Skip to content

Commit ba8e9a1

Browse files
committed
Add update project
1 parent 1140e18 commit ba8e9a1

File tree

2 files changed

+22
-1
lines changed

2 files changed

+22
-1
lines changed

modules/ingest-geoip/src/main/java/org/elasticsearch/ingest/geoip/GeoIpDownloader.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ protected void updateTimestamp(String name, Metadata old) {
244244

245245
void updateTaskState() {
246246
PlainActionFuture<PersistentTask<?>> future = new PlainActionFuture<>();
247-
updatePersistentTaskState(state, future);
247+
updateProjectPersistentTaskState(projectId, state, future);
248248
state = ((GeoIpTaskState) future.actionGet().getState());
249249
}
250250

server/src/main/java/org/elasticsearch/persistent/AllocatedPersistentTask.java

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import org.apache.logging.log4j.Logger;
1313
import org.elasticsearch.action.ActionListener;
1414
import org.elasticsearch.action.admin.cluster.node.tasks.cancel.CancelTasksRequest;
15+
import org.elasticsearch.cluster.metadata.ProjectId;
1516
import org.elasticsearch.core.Nullable;
1617
import org.elasticsearch.core.TimeValue;
1718
import org.elasticsearch.tasks.CancellableTask;
@@ -74,6 +75,26 @@ public void updatePersistentTaskState(
7475
);
7576
}
7677

78+
/**
79+
* Updates the persistent state for the corresponding project scope persistent task.
80+
* <p>
81+
* This doesn't affect the status of this allocated task.
82+
*/
83+
public void updateProjectPersistentTaskState(
84+
final ProjectId projectId,
85+
final PersistentTaskState state,
86+
final ActionListener<PersistentTasksCustomMetadata.PersistentTask<?>> listener
87+
) {
88+
persistentTasksService.sendProjectUpdateStateRequest(
89+
projectId,
90+
persistentTaskId,
91+
allocationId,
92+
state,
93+
TimeValue.THIRTY_SECONDS /* TODO should this be longer? infinite? */,
94+
listener
95+
);
96+
}
97+
7798
public String getPersistentTaskId() {
7899
return persistentTaskId;
79100
}

0 commit comments

Comments
 (0)