tech.ydb.coordination.impl.RpcImpl 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
There is a newer version: 2.3.6
package tech.ydb.coordination.impl;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ScheduledExecutorService;
import java.util.function.Function;

import tech.ydb.coordination.description.NodeConfig;
import tech.ydb.core.Result;
import tech.ydb.core.Status;
import tech.ydb.core.grpc.GrpcReadWriteStream;
import tech.ydb.core.grpc.GrpcRequestSettings;
import tech.ydb.core.grpc.GrpcTransport;
import tech.ydb.core.operation.OperationBinder;
import tech.ydb.proto.coordination.AlterNodeRequest;
import tech.ydb.proto.coordination.AlterNodeResponse;
import tech.ydb.proto.coordination.CreateNodeRequest;
import tech.ydb.proto.coordination.CreateNodeResponse;
import tech.ydb.proto.coordination.DescribeNodeRequest;
import tech.ydb.proto.coordination.DescribeNodeResponse;
import tech.ydb.proto.coordination.DescribeNodeResult;
import tech.ydb.proto.coordination.DropNodeRequest;
import tech.ydb.proto.coordination.DropNodeResponse;
import tech.ydb.proto.coordination.SessionRequest;
import tech.ydb.proto.coordination.SessionResponse;
import tech.ydb.proto.coordination.v1.CoordinationServiceGrpc;

/**
 * @author Kirill Kurdyukov
 */
class RpcImpl implements Rpc {
    private static final Function, Status> CREATE_NODE_STATUS = OperationBinder
            .bindSync(CreateNodeResponse::getOperation);

    private static final Function, Status> ALTER_NODE_STATUS = OperationBinder
            .bindSync(AlterNodeResponse::getOperation);

    private static final Function, Status> DROP_NODE_STATUS = OperationBinder
            .bindSync(DropNodeResponse::getOperation);

    private static final Function, Result> DESCRIBE_NODE_RESULT =
            OperationBinder.bindSync(DescribeNodeResponse::getOperation, DescribeNodeResult.class);

    private final GrpcTransport transport;

    RpcImpl(GrpcTransport grpcTransport) {
        this.transport = grpcTransport;
    }

    @Override
    public GrpcReadWriteStream createSession(GrpcRequestSettings settings) {
        return transport.readWriteStreamCall(CoordinationServiceGrpc.getSessionMethod(), settings);
    }

    @Override
    public CompletableFuture createNode(CreateNodeRequest request, GrpcRequestSettings settings) {
        return transport
                .unaryCall(CoordinationServiceGrpc.getCreateNodeMethod(), settings, request)
                .thenApply(CREATE_NODE_STATUS);
    }

    @Override
    public CompletableFuture alterNode(AlterNodeRequest request, GrpcRequestSettings settings) {
        return transport
                .unaryCall(CoordinationServiceGrpc.getAlterNodeMethod(), settings, request)
                .thenApply(ALTER_NODE_STATUS);
    }

    @Override
    public CompletableFuture dropNode(DropNodeRequest request, GrpcRequestSettings settings) {
        return transport
                .unaryCall(CoordinationServiceGrpc.getDropNodeMethod(), settings, request)
                .thenApply(DROP_NODE_STATUS);
    }

    @Override
    public CompletableFuture> describeNode(DescribeNodeRequest req, GrpcRequestSettings settings) {
        return transport
                .unaryCall(CoordinationServiceGrpc.getDescribeNodeMethod(), settings, req)
                .thenApply(DESCRIBE_NODE_RESULT)
                .thenApply(r -> r.map(NodeConfig::fromProto));
    }

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

    @Override
    public ScheduledExecutorService getScheduler() {
        return transport.getScheduler();
    }
}