|
@@ -19,6 +19,7 @@ import org.elasticsearch.action.admin.cluster.node.tasks.list.ListTasksResponse;
|
|
|
import org.elasticsearch.action.admin.cluster.snapshots.features.ResetFeatureStateResponse;
|
|
|
import org.elasticsearch.action.support.master.AcknowledgedResponse;
|
|
|
import org.elasticsearch.client.Client;
|
|
|
+import org.elasticsearch.client.OriginSettingClient;
|
|
|
import org.elasticsearch.cluster.metadata.IndexNameExpressionResolver;
|
|
|
import org.elasticsearch.cluster.metadata.IndexTemplateMetadata;
|
|
|
import org.elasticsearch.cluster.metadata.Metadata;
|
|
@@ -111,6 +112,7 @@ import java.util.Map;
|
|
|
import java.util.function.Supplier;
|
|
|
import java.util.function.UnaryOperator;
|
|
|
|
|
|
+import static org.elasticsearch.xpack.core.ClientHelper.TRANSFORM_ORIGIN;
|
|
|
import static org.elasticsearch.xpack.core.transform.TransformMessages.FAILED_TO_UNSET_RESET_MODE;
|
|
|
import static org.elasticsearch.xpack.core.transform.transforms.persistence.TransformInternalIndexConstants.AUDIT_INDEX_PATTERN;
|
|
|
import static org.elasticsearch.xpack.core.transform.transforms.persistence.TransformInternalIndexConstants.TRANSFORM_PREFIX;
|
|
@@ -308,10 +310,10 @@ public class Transform extends Plugin implements SystemIndexPlugin, PersistentTa
|
|
|
@Override
|
|
|
public void cleanUpFeature(
|
|
|
ClusterService clusterService,
|
|
|
- Client client,
|
|
|
+ Client unwrappedClient,
|
|
|
ActionListener<ResetFeatureStateResponse.ResetFeatureStateStatus> finalListener
|
|
|
) {
|
|
|
-
|
|
|
+ OriginSettingClient client = new OriginSettingClient(unwrappedClient, TRANSFORM_ORIGIN);
|
|
|
ActionListener<ResetFeatureStateResponse.ResetFeatureStateStatus> unsetResetModeListener = ActionListener.wrap(
|
|
|
success -> client.execute(
|
|
|
SetResetModeAction.INSTANCE,
|