Exemplo n.º 1
0
  public void callMethod(
      MethodDescriptor method,
      RpcController controller,
      Message request,
      Message responsePrototype,
      RpcCallback<Message> done) {
    ClientRpcController rpcController = (ClientRpcController) controller;

    int correlationId = getNextCorrelationId();
    rpcController.setCorrelationId(correlationId);

    PendingClientCallState state =
        new PendingClientCallState(rpcController, method, responsePrototype, request, done);
    registerPendingRequest(correlationId, state);

    RpcRequest rpcRequest =
        RpcRequest.newBuilder()
            .setCorrelationId(correlationId)
            .setServiceIdentifier(state.getServiceIdentifier())
            .setMethodIdentifier(state.getMethodIdentifier())
            .setRequestBytes(request.toByteString())
            .build();

    if (channel.isConnected()) {
      if (LOGGER.isDebugEnabled())
        //				LOGGER.debug("Sending ["+rpcRequest.getCorrelationId()+"]RpcRequest.");
        LOGGER.debug(
            "Sending [" + rpcRequest.getCorrelationId() + "]RpcRequest. signature {}",
            state.getMethodDesc().getFullName());
      WirePayload payload = WirePayload.newBuilder().setRpcRequest(rpcRequest).build();
      channel.write(payload);
    } else {
      LOGGER.error("Sending [" + rpcRequest.getCorrelationId() + "]RpcRequest. FAILURE");
      String errorMessage = "Session [" + channel + "]closed";
      RpcError rpcError =
          RpcError.newBuilder()
              .setCorrelationId(correlationId)
              .setErrorMessage(errorMessage)
              .build();
      error(rpcError);
      doLog(state, rpcError, errorMessage);
    }
  }
Exemplo n.º 2
0
  void startCancel(int correlationId) {
    PendingClientCallState state = removePendingRequest(correlationId);
    if (state != null) {
      RpcCancel rpcCancel = RpcCancel.newBuilder().setCorrelationId(correlationId).build();
      if (LOGGER.isDebugEnabled())
        LOGGER.debug("Sending [" + rpcCancel.getCorrelationId() + "]RpcCancel.");
      WirePayload payload = WirePayload.newBuilder().setRpcCancel(rpcCancel).build();
      channel.write(payload).awaitUninterruptibly();

      String errorMessage = "Cancel";

      doLog(state, rpcCancel, errorMessage);

      state.handleFailure(errorMessage);

    } else {
      // this can happen if the server call completed in the meantime.
      if (LOGGER.isDebugEnabled())
        LOGGER.debug("No PendingClientCallState found for correlationId " + correlationId);
    }
  }