tech.ydb.coordination.impl.ClientImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ydb-sdk-coordination Show documentation
Coordination node client implementation
The newest version!
package tech.ydb.coordination.impl;

import java.time.Clock;
import java.util.UUID;
import java.util.concurrent.CompletableFuture;

import tech.ydb.coordination.CoordinationClient;
import tech.ydb.coordination.CoordinationSession;
import tech.ydb.coordination.description.NodeConfig;
import tech.ydb.coordination.settings.CoordinationNodeSettings;
import tech.ydb.coordination.settings.CoordinationSessionSettings;
import tech.ydb.coordination.settings.DescribeCoordinationNodeSettings;
import tech.ydb.coordination.settings.DropCoordinationNodeSettings;
import tech.ydb.core.Result;
import tech.ydb.core.Status;
import tech.ydb.core.grpc.GrpcRequestSettings;
import tech.ydb.core.operation.Operation;
import tech.ydb.core.settings.BaseRequestSettings;
import tech.ydb.proto.coordination.AlterNodeRequest;
import tech.ydb.proto.coordination.CreateNodeRequest;
import tech.ydb.proto.coordination.DescribeNodeRequest;
import tech.ydb.proto.coordination.DropNodeRequest;

/**
 * @author Kirill Kurdyukov
 * @author Aleksandr Gorshenin
 */
class ClientImpl implements CoordinationClient {

    private final Rpc rpc;

    ClientImpl(Rpc rpc) {
        this.rpc = rpc;
    }

    private String validatePath(String path) {
        if (path == null || path.isEmpty()) {
            throw new IllegalArgumentException("Coordination node path cannot be empty");
        }

        return path.startsWith("/") ? path : rpc.getDatabase() + "/" + path;
    }

    private static String getTraceIdOrGenerateNew(String traceId) {
        return traceId == null ? UUID.randomUUID().toString() : traceId;
    }

    private GrpcRequestSettings makeGrpcRequestSettings(BaseRequestSettings settings, String traceId) {
        return GrpcRequestSettings.newBuilder()
                .withDeadline(settings.getRequestTimeout())
                .withTraceId(traceId)
                .build();
    }

    @Override
    public CoordinationSession createSession(String path, CoordinationSessionSettings settings) {
        return new SessionImpl(rpc, Clock.systemUTC(), validatePath(path), settings);
    }

    @Override
    public CompletableFuture createNode(String path, CoordinationNodeSettings settings) {
        CreateNodeRequest request = CreateNodeRequest.newBuilder()
                .setPath(validatePath(path))
                .setOperationParams(Operation.buildParams(settings))
                .setConfig(settings.getConfig().toProto())
                .build();

        String traceId = getTraceIdOrGenerateNew(settings.getTraceId());
        GrpcRequestSettings grpcSettings = makeGrpcRequestSettings(settings, traceId);
        return rpc.createNode(request, grpcSettings);
    }

    @Override
    public CompletableFuture alterNode(String path, CoordinationNodeSettings settings) {
        AlterNodeRequest request = AlterNodeRequest.newBuilder()
                .setPath(validatePath(path))
                .setOperationParams(Operation.buildParams(settings))
                .setConfig(settings.getConfig().toProto())
                .build();

        String traceId = getTraceIdOrGenerateNew(settings.getTraceId());
        GrpcRequestSettings grpcSettings = makeGrpcRequestSettings(settings, traceId);
        return rpc.alterNode(request, grpcSettings);
    }

    @Override
    public CompletableFuture dropNode(String path, DropCoordinationNodeSettings settings) {
        DropNodeRequest request = DropNodeRequest.newBuilder()
                .setPath(validatePath(path))
                .setOperationParams(Operation.buildParams(settings))
                .build();

        String traceId = getTraceIdOrGenerateNew(settings.getTraceId());
        GrpcRequestSettings grpcSettings = makeGrpcRequestSettings(settings, traceId);
        return rpc.dropNode(request, grpcSettings);
    }

    @Override
    public CompletableFuture> describeNode(String path,
            DescribeCoordinationNodeSettings settings) {
        DescribeNodeRequest request = DescribeNodeRequest.newBuilder()
                .setPath(validatePath(path))
                .setOperationParams(Operation.buildParams(settings))
                .build();

        String traceId = getTraceIdOrGenerateNew(settings.getTraceId());
        GrpcRequestSettings grpcSettings = makeGrpcRequestSettings(settings, traceId);
        return rpc.describeNode(request, grpcSettings);
    }

    @Override
    public String getDatabase() {
        return rpc.getDatabase();
    }
}