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

Make ConsumerContext disposable

Browse Source 
- This will allow shutting down the consumer connector for the consumer
configuration when the associated context no longer exists.
This commit is contained in:
Ilayaperumal Gopinathan
2014-08-18 06:33:20 -07:00
committed by Artem Bilan
parent bacf8d2ceb
commit 1829bd879f
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
spring-integration-kafka/src/main/java/org/springframework/integration/kafka/support/KafkaConsumerContext.java
View File

@@ -23,6 +23,7 @@ import java.util.Map;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.ListableBeanFactory;
import org.springframework.integration.kafka.core.KafkaConsumerDefaults;
import org.springframework.integration.support.MessageBuilder;
@@ -31,9 +32,10 @@ import org.springframework.util.CollectionUtils;
/**
* @author Soby Chacko
* @author Ilayaperumal Gopinathan
* @since 0.5
*/
public class KafkaConsumerContext<K,V> implements BeanFactoryAware {
public class KafkaConsumerContext<K,V> implements BeanFactoryAware, DisposableBean {
private Map<String, ConsumerConfiguration<K,V>> consumerConfigurations;
private String consumerTimeout = KafkaConsumerDefaults.CONSUMER_TIMEOUT;
private ZookeeperConnect zookeeperConnect;
@@ -77,4 +79,11 @@ public class KafkaConsumerContext<K,V> implements BeanFactoryAware {
public void setZookeeperConnect(final ZookeeperConnect zookeeperConnect) {
this.zookeeperConnect = zookeeperConnect;
}
@Override
public void destroy() throws Exception {
for (ConsumerConfiguration<K,V> config: consumerConfigurations.values()) {
config.getConsumerConnector().shutdown();
}
}
}