diff --git a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportDeleteConnectorAction.java b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportDeleteConnectorAction.java index e534d969fdaaa..db030e0b2cf47 100644 --- a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportDeleteConnectorAction.java +++ b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportDeleteConnectorAction.java @@ -12,12 +12,15 @@ import org.elasticsearch.action.support.HandledTransportAction; import org.elasticsearch.action.support.master.AcknowledgedResponse; import org.elasticsearch.client.internal.Client; +import org.elasticsearch.client.internal.OriginSettingClient; import org.elasticsearch.common.util.concurrent.EsExecutors; import org.elasticsearch.injection.guice.Inject; import org.elasticsearch.tasks.Task; import org.elasticsearch.transport.TransportService; import org.elasticsearch.xpack.application.connector.ConnectorIndexService; +import static org.elasticsearch.xpack.core.ClientHelper.ENT_SEARCH_ORIGIN; + public class TransportDeleteConnectorAction extends HandledTransportAction { protected final ConnectorIndexService connectorIndexService; @@ -31,7 +34,7 @@ public TransportDeleteConnectorAction(TransportService transportService, ActionF DeleteConnectorAction.Request::new, EsExecutors.DIRECT_EXECUTOR_SERVICE ); - this.connectorIndexService = new ConnectorIndexService(client); + this.connectorIndexService = new ConnectorIndexService(new OriginSettingClient(client, ENT_SEARCH_ORIGIN)); } @Override diff --git a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportGetConnectorAction.java b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportGetConnectorAction.java index e0beeb9b19515..d39e68622a5e1 100644 --- a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportGetConnectorAction.java +++ b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportGetConnectorAction.java @@ -11,12 +11,15 @@ import org.elasticsearch.action.support.ActionFilters; import org.elasticsearch.action.support.HandledTransportAction; import org.elasticsearch.client.internal.Client; +import org.elasticsearch.client.internal.OriginSettingClient; import org.elasticsearch.common.util.concurrent.EsExecutors; import org.elasticsearch.injection.guice.Inject; import org.elasticsearch.tasks.Task; import org.elasticsearch.transport.TransportService; import org.elasticsearch.xpack.application.connector.ConnectorIndexService; +import static org.elasticsearch.xpack.core.ClientHelper.ENT_SEARCH_ORIGIN; + public class TransportGetConnectorAction extends HandledTransportAction { protected final ConnectorIndexService connectorIndexService; @@ -29,7 +32,7 @@ public TransportGetConnectorAction(TransportService transportService, ActionFilt GetConnectorAction.Request::new, EsExecutors.DIRECT_EXECUTOR_SERVICE ); - this.connectorIndexService = new ConnectorIndexService(client); + this.connectorIndexService = new ConnectorIndexService(new OriginSettingClient(client, ENT_SEARCH_ORIGIN)); } @Override diff --git a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPostConnectorAction.java b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPostConnectorAction.java index 4762f125fe9df..4adcdeab08d8d 100644 --- a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPostConnectorAction.java +++ b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPostConnectorAction.java @@ -11,12 +11,15 @@ import org.elasticsearch.action.support.ActionFilters; import org.elasticsearch.action.support.HandledTransportAction; import org.elasticsearch.client.internal.Client; +import org.elasticsearch.client.internal.OriginSettingClient; import org.elasticsearch.common.util.concurrent.EsExecutors; import org.elasticsearch.injection.guice.Inject; import org.elasticsearch.tasks.Task; import org.elasticsearch.transport.TransportService; import org.elasticsearch.xpack.application.connector.ConnectorIndexService; +import static org.elasticsearch.xpack.core.ClientHelper.ENT_SEARCH_ORIGIN; + public class TransportPostConnectorAction extends HandledTransportAction { protected final ConnectorIndexService connectorIndexService; @@ -30,7 +33,7 @@ public TransportPostConnectorAction(TransportService transportService, ActionFil PostConnectorAction.Request::new, EsExecutors.DIRECT_EXECUTOR_SERVICE ); - this.connectorIndexService = new ConnectorIndexService(client); + this.connectorIndexService = new ConnectorIndexService(new OriginSettingClient(client, ENT_SEARCH_ORIGIN)); } @Override diff --git a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPutConnectorAction.java b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPutConnectorAction.java index b409670e44164..3719aed07fb97 100644 --- a/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPutConnectorAction.java +++ b/x-pack/plugin/ent-search/src/main/java/org/elasticsearch/xpack/application/connector/action/TransportPutConnectorAction.java @@ -11,12 +11,15 @@ import org.elasticsearch.action.support.ActionFilters; import org.elasticsearch.action.support.HandledTransportAction; import org.elasticsearch.client.internal.Client; +import org.elasticsearch.client.internal.OriginSettingClient; import org.elasticsearch.common.util.concurrent.EsExecutors; import org.elasticsearch.injection.guice.Inject; import org.elasticsearch.tasks.Task; import org.elasticsearch.transport.TransportService; import org.elasticsearch.xpack.application.connector.ConnectorIndexService; +import static org.elasticsearch.xpack.core.ClientHelper.ENT_SEARCH_ORIGIN; + public class TransportPutConnectorAction extends HandledTransportAction { protected final ConnectorIndexService connectorIndexService; @@ -30,7 +33,7 @@ public TransportPutConnectorAction(TransportService transportService, ActionFilt PutConnectorAction.Request::new, EsExecutors.DIRECT_EXECUTOR_SERVICE ); - this.connectorIndexService = new ConnectorIndexService(client); + this.connectorIndexService = new ConnectorIndexService(new OriginSettingClient(client, ENT_SEARCH_ORIGIN)); } @Override