use of com.yahoo.messagebus.network.Identity in project vespa by vespa-engine.
the class SessionCache method createSharedMessageBus.
private static SharedMessageBus createSharedMessageBus(final ContainerMbusConfig mbusConfig, final String slobrokConfigId, final String identity, Protocol protocol) {
final MessageBusParams mbusParams = new MessageBusParams().addProtocol(protocol);
final int maxPendingSize = DocumentUtil.calculateMaxPendingSize(mbusConfig.maxConcurrentFactor(), mbusConfig.documentExpansionFactor(), mbusConfig.containerCoreMemory());
logSystemInfo(mbusConfig, maxPendingSize);
mbusParams.setMaxPendingCount(mbusConfig.maxpendingcount());
mbusParams.setMaxPendingSize(maxPendingSize);
final RPCNetworkParams netParams = new RPCNetworkParams().setSlobrokConfigId(slobrokConfigId).setIdentity(new Identity(identity)).setListenPort(mbusConfig.port());
return SharedMessageBus.newInstance(mbusParams, netParams);
}
use of com.yahoo.messagebus.network.Identity in project vespa by vespa-engine.
the class VdsVisitTarget method run.
@SuppressWarnings("unchecked")
public void run() throws Exception {
initShutdownHook();
log.log(LogLevel.DEBUG, "Starting VdsVisitTarget");
MessageBusParams mbusParams = new MessageBusParams(new LoadTypeSet());
mbusParams.getRPCNetworkParams().setIdentity(new Identity(slobrokAddress));
if (port > 0) {
mbusParams.getRPCNetworkParams().setListenPort(port);
}
access = new MessageBusDocumentAccess(mbusParams);
VdsVisitHandler handler;
Class<?> cls = Thread.currentThread().getContextClassLoader().loadClass(handlerClassName);
try {
// Any custom data handlers may have a constructor that takes in args,
// so that the user can pass cmd line options to them
Class<?>[] consTypes = new Class<?>[] { boolean.class, boolean.class, boolean.class, boolean.class, boolean.class, boolean.class, int.class, String[].class };
Constructor<?> cons = cls.getConstructor(consTypes);
handler = (VdsVisitHandler) cons.newInstance(printIds, verbose, verbose, verbose, false, false, processTime, handlerArgs);
} catch (NoSuchMethodException e) {
// Retry, this time matching the StdOutVisitorHandler constructor
// arg list
Class<?>[] consTypes = new Class<?>[] { boolean.class, boolean.class, boolean.class, boolean.class, boolean.class, boolean.class, int.class, boolean.class };
Constructor<?> cons = cls.getConstructor(consTypes);
handler = (VdsVisitHandler) cons.newInstance(printIds, verbose, verbose, verbose, false, false, processTime, false);
}
VisitorDataHandler dataHandler = handler.getDataHandler();
VisitorControlHandler controlHandler = handler.getControlHandler();
VisitorDestinationParameters params = new VisitorDestinationParameters("visit-destination", dataHandler);
session = access.createVisitorDestinationSession(params);
while (!controlHandler.isDone()) {
Thread.sleep(1000);
}
}
use of com.yahoo.messagebus.network.Identity in project vespa by vespa-engine.
the class MessageBusDocumentApiTestCase method setUp.
@Before
public void setUp() throws ListenFailedException {
slobrok = new Slobrok();
String slobrokConfigId = "raw:slobrok[1]\n" + "slobrok[0].connectionspec tcp/localhost:" + slobrok.port() + "\n";
MessageBusParams params = new MessageBusParams();
params.getRPCNetworkParams().setIdentity(new Identity("test/feeder"));
params.getRPCNetworkParams().setSlobrokConfigId(slobrokConfigId);
params.setDocumentManagerConfigId("file:src/test/cfg/documentmanager.cfg");
params.setRouteName("Route");
params.setRoutingConfigId("file:src/test/cfg/messagebus.cfg");
params.setTraceLevel(9);
params.setSourceSessionParams(new SourceSessionParams().setThrottlePolicy(null));
access = new MessageBusDocumentAccess(params);
destination = new VisitableDestination(slobrokConfigId, params.getDocumentManagerConfigId());
}
use of com.yahoo.messagebus.network.Identity in project vespa by vespa-engine.
the class SendAdapterTestCase method setUp.
@Before
public void setUp() throws ListenFailedException, UnknownHostException {
slobrok = new Slobrok();
dstServer = new TestServer(new MessageBusParams().addProtocol(dstProtocol = new TestProtocol()), new RPCNetworkParams().setIdentity(new Identity("dst")).setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
dstSession = dstServer.mb.createDestinationSession(new DestinationSessionParams().setName("session").setMessageHandler(new Receptor()));
itrServer = new TestServer(new MessageBusParams().addProtocol(itrProtocol = new TestProtocol()), new RPCNetworkParams().setIdentity(new Identity("itr")).setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
itrSession = itrServer.mb.createIntermediateSession(new IntermediateSessionParams().setName("session").setMessageHandler(new Receptor()).setReplyHandler(new Receptor()));
srcServer = new TestServer(new MessageBusParams().addProtocol(srcProtocol = new TestProtocol()), new RPCNetworkParams().setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
srcSession = srcServer.mb.createSourceSession(new SourceSessionParams().setTimeout(600.0).setReplyHandler(new Receptor()));
assertTrue(srcServer.waitSlobrok("*/session", 2));
}
use of com.yahoo.messagebus.network.Identity in project vespa by vespa-engine.
the class ChokeTestCase method setUp.
@Before
public void setUp() throws ListenFailedException, UnknownHostException {
slobrok = new Slobrok();
dstServer = new TestServer(new MessageBusParams().addProtocol(new SimpleProtocol()), new RPCNetworkParams().setIdentity(new Identity("dst")).setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
dstSession = dstServer.mb.createDestinationSession(new DestinationSessionParams().setName("session").setMessageHandler(new Receptor()));
srcServer = new TestServer(new MessageBusParams().setRetryPolicy(null).addProtocol(new SimpleProtocol()), new RPCNetworkParams().setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
srcSession = srcServer.mb.createSourceSession(new SourceSessionParams().setTimeout(600.0).setThrottlePolicy(null).setReplyHandler(new Receptor()));
assertTrue(srcServer.waitSlobrok("dst/session", 1));
}
Aggregations