SpringCloud/spring-cloud-stream
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Also honour ApplicationContextAware

Browse Source
This commit is contained in:
Marius Bogoevici
2015-08-18 17:05:08 -04:00
parent 79c13a8ab2
commit 108ac82df3
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
spring-cloud-stream/src/main/java/org/springframework/cloud/stream/config/DirectChannelFactoryBean.java
View File

@@ -21,7 +21,10 @@ import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.beans.factory.BeanNameAware;
import org.springframework.beans.factory.FactoryBean;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.integration.channel.DirectChannel;
import org.springframework.util.Assert;
/**
* {@link FactoryBean} for creating channels for fields annotated with
@@ -30,13 +33,16 @@ import org.springframework.integration.channel.DirectChannel;
*
* @author Marius Bogoevici
*/
public class DirectChannelFactoryBean implements FactoryBean<DirectChannel>, BeanNameAware, BeanFactoryAware {
public class DirectChannelFactoryBean implements FactoryBean<DirectChannel>, BeanNameAware, BeanFactoryAware, ApplicationContextAware {
private DirectChannel directChannel;
private String beanName;
private BeanFactory beanFactory;
private ApplicationContext applicationContext;
@Override
public void setBeanName(String beanName) {
this.beanName = beanName;
@@ -44,9 +50,16 @@ public class DirectChannelFactoryBean implements FactoryBean<DirectChannel>, Bea
@Override
public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
Assert.notNull(beanFactory, "'beanFactory' must not be null");
this.beanFactory = beanFactory;
}
@Override
public void setApplicationContext(ApplicationContext applicationContext) {
Assert.notNull(beanFactory, "'applicationContext' must not be null");
this.applicationContext = applicationContext;
}
@Override
public synchronized DirectChannel getObject() throws Exception {
if (directChannel == null) {
@@ -54,6 +67,7 @@ public class DirectChannelFactoryBean implements FactoryBean<DirectChannel>, Bea
}
directChannel.setBeanName(beanName);
directChannel.setBeanFactory(beanFactory);
directChannel.setApplicationContext(applicationContext);
return directChannel;
}