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

INT-1410 SourcePollingChannelAdapter is now a TrackableComponent

Browse Source
This commit is contained in:
Mark Fisher
2010-09-02 20:10:50 +00:00
parent f78794f900
commit 4948058482
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
spring-integration-core/src/main/java/org/springframework/integration/endpoint/SourcePollingChannelAdapter.java
View File

@@ -20,6 +20,8 @@ import org.springframework.integration.Message;
import org.springframework.integration.MessageChannel;
import org.springframework.integration.core.MessageSource;
import org.springframework.integration.core.MessagingTemplate;
import org.springframework.integration.history.MessageHistory;
import org.springframework.integration.history.TrackableComponent;
import org.springframework.util.Assert;
/**
@@ -28,12 +30,14 @@ import org.springframework.util.Assert;
*
* @author Mark Fisher
*/
public class SourcePollingChannelAdapter extends AbstractPollingEndpoint {
public class SourcePollingChannelAdapter extends AbstractPollingEndpoint implements TrackableComponent {
private volatile MessageSource<?> source;
private volatile MessageChannel outputChannel;
private volatile boolean shouldTrack;
private final MessagingTemplate messagingTemplate = new MessagingTemplate();
@@ -59,6 +63,13 @@ public class SourcePollingChannelAdapter extends AbstractPollingEndpoint {
this.messagingTemplate.setSendTimeout(sendTimeout);
}
/**
* Specify whether this component should be tracked in the Message History.
*/
public void setShouldTrack(boolean shouldTrack) {
this.shouldTrack = shouldTrack;
}
@Override
protected void onInit() {
Assert.notNull(this.source, "source must not be null");
@@ -75,6 +86,9 @@ public class SourcePollingChannelAdapter extends AbstractPollingEndpoint {
protected boolean doPoll() {
Message<?> message = this.source.receive();
if (message != null) {
if (this.shouldTrack) {
message = MessageHistory.write(message, this);
}
this.messagingTemplate.send(this.outputChannel, message);
return true;
}