Examples with DocumentProcessingHandler com.yahoo.docproc.jdisc.DocumentProcessingHandler used on opensource projects

Example 1 with DocumentProcessingHandler

use of com.yahoo.docproc.jdisc.DocumentProcessingHandler in project vespa by vespa-engine.

the class VespaFeedHandlerTestCase method setup.

public void setup(com.yahoo.messagebus.Error e, LoadTypeConfig loadTypeCfg, boolean autoReply, DummySessionFactory.ReplyFactory autoReplyFactory) throws Exception {
    DocumentTypeManager docMan = new DocumentTypeManager();
    DocumentTypeManagerConfigurer.configure(docMan, "file:" + xmlFilesPath + "documentmanager.cfg");
    if (autoReply) {
        if (autoReplyFactory != null) {
            factory = DummySessionFactory.createWithAutoReplyFactory(autoReplyFactory);
        } else {
            factory = DummySessionFactory.createWithErrorAutoReply(e);
        }
    } else {
        factory = DummySessionFactory.createDefault();
    }
    context = new FeedContext(new MessagePropertyProcessor(new FeederConfig(new FeederConfig.Builder()), loadTypeCfg), factory, docMan, new ClusterList(), new NullFeedMetric());
    Executor threadPool = Executors.newCachedThreadPool();
    feedHandler = new VespaFeedHandler(context, threadPool);
    removeHandler = new VespaFeedHandlerRemove(context, threadPool);
    statusHandler = new VespaFeedHandlerStatus(context, false, false, threadPool);
    removeLocationHandler = new VespaFeedHandlerRemoveLocation(context, threadPool);
    CallStack dpCallstack = new CallStack("bar");
    dpCallstack.addLast(new TestDocProc());
    dpCallstack.addLast(new TestLaterDocProc());
    DocprocService myservice = new DocprocService("bar");
    myservice.setCallStack(dpCallstack);
    myservice.setInService(true);
    ComponentRegistry<DocprocService> registry = new ComponentRegistry<DocprocService>();
    registry.register(new ComponentId(myservice.getName()), myservice);
    DocumentProcessingHandler handler = new DocumentProcessingHandler(registry, new ComponentRegistry<>(), new ComponentRegistry<>(), new DocumentProcessingHandlerParameters());
    Container container = Container.get();
    ComponentRegistry<RequestHandler> requestHandlerComponentRegistry = new ComponentRegistry<>();
    requestHandlerComponentRegistry.register(new ComponentId(DocumentProcessingHandler.class.getName()), handler);
    container.setRequestHandlerRegistry(requestHandlerComponentRegistry);
}

Example 2 with DocumentProcessingHandler

use of com.yahoo.docproc.jdisc.DocumentProcessingHandler in project vespa by vespa-engine.

the class MessagePropertyProcessor method getDocprocServiceRegistry.

public synchronized ComponentRegistry<DocprocService> getDocprocServiceRegistry(HttpRequest request) {
    String docprocChain = getDocprocChainParameter(request);
    if (docprocChain == null) {
        return null;
    }
    Container container = Container.get();
    if (container == null) {
        throw new IllegalStateException("Could not get Container instance.");
    }
    ComponentRegistry<RequestHandler> requestHandlerRegistry = container.getRequestHandlerRegistry();
    if (requestHandlerRegistry == null) {
        throw new IllegalStateException("Could not get requesthandlerregistry.");
    }
    DocumentProcessingHandler handler = (DocumentProcessingHandler) requestHandlerRegistry.getComponent(DocumentProcessingHandler.class.getName());
    if (handler == null) {
        return null;
    }
    ComponentRegistry<DocprocService> services = handler.getDocprocServiceRegistry();
    if (services == null) {
        throw new IllegalStateException("Could not get DocprocServiceRegistry.");
    }
    return services;
}