resolves INT-582
Includes refactoring of router hierarchy to avoid duplication of channel resolution logic Subclasses of AbstractChannelNameResolvingMessageRouter can return MessageChannel , MessageChannel[], collection ... in addition to String MethodInvokingRouter now extends AbstractChannelNameResolvingMessageRouter and is simplified as a result
This commit is contained in:
@@ -19,6 +19,7 @@ package org.springframework.integration.config;
|
||||
import org.springframework.integration.channel.ChannelResolver;
|
||||
import org.springframework.integration.core.MessageChannel;
|
||||
import org.springframework.integration.message.MessageHandler;
|
||||
import org.springframework.integration.router.AbstractChannelNameResolvingMessageRouter;
|
||||
import org.springframework.integration.router.AbstractMessageRouter;
|
||||
import org.springframework.integration.router.MethodInvokingRouter;
|
||||
import org.springframework.util.Assert;
|
||||
@@ -28,13 +29,17 @@ import org.springframework.util.StringUtils;
|
||||
* Factory bean for creating a Message Router.
|
||||
*
|
||||
* @author Mark Fisher
|
||||
* @author Jonas Partner
|
||||
*/
|
||||
public class RouterFactoryBean extends AbstractMessageHandlerFactoryBean {
|
||||
|
||||
private volatile ChannelResolver channelResolver;
|
||||
|
||||
private volatile MessageChannel defaultOutputChannel;
|
||||
|
||||
private volatile boolean resolutionRequired;
|
||||
|
||||
private volatile Boolean ignoreChannelNameResolutionFailures;
|
||||
|
||||
public void setChannelResolver(ChannelResolver channelResolver) {
|
||||
this.channelResolver = channelResolver;
|
||||
@@ -43,6 +48,14 @@ public class RouterFactoryBean extends AbstractMessageHandlerFactoryBean {
|
||||
public void setDefaultOutputChannel(MessageChannel defaultOutputChannel) {
|
||||
this.defaultOutputChannel = defaultOutputChannel;
|
||||
}
|
||||
|
||||
public void setResolutionRequired(boolean resolutionRequired){
|
||||
this.resolutionRequired = resolutionRequired;
|
||||
}
|
||||
|
||||
public void setIgnoreChannelNameResolutionFailures(boolean ignoreChannelNameResolutionFailures){
|
||||
this.ignoreChannelNameResolutionFailures = ignoreChannelNameResolutionFailures;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected MessageHandler createHandler(Object targetObject, String targetMethodName) {
|
||||
@@ -51,6 +64,12 @@ public class RouterFactoryBean extends AbstractMessageHandlerFactoryBean {
|
||||
if (this.defaultOutputChannel != null) {
|
||||
router.setDefaultOutputChannel(this.defaultOutputChannel);
|
||||
}
|
||||
if(ignoreChannelNameResolutionFailures != null ){
|
||||
Assert.isTrue(router instanceof AbstractChannelNameResolvingMessageRouter,
|
||||
"Ignore channel name resolution failures can only set on rotuers extending " + AbstractChannelNameResolvingMessageRouter.class.getName());
|
||||
((AbstractChannelNameResolvingMessageRouter)router).setIgnoreChannelNameResolutionFailures(ignoreChannelNameResolutionFailures);
|
||||
}
|
||||
router.setResolutionRequired(resolutionRequired);
|
||||
return router;
|
||||
}
|
||||
|
||||
|
||||
@@ -49,6 +49,8 @@ public class RouterParser extends AbstractConsumerEndpointParser {
|
||||
resolverBuilder.getBeanDefinition(), parserContext.getRegistry());
|
||||
builder.addPropertyReference("channelResolver", resolverBeanName);
|
||||
IntegrationNamespaceUtils.setReferenceIfAttributeDefined(builder, element, "default-output-channel");
|
||||
IntegrationNamespaceUtils.setValueIfAttributeDefined(builder, element, "resolution-required");
|
||||
IntegrationNamespaceUtils.setValueIfAttributeDefined(builder, element, "ignore-channel-name-resolution-failures");
|
||||
return builder;
|
||||
}
|
||||
|
||||
|
||||
@@ -788,6 +788,8 @@
|
||||
</xsd:annotation>
|
||||
</xsd:attribute>
|
||||
<xsd:attribute name="method" type="xsd:string" />
|
||||
<xsd:attribute name="resolution-required" type="xsd:boolean" />
|
||||
<xsd:attribute name="ignore-channel-name-resolution-failures" type="xsd:boolean" />
|
||||
</xsd:extension>
|
||||
</xsd:complexContent>
|
||||
</xsd:complexType>
|
||||
|
||||
@@ -17,12 +17,15 @@
|
||||
package org.springframework.integration.router;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
import java.util.List;
|
||||
|
||||
import org.springframework.beans.factory.BeanFactory;
|
||||
import org.springframework.beans.factory.BeanFactoryAware;
|
||||
import org.springframework.beans.factory.InitializingBean;
|
||||
import org.springframework.integration.channel.BeanFactoryChannelResolver;
|
||||
import org.springframework.integration.channel.ChannelResolutionException;
|
||||
import org.springframework.integration.channel.ChannelResolver;
|
||||
import org.springframework.integration.core.Message;
|
||||
import org.springframework.integration.core.MessageChannel;
|
||||
@@ -30,12 +33,14 @@ import org.springframework.integration.core.MessagingException;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
/**
|
||||
* A base class for router implementations that return only
|
||||
* the channel name(s) rather than {@link MessageChannel} instances.
|
||||
* A base class for router implementations that return only the channel name(s)
|
||||
* rather than {@link MessageChannel} instances.
|
||||
*
|
||||
* @author Mark Fisher
|
||||
* @author Jonas Partner
|
||||
*/
|
||||
public abstract class AbstractChannelNameResolvingMessageRouter extends AbstractMessageRouter implements BeanFactoryAware, InitializingBean {
|
||||
public abstract class AbstractChannelNameResolvingMessageRouter extends
|
||||
AbstractMessageRouter implements BeanFactoryAware, InitializingBean {
|
||||
|
||||
private volatile ChannelResolver channelResolver;
|
||||
|
||||
@@ -45,6 +50,7 @@ public abstract class AbstractChannelNameResolvingMessageRouter extends Abstract
|
||||
|
||||
private volatile BeanFactory beanFactory;
|
||||
|
||||
private volatile boolean ignoreChannelNameResolutionFailures;
|
||||
|
||||
public void setChannelResolver(ChannelResolver channelResolver) {
|
||||
this.channelResolver = channelResolver;
|
||||
@@ -62,45 +68,97 @@ public abstract class AbstractChannelNameResolvingMessageRouter extends Abstract
|
||||
this.beanFactory = beanFactory;
|
||||
}
|
||||
|
||||
public void setIgnoreChannelNameResolutionFailures(
|
||||
boolean ignoreChannelNameResolutionFailures) {
|
||||
this.ignoreChannelNameResolutionFailures = ignoreChannelNameResolutionFailures;
|
||||
}
|
||||
|
||||
public void afterPropertiesSet() {
|
||||
if (this.channelResolver == null) {
|
||||
Assert.notNull(beanFactory, "either a ChannelResolver or BeanFactory is required");
|
||||
this.channelResolver = new BeanFactoryChannelResolver(this.beanFactory);
|
||||
Assert.notNull(beanFactory,
|
||||
"either a ChannelResolver or BeanFactory is required");
|
||||
this.channelResolver = new BeanFactoryChannelResolver(
|
||||
this.beanFactory);
|
||||
}
|
||||
}
|
||||
|
||||
protected MessageChannel resolveChannelForName(String channelName,
|
||||
Message<?> message) {
|
||||
Assert.state(this.channelResolver != null,
|
||||
"unable to resolve channel names, no ChannelResolver available");
|
||||
|
||||
MessageChannel channel = null;
|
||||
try {
|
||||
channel = this.channelResolver.resolveChannelName(channelName);
|
||||
} catch (ChannelResolutionException e) {
|
||||
if (!ignoreChannelNameResolutionFailures)
|
||||
throw new MessagingException(message,
|
||||
"failed to resolve channel name '" + channelName + "'", e);
|
||||
}
|
||||
if(channel == null && !ignoreChannelNameResolutionFailures){
|
||||
throw new MessagingException(message,
|
||||
"failed to resolve channel name '" + channelName + "'");
|
||||
}
|
||||
return channel;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected final Collection<MessageChannel> determineTargetChannels(Message<?> message) {
|
||||
protected Collection<MessageChannel> determineTargetChannels(
|
||||
Message<?> message) {
|
||||
this.afterPropertiesSet();
|
||||
Collection<MessageChannel> channels = new ArrayList<MessageChannel>();
|
||||
String[] channelNames = this.determineTargetChannelNames(message);
|
||||
if (channelNames == null) {
|
||||
Collection<Object> channelsReturned = this.getChannelIndicatorList(message);
|
||||
addToCollection(channels, channelsReturned, message);
|
||||
if (channels == null) {
|
||||
return null;
|
||||
}
|
||||
for (String channelName : channelNames) {
|
||||
if (channelName != null) {
|
||||
Assert.state(this.channelResolver != null,
|
||||
"unable to resolve channel names, no ChannelResolver available");
|
||||
if (this.prefix != null) {
|
||||
channelName = this.prefix + channelName;
|
||||
|
||||
return channels;
|
||||
}
|
||||
|
||||
|
||||
protected void addToCollection(Collection<MessageChannel> channels, Collection<?> channelIndicators, Message<?> message){
|
||||
if(channelIndicators == null){
|
||||
return;
|
||||
}
|
||||
for (Object channelIndicator : channelIndicators) {
|
||||
if (channelIndicator == null) {
|
||||
continue;
|
||||
} else if (channelIndicator instanceof String) {
|
||||
addChannelFromString(channels, (String)channelIndicator, message);
|
||||
} else if (channelIndicator instanceof MessageChannel){
|
||||
channels.add((MessageChannel)channelIndicator);
|
||||
} else if (channelIndicator instanceof Collection){
|
||||
addToCollection(channels, (Collection<?>)channelIndicator, message);
|
||||
} else if (channelIndicator instanceof MessageChannel[]) {
|
||||
channels.addAll(Arrays.asList((MessageChannel[]) channelIndicator));
|
||||
} else if (channelIndicator instanceof String[]) {
|
||||
for (String indicatorName : (String[]) channelIndicator) {
|
||||
addChannelFromString(channels, indicatorName, message);
|
||||
}
|
||||
if (this.suffix != null) {
|
||||
channelName = channelName + suffix;
|
||||
}
|
||||
MessageChannel channel = this.channelResolver.resolveChannelName(channelName);
|
||||
if (channel == null) {
|
||||
throw new MessagingException(message,
|
||||
"failed to resolve channel name '" + channelName + "'");
|
||||
}
|
||||
channels.add(channel);
|
||||
}else {
|
||||
throw new MessagingException("unsupported return type for router [" + channelIndicator.getClass() + "]");
|
||||
}
|
||||
}
|
||||
return channels;
|
||||
}
|
||||
|
||||
|
||||
protected void addChannelFromString(Collection<MessageChannel> channels, String channelName, Message<?> message){
|
||||
if (this.prefix != null) {
|
||||
channelName = this.prefix + channelName;
|
||||
}
|
||||
if (this.suffix != null) {
|
||||
channelName = channelName + suffix;
|
||||
}
|
||||
MessageChannel channel = resolveChannelForName(channelName,message);
|
||||
if (channel != null) {
|
||||
channels.add(channel);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Subclasses must implement this method to return the channel name(s).
|
||||
* Subclasses must implement this method to return the channel indicators.
|
||||
*/
|
||||
protected abstract String[] determineTargetChannelNames(Message<?> message);
|
||||
protected abstract List<Object> getChannelIndicatorList(Message<?> message);
|
||||
|
||||
}
|
||||
|
||||
@@ -16,6 +16,10 @@
|
||||
|
||||
package org.springframework.integration.router;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collection;
|
||||
import java.util.List;
|
||||
|
||||
import org.springframework.integration.core.Message;
|
||||
|
||||
/**
|
||||
@@ -27,9 +31,13 @@ import org.springframework.integration.core.Message;
|
||||
public abstract class AbstractSingleChannelNameRouter extends AbstractChannelNameResolvingMessageRouter {
|
||||
|
||||
@Override
|
||||
protected final String[] determineTargetChannelNames(Message<?> message) {
|
||||
String channelName = this.determineTargetChannelName(message);
|
||||
return (channelName != null) ? new String[] { channelName } : null;
|
||||
protected final List<Object> getChannelIndicatorList(Message<?> message) {
|
||||
List<Object> channelList = new ArrayList<Object>();
|
||||
String channelName = determineTargetChannelName(message);
|
||||
if(channelName != null){
|
||||
channelList.add(channelName);
|
||||
}
|
||||
return channelList;
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -18,17 +18,13 @@ package org.springframework.integration.router;
|
||||
|
||||
import java.lang.reflect.Method;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
import java.util.List;
|
||||
|
||||
import org.springframework.integration.annotation.Router;
|
||||
import org.springframework.integration.channel.ChannelResolver;
|
||||
import org.springframework.integration.core.Message;
|
||||
import org.springframework.integration.core.MessageChannel;
|
||||
import org.springframework.integration.core.MessagingException;
|
||||
import org.springframework.integration.handler.MessageMappingMethodInvoker;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
/**
|
||||
* A Message Router that invokes the specified method on the given object. The
|
||||
@@ -39,12 +35,11 @@ import org.springframework.util.Assert;
|
||||
*
|
||||
* @author Mark Fisher
|
||||
*/
|
||||
public class MethodInvokingRouter extends AbstractMessageRouter {
|
||||
public class MethodInvokingRouter extends AbstractChannelNameResolvingMessageRouter {
|
||||
|
||||
private final MessageMappingMethodInvoker invoker;
|
||||
|
||||
private volatile ChannelResolver channelResolver;
|
||||
|
||||
|
||||
|
||||
public MethodInvokingRouter(Object object, Method method) {
|
||||
this.invoker = new MessageMappingMethodInvoker(object, method);
|
||||
@@ -58,68 +53,12 @@ public class MethodInvokingRouter extends AbstractMessageRouter {
|
||||
this.invoker = new MessageMappingMethodInvoker(object, Router.class);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Provide the {@link ChannelResolver} strategy to use for methods that
|
||||
* return a channel name rather than a {@link MessageChannel} instance.
|
||||
*/
|
||||
public void setChannelResolver(ChannelResolver channelResolver) {
|
||||
this.channelResolver = channelResolver;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected final Collection<MessageChannel> determineTargetChannels(Message<?> message) {
|
||||
protected List<Object> getChannelIndicatorList(Message<?> message) {
|
||||
Object result = this.invoker.invokeMethod(message);
|
||||
if (result == null) {
|
||||
return null;
|
||||
}
|
||||
List<MessageChannel> channels = new ArrayList<MessageChannel>();
|
||||
if (result instanceof Collection) {
|
||||
for (Object next : (Collection<?>) result) {
|
||||
this.addChannel(next, channels);
|
||||
}
|
||||
}
|
||||
else if (result instanceof MessageChannel[]) {
|
||||
channels.addAll(Arrays.asList((MessageChannel[]) result));
|
||||
}
|
||||
else if (result instanceof String[]) {
|
||||
for (String channelName : (String[]) result) {
|
||||
this.addChannel(channelName, channels);
|
||||
}
|
||||
}
|
||||
else if (result instanceof MessageChannel) {
|
||||
channels.add((MessageChannel) result);
|
||||
}
|
||||
else if (result instanceof String) {
|
||||
this.addChannel((String) result, channels);
|
||||
}
|
||||
else {
|
||||
throw new IllegalStateException(
|
||||
"router method must return type 'MessageChannel' or 'String'");
|
||||
}
|
||||
return channels;
|
||||
}
|
||||
|
||||
private void addChannel(Object channelOrName, List<MessageChannel> channels) {
|
||||
if (channelOrName == null) {
|
||||
return;
|
||||
}
|
||||
if (channelOrName instanceof MessageChannel) {
|
||||
channels.add((MessageChannel) channelOrName);
|
||||
}
|
||||
else if (channelOrName instanceof String) {
|
||||
String channelName = (String) channelOrName;
|
||||
Assert.state(this.channelResolver != null,
|
||||
"unable to resolve channel names, no ChannelResolver available");
|
||||
MessageChannel channel = this.channelResolver.resolveChannelName(channelName);
|
||||
if (channel == null) {
|
||||
throw new MessagingException("failed to resolve channel name '" + channelName + "'");
|
||||
}
|
||||
channels.add(channel);
|
||||
}
|
||||
else {
|
||||
throw new MessagingException("unsupported return type for router [" + channelOrName.getClass() + "]");
|
||||
}
|
||||
List<Object> asList = new ArrayList<Object>();
|
||||
asList.add(result);
|
||||
return asList;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user