Examples with DocumentProtocol com.yahoo.documentapi.messagebus.protocol.DocumentProtocol used on opensource projects

Example 1 with DocumentProtocol

use of com.yahoo.documentapi.messagebus.protocol.DocumentProtocol in project vespa by vespa-engine.

the class DocumentProcessingHandlerTestBase method createHandler.

@Before
public void createHandler() {
    documentTypeManager.register(getType());
    Protocol protocol = new DocumentProtocol(documentTypeManager);
    driver = ServerTestDriver.newInactiveInstanceWithProtocol(protocol);
    sessionCache = new SessionCache("raw:", driver.client().slobrokId(), "test", "raw:", null, "raw:", documentTypeManager);
    ContainerBuilder builder = driver.parent().newContainerBuilder();
    ComponentRegistry<DocprocService> registry = new ComponentRegistry<>();
    handler = new DocumentProcessingHandler(registry, new ComponentRegistry<>(), new ComponentRegistry<>(), new DocumentProcessingHandlerParameters().setDocumentTypeManager(documentTypeManager).setContainerDocumentConfig(new ContainerDocumentConfig(new ContainerDocumentConfig.Builder())));
    builder.serverBindings().bind("mbus://*/*", handler);
    ReferencedResource<SharedSourceSession> sessionRef = sessionCache.retainSource(new SourceSessionParams());
    MbusClient sourceClient = new MbusClient(sessionRef.getResource());
    builder.clientBindings().bind("mbus://*/source", sourceClient);
    builder.clientBindings().bind("mbus://*/" + MbusRequestContext.internalNoThrottledSource, sourceClient);
    sourceClient.start();
    List<Pair<String, CallStack>> callStacks = getCallStacks();
    List<AbstractResource> resources = new ArrayList<>();
    for (Pair<String, CallStack> callStackPair : callStacks) {
        DocprocService service = new DocprocService(callStackPair.getFirst());
        service.setCallStack(callStackPair.getSecond());
        service.setInService(true);
        ComponentId serviceId = new ComponentId(service.getName());
        registry.register(serviceId, service);
        ComponentId sessionName = ComponentId.fromString("chain." + serviceId);
        MbusServerProvider serviceProvider = new MbusServerProvider(sessionName, sessionCache, driver.parent());
        serviceProvider.get().start();
        serviceProviders.add(serviceProvider);
        MbusClient intermediateClient = new MbusClient(serviceProvider.getSession());
        builder.clientBindings().bind("mbus://*/" + sessionName.stringValue(), intermediateClient);
        intermediateClient.start();
        resources.add(intermediateClient);
    }
    driver.parent().activateContainer(builder);
    sessionRef.getReference().close();
    sourceClient.release();
    for (AbstractResource resource : resources) {
        resource.release();
    }
    remoteServer = RemoteServer.newInstance(driver.client().slobrokId(), "foobar", protocol);
}

Example 2 with DocumentProtocol

use of com.yahoo.documentapi.messagebus.protocol.DocumentProtocol in project vespa by vespa-engine.

the class PolicyFactoryTestCase method testFactory.

// //////////////////////////////////////////////////////////////////////////////
// 
// Tests
// 
// //////////////////////////////////////////////////////////////////////////////
public void testFactory() {
    Route route = Route.parse("[MyPolicy]");
    assertTrue(src.send(createMessage(), route).isAccepted());
    Reply reply = ((Receptor) src.getReplyHandler()).getReply(60);
    assertNotNull(reply);
    System.out.println(reply.getTrace());
    assertEquals(1, reply.getNumErrors());
    assertEquals(ErrorCode.UNKNOWN_POLICY, reply.getError(0).getCode());
    Protocol obj = srv.mb.getProtocol(DocumentProtocol.NAME);
    assertTrue(obj instanceof DocumentProtocol);
    DocumentProtocol protocol = (DocumentProtocol) obj;
    protocol.putRoutingPolicyFactory("MyPolicy", new MyFactory());
    assertTrue(src.send(createMessage(), route).isAccepted());
    assertNotNull(reply = ((Receptor) src.getReplyHandler()).getReply(60));
    System.out.println(reply.getTrace());
    assertEquals(1, reply.getNumErrors());
    assertEquals(DocumentProtocol.ERROR_POLICY_FAILURE, reply.getError(0).getCode());
}

Example 3 with DocumentProtocol

use of com.yahoo.documentapi.messagebus.protocol.DocumentProtocol in project vespa by vespa-engine.

the class SessionCache method start.

private void start() {
    ContainerMbusConfig mbusConfig = ConfigGetter.getConfig(ContainerMbusConfig.class, containerMbusConfigId);
    if (documentManagerConfigId != null) {
        documentTypeManager.configure(documentManagerConfigId);
    }
    LoadTypeSet loadTypeSet = new LoadTypeSet(loadTypeConfigId);
    DocumentProtocol protocol = new DocumentProtocol(documentTypeManager, identity, loadTypeSet);
    messageBus = createSharedMessageBus(mbusConfig, slobrokConfigId, identity, protocol);
    // TODO: stop doing subscriptions to config when that is to be solved in slobrok as well
    configAgent = new ConfigAgent(messagebusConfigId, messageBus.messageBus());
    configAgent.subscribe();
}

Example 4 with DocumentProtocol

use of com.yahoo.documentapi.messagebus.protocol.DocumentProtocol in project vespa by vespa-engine.

the class PolicyFactoryTestCase method setUp.

@Override
public void setUp() throws ListenFailedException {
    slobrok = new Slobrok();
    srv = new TestServer(new MessageBusParams().addProtocol(new DocumentProtocol(new DocumentTypeManager())), new RPCNetworkParams().setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
    src = srv.mb.createSourceSession(new SourceSessionParams().setReplyHandler(new Receptor()));
}

Example 5 with DocumentProtocol

use of com.yahoo.documentapi.messagebus.protocol.DocumentProtocol in project vespa by vespa-engine.

the class RoutableFactoryTestCase method setUp.

@Override
public void setUp() throws ListenFailedException {
    slobrok = new Slobrok();
    DocumentTypeManager docMan = new DocumentTypeManager();
    dstProtocol = new DocumentProtocol(docMan);
    dstServer = new TestServer(new MessageBusParams().addProtocol(dstProtocol), new RPCNetworkParams().setIdentity(new Identity("dst")).setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
    dstSession = dstServer.mb.createDestinationSession(new DestinationSessionParams().setName("session").setMessageHandler(new Receptor()));
    srcProtocol = new DocumentProtocol(docMan);
    srcServer = new TestServer(new MessageBusParams().addProtocol(srcProtocol), new RPCNetworkParams().setSlobrokConfigId(TestServer.getSlobrokConfig(slobrok)));
    srcSession = srcServer.mb.createSourceSession(new SourceSessionParams().setReplyHandler(new Receptor()));
    assertTrue(srcServer.waitSlobrok("dst/session", 1));
}