diff options
author | Bjørn Christian Seime <bjorncs@verizonmedia.com> | 2020-06-11 18:35:09 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@verizonmedia.com> | 2020-06-15 13:18:01 +0200 |
commit | a7fc7785961dafa2276c97d207d6a5af322ba2ea (patch) | |
tree | 1f1951b02e5b604eec010d1abc65098896a4de1d /config-model/src/main/java/com | |
parent | 7d49d61de65cea5fd8980c25c5c2356ee1f6b852 (diff) |
Add threadpool configuration after cluster nodes are populated
Add unit test as well
Diffstat (limited to 'config-model/src/main/java/com')
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/clients/ContainerDocumentApi.java | 33 | ||||
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java | 8 |
2 files changed, 28 insertions, 13 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/clients/ContainerDocumentApi.java b/config-model/src/main/java/com/yahoo/vespa/model/clients/ContainerDocumentApi.java index 9ee62aca798..cdebf6a8c6c 100644 --- a/config-model/src/main/java/com/yahoo/vespa/model/clients/ContainerDocumentApi.java +++ b/config-model/src/main/java/com/yahoo/vespa/model/clients/ContainerDocumentApi.java @@ -22,28 +22,37 @@ public class ContainerDocumentApi { private static final int FALLBACK_MAX_POOL_SIZE = 0; // Use fallback based on actual logical core count on host private static final int FALLBACK_CORE_POOL_SIZE = 0; // Use fallback based on actual logical core count on host + private final ContainerCluster<?> cluster; + private final Options options; + private final Handler<AbstractConfigProducer<?>> feedHandler; + private final Handler<AbstractConfigProducer<?>> restApiHandler; + public ContainerDocumentApi(ContainerCluster<?> cluster, Options options) { - setupHandlers(cluster, options); + this.cluster = cluster; + this.options = options; + this.restApiHandler = addRestApiHandler(cluster, options); + this.feedHandler = addFeedHandler(cluster, options); } - private static void setupHandlers(ContainerCluster<?> cluster, Options options) { - addRestApiHandler(cluster, options); - addFeedHandler(cluster, options); + public void addNodesDependentThreadpoolConfiguration() { + if (cluster.getContainers().isEmpty()) throw new IllegalStateException("Cluster is empty"); + feedHandler.addComponent(newExecutorComponent("feedapi-handler", cluster, options)); + restApiHandler.addComponent(newExecutorComponent("restapi-handler", cluster, options)); } - private static void addFeedHandler(ContainerCluster<?> cluster, Options options) { - var executorComponent = newExecutorComponent("feedapi-handler", cluster, options); + private static Handler<AbstractConfigProducer<?>> addFeedHandler(ContainerCluster<?> cluster, Options options) { String bindingSuffix = ContainerCluster.RESERVED_URI_PREFIX + "/feedapi"; var handler = newVespaClientHandler( - "com.yahoo.vespa.http.server.FeedHandler", bindingSuffix, options, executorComponent); + "com.yahoo.vespa.http.server.FeedHandler", bindingSuffix, options); cluster.addComponent(handler); + return handler; } - private static void addRestApiHandler(ContainerCluster<?> cluster, Options options) { - var executorComponent = newExecutorComponent("restapi-handler", cluster, options); + private static Handler<AbstractConfigProducer<?>> addRestApiHandler(ContainerCluster<?> cluster, Options options) { var handler = newVespaClientHandler( - "com.yahoo.document.restapi.resource.RestApi", "document/v1/*", options, executorComponent); + "com.yahoo.document.restapi.resource.RestApi", "document/v1/*", options); cluster.addComponent(handler); + return handler; } private static ThreadPoolExecutorComponent newExecutorComponent(String name, ContainerCluster<?> cluster, Options options) { @@ -58,14 +67,12 @@ public class ContainerDocumentApi { private static Handler<AbstractConfigProducer<?>> newVespaClientHandler( String componentId, String bindingSuffix, - Options options, - ThreadPoolExecutorComponent executorComponent) { + Options options) { Handler<AbstractConfigProducer<?>> handler = new Handler<>(new ComponentModel( BundleInstantiationSpecification.getFromStrings(componentId, null, "vespaclient-container-plugin"), "")); for (String rootBinding : options.bindings) { handler.addServerBindings(rootBinding + bindingSuffix, rootBinding + bindingSuffix + '/'); } - handler.addComponent(executorComponent); return handler; } diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java b/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java index 2966e0e000d..4cf3108f928 100644 --- a/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java +++ b/config-model/src/main/java/com/yahoo/vespa/model/container/xml/ContainerModelBuilder.java @@ -207,6 +207,7 @@ public class ContainerModelBuilder extends ConfigModelBuilder<ContainerModel> { addClientProviders(deployState, spec, cluster); addServerProviders(deployState, spec, cluster); + addHandlerSpecificThreadpools(cluster); addAthensCopperArgos(cluster, context); // Must be added after nodes. } @@ -221,6 +222,13 @@ public class ContainerModelBuilder extends ConfigModelBuilder<ContainerModel> { } } + private void addHandlerSpecificThreadpools(ContainerCluster<?> cluster) { + ContainerDocumentApi documentApi = cluster.getDocumentApi(); + if (documentApi != null) { + documentApi.addNodesDependentThreadpoolConfiguration(); + } + } + private void addAthensCopperArgos(ApplicationContainerCluster cluster, ConfigModelContext context) { if ( ! context.getDeployState().isHosted()) return; app.getDeployment().map(DeploymentSpec::fromXml) |