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

Added receiveTimeout and sendTimeout properties to SourceEndpoint.

Browse Source
This commit is contained in:
Mark Fisher
2008-07-10 01:01:34 +00:00
parent 882cba8a07
commit a3a50488d8
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
org.springframework.integration/src/main/java/org/springframework/integration/endpoint/SourceEndpoint.java
View File

@@ -18,6 +18,7 @@ package org.springframework.integration.endpoint;
import org.springframework.integration.ConfigurationException;
import org.springframework.integration.channel.MessageChannel;
import org.springframework.integration.message.BlockingSource;
import org.springframework.integration.message.Message;
import org.springframework.integration.message.MessageDeliveryAware;
import org.springframework.integration.message.MessageDeliveryException;
@@ -34,6 +35,10 @@ public class SourceEndpoint extends AbstractEndpoint {
private final MessageSource<?> source;
private volatile long receiveTimeout = 5000;
private volatile long sendTimeout = -1;
public SourceEndpoint(MessageSource<?> source) {
Assert.notNull(source, "source must not be null");
@@ -41,6 +46,14 @@ public class SourceEndpoint extends AbstractEndpoint {
}
public void setReceiveTimeout(long receiveTimeout) {
this.receiveTimeout = receiveTimeout;
}
public void setSendTimeout(long sendTimeout) {
this.sendTimeout = sendTimeout;
}
@Override
public void initialize() {
if (this.getOutputChannelName() == null && this.getOutputChannel() == null) {
@@ -60,11 +73,12 @@ public class SourceEndpoint extends AbstractEndpoint {
}
public boolean poll() {
Message<?> message = this.source.receive();
Message<?> message = (this.source instanceof BlockingSource && this.receiveTimeout >= 0) ?
((BlockingSource<?>) this.source).receive(this.receiveTimeout) : this.source.receive();
if (message == null) {
return false;
}
boolean sent = this.getOutputChannel().send(message);
boolean sent = this.getOutputChannel().send(message, this.sendTimeout);
if (this.source instanceof MessageDeliveryAware) {
if (sent) {
((MessageDeliveryAware) this.source).onSend(message);