package org.jgroups.blocks.mux;
import java.util.Collection;
import org.jgroups.Address;
import org.jgroups.Channel;
import org.jgroups.MembershipListener;
import org.jgroups.Message;
import org.jgroups.MessageListener;
import org.jgroups.UpHandler;
import org.jgroups.blocks.GroupRequest;
import org.jgroups.blocks.MessageDispatcher;
import org.jgroups.blocks.RequestCorrelator;
import org.jgroups.blocks.RequestHandler;
import org.jgroups.blocks.RequestOptions;
import org.jgroups.blocks.RspFilter;
/**
* A multiplexed message dispatcher.
* When used in conjunction with a MuxUpHandler, allows multiple dispatchers to use the same channel.
* <br/>
* Usage:<br/>
* <code>
* Channel c = new JChannel(...);<br/>
* c.setUpHandler(new MuxUpHandler());<br/>
* <br/>
* MessageDispatcher d1 = new MuxMessageDispatcher((short) 1, c, ...);<br/>
* MessageDispatcher d2 = new MuxMessageDispatcher((short) 2, c, ...);<br/>
* <br/>
* c.connect(...);<br/>
* </code>
* @author Paul Ferraro
* @version $Id: MuxMessageDispatcher.java,v 1.1 2010/04/13 17:57:07 ferraro Exp $
*/
public class MuxMessageDispatcher extends MessageDispatcher {
private final short id;
public MuxMessageDispatcher(short id) {
this.id = id;
}
public MuxMessageDispatcher(short id, Channel channel, MessageListener messageListener, MembershipListener membershipListener, RequestHandler handler) {
this(id);
setMessageListener(messageListener);
setMembershipListener(membershipListener);
setChannel(channel);
setRequestHandler(handler);
start();
}
private Muxer<UpHandler> getMuxer() {
UpHandler handler = channel.getUpHandler();
return ((handler != null) && (handler instanceof MuxUpHandler)) ? (MuxUpHandler) handler : null;
}
@Override
protected RequestCorrelator createRequestCorrelator(Object transport, RequestHandler handler, Address localAddr) {
// We can't set the scope of the request correlator here
// since this method is called from start() triggered in the
// MessageDispatcher constructor, when this.scope is not yet defined
return new MuxRequestCorrelator(id, transport, handler, localAddr);
}
@Override
public void start() {
super.start();
Muxer<UpHandler> muxer = this.getMuxer();
if (muxer != null) {
muxer.add(id, this.getProtocolAdapter());
}
}
@Override
public void stop() {
Muxer<UpHandler> muxer = this.getMuxer();
if (muxer != null) {
muxer.remove(id);
}
super.stop();
}
@Override
protected GroupRequest cast(Collection<Address> dests, Message msg, RequestOptions options, boolean blockForResults) {
RspFilter filter = options.getRspFilter();
return super.cast(dests, msg, options.setRspFilter((filter != null) ? new NoMuxHandlerRspFilter(filter) : new NoMuxHandlerRspFilter()), blockForResults);
}
}