Skip to content

Commit e03c4f2

Browse files
droberts195droberts195
committed
Move persistent task registrations to core
Persistent tasks was moved from X-Pack to core in elastic#28455. However, registration of the named writables and named X-content was left in X-Pack. This change moves the registration of the named writables and named X-content into core. Additionally, the persistent task actions are no longer registered in the X-Pack client plugin, as they are already registered in ActionModule.
1 parent e639036 commit e03c4f2

File tree

4 files changed

+17
-32
lines changed

4 files changed

+17
-32
lines changed

‎server/src/main/java/org/elasticsearch/common/network/NetworkModule.java

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@
2020
package org.elasticsearch.common.network;
2121

2222
import org.elasticsearch.action.support.replication.ReplicationTask;
23+
import org.elasticsearch.cluster.NamedDiff;
24+
import org.elasticsearch.cluster.metadata.MetaData;
2325
import org.elasticsearch.cluster.routing.allocation.command.AllocateEmptyPrimaryAllocationCommand;
2426
import org.elasticsearch.cluster.routing.allocation.command.AllocateReplicaAllocationCommand;
2527
import org.elasticsearch.cluster.routing.allocation.command.AllocateStalePrimaryAllocationCommand;
@@ -39,6 +41,8 @@
3941
import org.elasticsearch.common.xcontent.XContentParser;
4042
import org.elasticsearch.http.HttpServerTransport;
4143
import org.elasticsearch.indices.breaker.CircuitBreakerService;
44+
import org.elasticsearch.persistent.PersistentTasksCustomMetaData;
45+
import org.elasticsearch.persistent.PersistentTasksNodeService;
4246
import org.elasticsearch.plugins.NetworkPlugin;
4347
import org.elasticsearch.tasks.RawTaskStatus;
4448
import org.elasticsearch.tasks.Task;
@@ -94,6 +98,18 @@ public final class NetworkModule {
9498
new NamedWriteableRegistry.Entry(Task.Status.class, ReplicationTask.Status.NAME, ReplicationTask.Status::new));
9599
namedWriteables.add(
96100
new NamedWriteableRegistry.Entry(Task.Status.class, RawTaskStatus.NAME, RawTaskStatus::new));
101+
namedWriteables.add(
102+
new NamedWriteableRegistry.Entry(Task.Status.class, PersistentTasksNodeService.Status.NAME,
103+
PersistentTasksNodeService.Status::new));
104+
namedWriteables.add(
105+
new NamedWriteableRegistry.Entry(MetaData.Custom.class, PersistentTasksCustomMetaData.TYPE,
106+
PersistentTasksCustomMetaData::new));
107+
namedWriteables.add(
108+
new NamedWriteableRegistry.Entry(NamedDiff.class, PersistentTasksCustomMetaData.TYPE,
109+
PersistentTasksCustomMetaData::readDiffFrom));
110+
namedXContents.add(
111+
new NamedXContentRegistry.Entry(MetaData.Custom.class, new ParseField(PersistentTasksCustomMetaData.TYPE),
112+
PersistentTasksCustomMetaData::fromXContent));
97113
}
98114

99115
private final Map<String, Supplier<Transport>> transportFactories = new HashMap<>();

‎server/src/main/java/org/elasticsearch/persistent/PersistentTasksCustomMetaData.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
import org.elasticsearch.cluster.ClusterState;
2626
import org.elasticsearch.cluster.NamedDiff;
2727
import org.elasticsearch.cluster.metadata.MetaData;
28-
import org.elasticsearch.cluster.node.DiscoveryNodes;
2928
import org.elasticsearch.common.Nullable;
3029
import org.elasticsearch.common.ParseField;
3130
import org.elasticsearch.common.Strings;

‎server/src/test/java/org/elasticsearch/persistent/TestPersistentTasksPlugin.java

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,7 @@
3333
import org.elasticsearch.client.Client;
3434
import org.elasticsearch.client.ElasticsearchClient;
3535
import org.elasticsearch.cluster.ClusterState;
36-
import org.elasticsearch.cluster.NamedDiff;
3736
import org.elasticsearch.cluster.metadata.IndexNameExpressionResolver;
38-
import org.elasticsearch.cluster.metadata.MetaData;
3937
import org.elasticsearch.cluster.node.DiscoveryNode;
4038
import org.elasticsearch.cluster.service.ClusterService;
4139
import org.elasticsearch.common.ParseField;
@@ -100,21 +98,13 @@ public List<PersistentTasksExecutor<?>> getPersistentTasksExecutor(ClusterServic
10098
public List<NamedWriteableRegistry.Entry> getNamedWriteables() {
10199
return Arrays.asList(
102100
new NamedWriteableRegistry.Entry(PersistentTaskParams.class, TestPersistentTasksExecutor.NAME, TestParams::new),
103-
new NamedWriteableRegistry.Entry(Task.Status.class,
104-
PersistentTasksNodeService.Status.NAME, PersistentTasksNodeService.Status::new),
105-
new NamedWriteableRegistry.Entry(MetaData.Custom.class, PersistentTasksCustomMetaData.TYPE,
106-
PersistentTasksCustomMetaData::new),
107-
new NamedWriteableRegistry.Entry(NamedDiff.class, PersistentTasksCustomMetaData.TYPE,
108-
PersistentTasksCustomMetaData::readDiffFrom),
109101
new NamedWriteableRegistry.Entry(Task.Status.class, TestPersistentTasksExecutor.NAME, Status::new)
110102
);
111103
}
112104

113105
@Override
114106
public List<NamedXContentRegistry.Entry> getNamedXContent() {
115107
return Arrays.asList(
116-
new NamedXContentRegistry.Entry(MetaData.Custom.class, new ParseField(PersistentTasksCustomMetaData.TYPE),
117-
PersistentTasksCustomMetaData::fromXContent),
118108
new NamedXContentRegistry.Entry(PersistentTaskParams.class, new ParseField(TestPersistentTasksExecutor.NAME),
119109
TestParams::fromXContent),
120110
new NamedXContentRegistry.Entry(Task.Status.class, new ParseField(TestPersistentTasksExecutor.NAME), Status::fromXContent)

‎x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/XPackClientPlugin.java

Lines changed: 1 addition & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -90,13 +90,7 @@
9090
import org.elasticsearch.xpack.core.ml.datafeed.DatafeedState;
9191
import org.elasticsearch.xpack.core.ml.job.config.JobTaskStatus;
9292
import org.elasticsearch.xpack.core.monitoring.MonitoringFeatureSetUsage;
93-
import org.elasticsearch.persistent.CompletionPersistentTaskAction;
9493
import org.elasticsearch.persistent.PersistentTaskParams;
95-
import org.elasticsearch.persistent.PersistentTasksCustomMetaData;
96-
import org.elasticsearch.persistent.PersistentTasksNodeService;
97-
import org.elasticsearch.persistent.RemovePersistentTaskAction;
98-
import org.elasticsearch.persistent.StartPersistentTaskAction;
99-
import org.elasticsearch.persistent.UpdatePersistentTaskStatusAction;
10094
import org.elasticsearch.xpack.core.rollup.RollupFeatureSetUsage;
10195
import org.elasticsearch.xpack.core.rollup.RollupField;
10296
import org.elasticsearch.xpack.core.rollup.action.DeleteRollupJobAction;
@@ -253,11 +247,6 @@ public List<GenericAction> getClientActions() {
253247
GetCalendarEventsAction.INSTANCE,
254248
PostCalendarEventsAction.INSTANCE,
255249
PersistJobAction.INSTANCE,
256-
// licensing
257-
StartPersistentTaskAction.INSTANCE,
258-
UpdatePersistentTaskStatusAction.INSTANCE,
259-
RemovePersistentTaskAction.INSTANCE,
260-
CompletionPersistentTaskAction.INSTANCE,
261250
// security
262251
ClearRealmCacheAction.INSTANCE,
263252
ClearRolesCacheAction.INSTANCE,
@@ -322,18 +311,12 @@ public List<NamedWriteableRegistry.Entry> getNamedWriteables() {
322311
// ML - Custom metadata
323312
new NamedWriteableRegistry.Entry(MetaData.Custom.class, "ml", MlMetadata::new),
324313
new NamedWriteableRegistry.Entry(NamedDiff.class, "ml", MlMetadata.MlMetadataDiff::new),
325-
new NamedWriteableRegistry.Entry(MetaData.Custom.class, PersistentTasksCustomMetaData.TYPE,
326-
PersistentTasksCustomMetaData::new),
327-
new NamedWriteableRegistry.Entry(NamedDiff.class, PersistentTasksCustomMetaData.TYPE,
328-
PersistentTasksCustomMetaData::readDiffFrom),
329314
// ML - Persistent action requests
330315
new NamedWriteableRegistry.Entry(PersistentTaskParams.class, StartDatafeedAction.TASK_NAME,
331316
StartDatafeedAction.DatafeedParams::new),
332317
new NamedWriteableRegistry.Entry(PersistentTaskParams.class, OpenJobAction.TASK_NAME,
333318
OpenJobAction.JobParams::new),
334319
// ML - Task statuses
335-
new NamedWriteableRegistry.Entry(Task.Status.class, PersistentTasksNodeService.Status.NAME,
336-
PersistentTasksNodeService.Status::new),
337320
new NamedWriteableRegistry.Entry(Task.Status.class, JobTaskStatus.NAME, JobTaskStatus::new),
338321
new NamedWriteableRegistry.Entry(Task.Status.class, DatafeedState.NAME, DatafeedState::fromStream),
339322
new NamedWriteableRegistry.Entry(XPackFeatureSet.Usage.class, XPackField.MACHINE_LEARNING,
@@ -368,8 +351,6 @@ public List<NamedXContentRegistry.Entry> getNamedXContent() {
368351
// ML - Custom metadata
369352
new NamedXContentRegistry.Entry(MetaData.Custom.class, new ParseField("ml"),
370353
parser -> MlMetadata.METADATA_PARSER.parse(parser, null).build()),
371-
new NamedXContentRegistry.Entry(MetaData.Custom.class, new ParseField(PersistentTasksCustomMetaData.TYPE),
372-
PersistentTasksCustomMetaData::fromXContent),
373354
// ML - Persistent action requests
374355
new NamedXContentRegistry.Entry(PersistentTaskParams.class, new ParseField(StartDatafeedAction.TASK_NAME),
375356
StartDatafeedAction.DatafeedParams::fromXContent),
@@ -385,8 +366,7 @@ public List<NamedXContentRegistry.Entry> getNamedXContent() {
385366
new NamedXContentRegistry.Entry(MetaData.Custom.class, new ParseField(LicensesMetaData.TYPE),
386367
LicensesMetaData::fromXContent),
387368
//rollup
388-
new NamedXContentRegistry.Entry(PersistentTaskParams.class, new ParseField(RollupField.TASK_NAME),
389-
parser -> RollupJob.fromXContent(parser)),
369+
new NamedXContentRegistry.Entry(PersistentTaskParams.class, new ParseField(RollupField.TASK_NAME), RollupJob::fromXContent),
390370
new NamedXContentRegistry.Entry(Task.Status.class, new ParseField(RollupJobStatus.NAME), RollupJobStatus::fromXContent)
391371
);
392372
}

0 commit comments

Comments
 (0)