// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.clustercontroller.core.restapiv2.requests;

import com.yahoo.vdslib.state.ClusterState;
import com.yahoo.vdslib.state.Node;
import com.yahoo.vespa.clustercontroller.core.ContentCluster;
import com.yahoo.vespa.clustercontroller.core.listeners.NodeStateOrHostInfoChangeHandler;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.errors.StateRestApiException;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.requests.SetUnitStateRequest;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.response.SetResponse;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.response.UnitState;

import java.util.Map;

/**
 * @author hakon
 */
@FunctionalInterface
public interface WantedStateSetter {
    SetResponse set(ContentCluster cluster,
                    SetUnitStateRequest.Condition condition,
                    Map<String, UnitState> newStates,
                    Node node,
                    NodeStateOrHostInfoChangeHandler stateListener,
                    ClusterState currentClusterState,
                    boolean inMasterMoratorium,
                    boolean probe) throws StateRestApiException;
}