Examples with Payload com.alibaba.nacos.api.grpc.auto.Payload used on opensource projects

Example 1 with Payload

use of com.alibaba.nacos.api.grpc.auto.Payload in project nacos by alibaba.

the class GrpcClient method connectToServer.

@Override
public Connection connectToServer(ServerInfo serverInfo) {
    try {
        if (grpcExecutor == null) {
            this.grpcExecutor = createGrpcExecutor(serverInfo.getServerIp());
        }
        int port = serverInfo.getServerPort() + rpcPortOffset();
        RequestGrpc.RequestFutureStub newChannelStubTemp = createNewChannelStub(serverInfo.getServerIp(), port);
        if (newChannelStubTemp != null) {
            Response response = serverCheck(serverInfo.getServerIp(), port, newChannelStubTemp);
            if (response == null || !(response instanceof ServerCheckResponse)) {
                shuntDownChannel((ManagedChannel) newChannelStubTemp.getChannel());
                return null;
            }
            BiRequestStreamGrpc.BiRequestStreamStub biRequestStreamStub = BiRequestStreamGrpc.newStub(newChannelStubTemp.getChannel());
            GrpcConnection grpcConn = new GrpcConnection(serverInfo, grpcExecutor);
            grpcConn.setConnectionId(((ServerCheckResponse) response).getConnectionId());
            // create stream request and bind connection event to this connection.
            StreamObserver<Payload> payloadStreamObserver = bindRequestStream(biRequestStreamStub, grpcConn);
            // stream observer to send response to server
            grpcConn.setPayloadStreamObserver(payloadStreamObserver);
            grpcConn.setGrpcFutureServiceStub(newChannelStubTemp);
            grpcConn.setChannel((ManagedChannel) newChannelStubTemp.getChannel());
            // send a  setup request.
            ConnectionSetupRequest conSetupRequest = new ConnectionSetupRequest();
            conSetupRequest.setClientVersion(VersionUtils.getFullClientVersion());
            conSetupRequest.setLabels(super.getLabels());
            conSetupRequest.setAbilities(super.clientAbilities);
            conSetupRequest.setTenant(super.getTenant());
            grpcConn.sendRequest(conSetupRequest);
            // wait to register connection setup
            Thread.sleep(100L);
            return grpcConn;
        }
        return null;
    } catch (Exception e) {
        LOGGER.error("[{}]Fail to connect to server!,error={}", GrpcClient.this.getName(), e);
    }
    return null;
}

Example 2 with Payload

use of com.alibaba.nacos.api.grpc.auto.Payload in project nacos by alibaba.

the class GrpcConnection method sendResponse.

public void sendResponse(Response response) {
    Payload convert = GrpcUtils.convert(response);
    payloadStreamObserver.onNext(convert);
}

Example 3 with Payload

use of com.alibaba.nacos.api.grpc.auto.Payload in project nacos by alibaba.

the class GrpcConnection method requestFuture.

@Override
public RequestFuture requestFuture(Request request) throws NacosException {
    Payload grpcRequest = GrpcUtils.convert(request);
    final ListenableFuture<Payload> requestFuture = grpcFutureServiceStub.request(grpcRequest);
    return new RequestFuture() {

        @Override
        public boolean isDone() {
            return requestFuture.isDone();
        }

        @Override
        public Response get() throws Exception {
            Payload grpcResponse = requestFuture.get();
            Response response = (Response) GrpcUtils.parse(grpcResponse);
            if (response instanceof ErrorResponse) {
                throw new NacosException(response.getErrorCode(), response.getMessage());
            }
            return response;
        }

        @Override
        public Response get(long timeout) throws Exception {
            Payload grpcResponse = requestFuture.get(timeout, TimeUnit.MILLISECONDS);
            Response response = (Response) GrpcUtils.parse(grpcResponse);
            if (response instanceof ErrorResponse) {
                throw new NacosException(response.getErrorCode(), response.getMessage());
            }
            return response;
        }
    };
}

Example 4 with Payload

use of com.alibaba.nacos.api.grpc.auto.Payload in project nacos by alibaba.

the class GrpcConnection method request.

@Override
public Response request(Request request, long timeouts) throws NacosException {
    Payload grpcRequest = GrpcUtils.convert(request);
    ListenableFuture<Payload> requestFuture = grpcFutureServiceStub.request(grpcRequest);
    Payload grpcResponse;
    try {
        grpcResponse = requestFuture.get(timeouts, TimeUnit.MILLISECONDS);
    } catch (Exception e) {
        throw new NacosException(NacosException.SERVER_ERROR, e);
    }
    return (Response) GrpcUtils.parse(grpcResponse);
}

Example 5 with Payload

use of com.alibaba.nacos.api.grpc.auto.Payload in project nacos by alibaba.

the class GrpcUtils method convert.

/**
 * convert request to payload.
 *
 * @param request request.
 * @param meta    request meta.
 * @return payload.
 */
public static Payload convert(Request request, RequestMeta meta) {
    // meta.
    Payload.Builder payloadBuilder = Payload.newBuilder();
    Metadata.Builder metaBuilder = Metadata.newBuilder();
    if (meta != null) {
        metaBuilder.putAllHeaders(request.getHeaders()).setType(request.getClass().getSimpleName());
    }
    metaBuilder.setClientIp(NetUtils.localIP());
    payloadBuilder.setMetadata(metaBuilder.build());
    // request body .
    request.clearHeaders();
    String jsonString = toJson(request);
    return payloadBuilder.setBody(Any.newBuilder().setValue(ByteString.copyFrom(jsonString, StandardCharsets.UTF_8))).build();
}