org.elasticsearch.cluster.AckedClusterStateUpdateTask Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch Show documentation
Show all versions of elasticsearch Show documentation
Elasticsearch subproject :server
/*
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
* or more contributor license agreements. Licensed under the Elastic License
* 2.0 and the Server Side Public License, v 1; you may not use this file except
* in compliance with, at your election, the Elastic License 2.0 or the Server
* Side Public License, v 1.
*/
package org.elasticsearch.cluster;
import org.elasticsearch.action.ActionListener;
import org.elasticsearch.action.support.master.AcknowledgedResponse;
import org.elasticsearch.cluster.ack.AckedRequest;
import org.elasticsearch.cluster.node.DiscoveryNode;
import org.elasticsearch.common.Priority;
import org.elasticsearch.core.TimeValue;
/**
* An extension interface to {@link ClusterStateUpdateTask} that allows to be notified when
* all the nodes have acknowledged a cluster state update request
*/
public abstract class AckedClusterStateUpdateTask extends ClusterStateUpdateTask implements ClusterStateAckListener {
private final ActionListener listener;
private final AckedRequest request;
protected AckedClusterStateUpdateTask(AckedRequest request, ActionListener listener) {
this(Priority.NORMAL, request, listener);
}
@SuppressWarnings("unchecked")
protected AckedClusterStateUpdateTask(
Priority priority,
AckedRequest request,
ActionListener listener
) {
super(priority, request.masterNodeTimeout());
this.listener = (ActionListener) listener;
this.request = request;
}
/**
* Called to determine which nodes the acknowledgement is expected from
*
* @param discoveryNode a node
* @return true if the node is expected to send ack back, false otherwise
*/
public boolean mustAck(DiscoveryNode discoveryNode) {
return true;
}
@Override
public void onAllNodesAcked() {
listener.onResponse(newResponse(true));
}
@Override
public void onAckFailure(Exception e) {
listener.onResponse(newResponse(false));
}
protected AcknowledgedResponse newResponse(boolean acknowledged) {
return AcknowledgedResponse.of(acknowledged);
}
/**
* Called once the acknowledgement timeout defined by
* {@link AckedClusterStateUpdateTask#ackTimeout()} has expired
*/
public void onAckTimeout() {
listener.onResponse(newResponse(false));
}
@Override
public void onFailure(Exception e) {
listener.onFailure(e);
}
/**
* Acknowledgement timeout, maximum time interval to wait for acknowledgements
*/
public final TimeValue ackTimeout() {
return request.ackTimeout();
}
}