Added namespace support for the WireTap (INT-323).
This commit is contained in:
@@ -16,9 +16,6 @@
|
||||
|
||||
package org.springframework.integration.channel.interceptor;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.concurrent.CopyOnWriteArrayList;
|
||||
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
|
||||
@@ -39,40 +36,51 @@ import org.springframework.util.Assert;
|
||||
*/
|
||||
public class WireTap extends ChannelInterceptorAdapter implements Lifecycle {
|
||||
|
||||
private final Log logger = LogFactory.getLog(this.getClass());
|
||||
private static final Log logger = LogFactory.getLog(WireTap.class);
|
||||
|
||||
private final MessageTarget target;
|
||||
|
||||
private final List<MessageSelector> selectors = new CopyOnWriteArrayList<MessageSelector>();
|
||||
private volatile long timeout = 0;
|
||||
|
||||
private final MessageSelector selector;
|
||||
|
||||
private volatile boolean running = true;
|
||||
|
||||
|
||||
/**
|
||||
* Create a new wire tap with <em>no</em> {@link MessageSelector MessageSelectors}.
|
||||
* Create a new wire tap with <em>no</em> {@link MessageSelector}.
|
||||
*
|
||||
* @param target the MessageTarget to which intercepted messages will be sent
|
||||
*/
|
||||
public WireTap(MessageTarget target) {
|
||||
Assert.notNull(target, "target must not be null");
|
||||
this.target = target;
|
||||
this(target, null);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new wire tap with {@link MessageSelector MessageSelectors}.
|
||||
* Create a new wire tap with the provided {@link MessageSelector}.
|
||||
*
|
||||
* @param target the target to which intercepted messages will be sent
|
||||
* @param selectors the list of selectors that must accept a message for it to
|
||||
* @param selector the selector that must accept a message for it to
|
||||
* be sent to the intercepting target
|
||||
*/
|
||||
public WireTap(MessageTarget target, List<MessageSelector> selectors) {
|
||||
this(target);
|
||||
if (selectors != null) {
|
||||
this.selectors.addAll(selectors);
|
||||
}
|
||||
public WireTap(MessageTarget target, MessageSelector selector) {
|
||||
Assert.notNull(target, "target must not be null");
|
||||
this.target = target;
|
||||
this.selector = selector;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Specify the timeout value for sending to the intercepting target. Note
|
||||
* that this value will only apply if the target is a {@link BlockingTarget}.
|
||||
* The default value is 0.
|
||||
*
|
||||
* @param timeout the timeout in milliseconds
|
||||
*/
|
||||
public void setTimeout(long timeout) {
|
||||
this.timeout = timeout;
|
||||
}
|
||||
|
||||
/**
|
||||
* Check whether the wire tap is currently running.
|
||||
*/
|
||||
@@ -94,30 +102,22 @@ public class WireTap extends ChannelInterceptorAdapter implements Lifecycle {
|
||||
this.running = false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Intercept the Message and, <em>if accepted</em> by the {@link MessageSelector},
|
||||
* send it to the secondary target. If this wire tap's {@link MessageSelector} is
|
||||
* <code>null</code>, it will accept all messages.
|
||||
*/
|
||||
@Override
|
||||
public Message<?> preSend(Message<?> message, MessageChannel channel) {
|
||||
if (this.running && this.selectorsAccept(message)) {
|
||||
boolean sent = (this.target instanceof BlockingTarget) ?
|
||||
((BlockingTarget) this.target).send(message, 0) : this.target.send(message);
|
||||
if (this.running && (this.selector == null || this.selector.accept(message))) {
|
||||
boolean sent = (this.target instanceof BlockingTarget)
|
||||
? ((BlockingTarget) this.target).send(message, this.timeout)
|
||||
: this.target.send(message);
|
||||
if (!sent && logger.isWarnEnabled()) {
|
||||
logger.warn("failed to send message to WireTap target '" + this.target + "'");
|
||||
logger.warn("failed to send message to WireTap target '" + this.target + "'");
|
||||
}
|
||||
}
|
||||
return message;
|
||||
}
|
||||
|
||||
/**
|
||||
* If this wire tap has any {@link MessageSelector MessageSelectors}, check
|
||||
* whether they accept the current message. If any of them do not accept it,
|
||||
* the message will <em>not</em> be sent to the intercepting target.
|
||||
*/
|
||||
private boolean selectorsAccept(Message<?> message) {
|
||||
for (MessageSelector selector : this.selectors) {
|
||||
if (!selector.accept(message)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -16,6 +16,7 @@
|
||||
|
||||
package org.springframework.integration.config;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
@@ -25,7 +26,9 @@ public class ChannelInterceptorParser extends AbstractInterceptorParser {
|
||||
|
||||
@Override
|
||||
protected Map<String, BeanDefinitionRegisteringParser> getParserMap() {
|
||||
return null;
|
||||
Map<String, BeanDefinitionRegisteringParser> parsers = new HashMap<String, BeanDefinitionRegisteringParser>();
|
||||
parsers.put("wire-tap", new WireTapParser());
|
||||
return parsers;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,54 @@
|
||||
/*
|
||||
* Copyright 2002-2008 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.integration.config;
|
||||
|
||||
import org.w3c.dom.Element;
|
||||
|
||||
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
|
||||
import org.springframework.beans.factory.support.BeanDefinitionReaderUtils;
|
||||
import org.springframework.beans.factory.xml.ParserContext;
|
||||
import org.springframework.integration.ConfigurationException;
|
||||
import org.springframework.integration.channel.interceptor.WireTap;
|
||||
import org.springframework.util.StringUtils;
|
||||
|
||||
/**
|
||||
* Parser for the <wire-tap> element.
|
||||
*
|
||||
* @author Mark Fisher
|
||||
*/
|
||||
public class WireTapParser implements BeanDefinitionRegisteringParser {
|
||||
|
||||
public String parse(Element element, ParserContext parserContext) {
|
||||
BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition(WireTap.class);
|
||||
String targetRef = element.getAttribute("target");
|
||||
if (!StringUtils.hasText(targetRef)) {
|
||||
throw new ConfigurationException("the 'target' attribute is required");
|
||||
}
|
||||
builder.addConstructorArgReference(targetRef);
|
||||
String selectorRef = element.getAttribute("selector");
|
||||
if (StringUtils.hasText(selectorRef)) {
|
||||
builder.addConstructorArgReference(selectorRef);
|
||||
}
|
||||
String timeout = element.getAttribute("timeout");
|
||||
if (StringUtils.hasText(timeout)) {
|
||||
builder.addPropertyValue("timeout", Long.parseLong(timeout));
|
||||
}
|
||||
return BeanDefinitionReaderUtils.registerWithGeneratedName(
|
||||
builder.getBeanDefinition(), parserContext.getRegistry());
|
||||
}
|
||||
|
||||
}
|
||||
@@ -418,11 +418,23 @@
|
||||
<xsd:attribute name="bean" use="required"/>
|
||||
</xsd:complexType>
|
||||
</xsd:element>
|
||||
<xsd:element name="wire-tap" type="wireTapType" minOccurs="0" maxOccurs="unbounded"/>
|
||||
<xsd:any namespace="##other" processContents="strict" minOccurs="0" maxOccurs="unbounded"/>
|
||||
</xsd:choice>
|
||||
</xsd:sequence>
|
||||
</xsd:complexType>
|
||||
|
||||
<xsd:complexType name="wireTapType">
|
||||
<xsd:annotation>
|
||||
<xsd:documentation>
|
||||
Defines a Wire Tap Channel Interceptor.
|
||||
</xsd:documentation>
|
||||
</xsd:annotation>
|
||||
<xsd:attribute name="target" type="xsd:string" use="required"/>
|
||||
<xsd:attribute name="selector" type="xsd:string" use="optional"/>
|
||||
<xsd:attribute name="timeout" type="xsd:long" use="optional"/>
|
||||
</xsd:complexType>
|
||||
|
||||
<xsd:complexType name="endpointInterceptorsType">
|
||||
<xsd:annotation>
|
||||
<xsd:documentation>
|
||||
|
||||
Reference in New Issue
Block a user