Examples with SourceSessionParams com.yahoo.messagebus.SourceSessionParams used on opensource projects

Example 6 with SourceSessionParams

use of com.yahoo.messagebus.SourceSessionParams 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 7 with SourceSessionParams

use of com.yahoo.messagebus.SourceSessionParams in project vespa by vespa-engine.

the class MbusSessionKeyTestCase method sourceSessionKey.

// TODO the session key tests are just smoke tests
@Test
public final void sourceSessionKey() {
    final SourceSessionParams base = new SourceSessionParams();
    final SourceSessionParams other = new SourceSessionParams();
    assertEquals(new SourceSessionKey(base), new SourceSessionKey(other));
    other.setTimeout(other.getTimeout() + 1);
    assertFalse(new SourceSessionKey(base).equals(new SourceSessionKey(other)));
}

Example 8 with SourceSessionParams

use of com.yahoo.messagebus.SourceSessionParams in project vespa by vespa-engine.

the class FeedHandlerV3 method sourceSessionParams.

private SourceSessionParams sourceSessionParams(HttpRequest request) {
    SourceSessionParams params = new SourceSessionParams();
    String timeout = request.getHeader(Headers.TIMEOUT);
    if (timeout != null) {
        try {
            params.setTimeout(Double.parseDouble(timeout));
        } catch (NumberFormatException e) {
        // NOP
        }
    }
    return params;
}

Example 9 with SourceSessionParams

use of com.yahoo.messagebus.SourceSessionParams in project vespa by vespa-engine.

the class FeedHandlerV3 method handle.

// TODO: If this is set up to run without first invoking the old FeedHandler code, we should
// verify the version header first. This is done in the old code.
@Override
public HttpResponse handle(HttpRequest request) {
    final String clientId = clientId(request);
    final ClientFeederV3 clientFeederV3;
    synchronized (monitor) {
        if (!clientFeederByClientId.containsKey(clientId)) {
            SourceSessionParams sourceSessionParams = sourceSessionParams(request);
            clientFeederByClientId.put(clientId, new ClientFeederV3(retainSource(sessionCache, sourceSessionParams), new FeedReaderFactory(), docTypeManager, clientId, metric, feedReplyHandler, threadsAvailableForFeeding));
        }
        clientFeederV3 = clientFeederByClientId.get(clientId);
    }
    try {
        return clientFeederV3.handleRequest(request);
    } catch (UnknownClientException uce) {
        String msg = Exceptions.toMessageString(uce);
        log.log(LogLevel.WARNING, msg);
        return new ErrorHttpResponse(com.yahoo.jdisc.http.HttpResponse.Status.BAD_REQUEST, msg);
    } catch (Exception e) {
        String msg = "Could not initialize document parsing: " + Exceptions.toMessageString(e);
        log.log(LogLevel.WARNING, msg);
        return new ErrorHttpResponse(com.yahoo.jdisc.http.HttpResponse.Status.INTERNAL_SERVER_ERROR, msg);
    }
}

Example 10 with SourceSessionParams

use of com.yahoo.messagebus.SourceSessionParams in project vespa by vespa-engine.

the class FeedHandler method sourceSessionParams.

/**
 * Exposed for use when creating mocks.
 */
protected SourceSessionParams sourceSessionParams(HttpRequest request) {
    SourceSessionParams params = new SourceSessionParams();
    String timeout = request.getHeader(Headers.TIMEOUT);
    if (timeout != null) {
        try {
            params.setTimeout(Double.parseDouble(timeout));
        } catch (NumberFormatException e) {
        // NOP
        }
    }
    return params;
}