Spring/spring-integration
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

INT-1397 MessageProducerSupport now implements TrackableComponent

Browse Source
This commit is contained in:
Mark Fisher
2010-09-01 20:00:23 +00:00
parent 3581ae95d7
commit 110cde02a9
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
spring-integration-core/src/main/java/org/springframework/integration/endpoint/MessageProducerSupport.java
View File

@@ -18,9 +18,11 @@ package org.springframework.integration.endpoint;
import org.springframework.integration.Message;
import org.springframework.integration.MessageChannel;
import org.springframework.integration.MessagingException;
import org.springframework.integration.core.MessageProducer;
import org.springframework.integration.core.MessagingTemplate;
import org.springframework.integration.history.MessageHistory;
import org.springframework.integration.history.TrackableComponent;
import org.springframework.util.Assert;
/**
@@ -29,10 +31,12 @@ import org.springframework.util.Assert;
*
* @author Mark Fisher
*/
public abstract class MessageProducerSupport extends AbstractEndpoint implements MessageProducer {
public abstract class MessageProducerSupport extends AbstractEndpoint implements MessageProducer, TrackableComponent {
private volatile MessageChannel outputChannel;
private volatile boolean shouldTrack = false;
private final MessagingTemplate messagingTemplate = new MessagingTemplate();
@@ -44,13 +48,20 @@ public abstract class MessageProducerSupport extends AbstractEndpoint implements
this.messagingTemplate.setSendTimeout(sendTimeout);
}
public void setShouldTrack(boolean shouldTrack) {
this.shouldTrack = shouldTrack;
}
@Override
protected void onInit() {
Assert.notNull(this.outputChannel, "outputChannel is required");
}
protected void sendMessage(Message<?> message) {
if (message != null) {
if (message == null) {
throw new MessagingException("cannot send a null message");
}
if (this.shouldTrack) {
message = MessageHistory.write(message, this);
}
this.messagingTemplate.send(this.outputChannel, message);