tech.ydb.core.grpc.BalancingSettings Maven / Gradle / Ivy

Go to download
package tech.ydb.core.grpc;

/**
 * @author Nikolay Perfilov
 */
public class BalancingSettings {
    public static final BalancingPolicy DEFAULT_BALANCING_POLICY = BalancingPolicy.USE_ALL_NODES;

    private final BalancingPolicy policy;
    private final String preferableLocation;

    public BalancingSettings(BalancingPolicy policy, String preferableLocation) {
        this.policy = policy;
        this.preferableLocation = preferableLocation;
    }

    public BalancingSettings() {
        this.policy = DEFAULT_BALANCING_POLICY;
        this.preferableLocation = null;
    }

    public BalancingPolicy getPolicy() {
        return policy;
    }

    public String getPreferableLocation() {
        return preferableLocation;
    }

    public static BalancingSettings fromPolicy(BalancingPolicy balancingPolicy) {
        return new BalancingSettings(balancingPolicy, null);
    }

    public static BalancingSettings fromLocation(String preferableLocation) {
        return new BalancingSettings(BalancingPolicy.USE_PREFERABLE_LOCATION, preferableLocation);
    }

    @Override
    public String toString() {
        return "BalancingSettings{" +
                "policy=" + policy +
                ", preferableLocation='" + preferableLocation + '\'' +
                '}';
    }
}