Examples with RaftClientRequestProto org.apache.ratis.shaded.proto.RaftProtos.RaftClientRequestProto used on opensource projects

Example 1 with RaftClientRequestProto

use of org.apache.ratis.shaded.proto.RaftProtos.RaftClientRequestProto in project incubator-ratis by apache.

the class AppendStreamer method refreshLeader.

private void refreshLeader(RaftPeerId suggestedLeader, RaftPeerId oldLeader) {
    running = RunningState.LOOK_FOR_LEADER;
    refreshLeaderProxy(suggestedLeader, oldLeader);
    reQueuePendingRequests(leaderId);
    final RaftClientRequestProto request = Objects.requireNonNull(dataQueue.poll());
    ackQueue.offer(request);
    try {
        exceptionAndRetry.retryInterval.sleep();
    } catch (InterruptedException ignored) {
    }
    leaderProxy.onNext(request);
}

Example 2 with RaftClientRequestProto

use of org.apache.ratis.shaded.proto.RaftProtos.RaftClientRequestProto in project incubator-ratis by apache.

the class AppendStreamer method write.

synchronized void write(ByteString content, long seqNum) throws IOException {
    checkState();
    while (isRunning() && dataQueue.size() >= maxPendingNum) {
        try {
            wait();
        } catch (InterruptedException ignored) {
        }
    }
    if (isRunning()) {
        // wrap the current buffer into a RaftClientRequestProto
        final RaftClientRequestProto request = ClientProtoUtils.toRaftClientRequestProto(clientId, leaderId, groupId, seqNum, seqNum, content);
        if (request.getSerializedSize() > maxMessageSize.getSizeInt()) {
            throw new IOException("msg size:" + request.getSerializedSize() + " exceeds maximum:" + maxMessageSize.getSizeInt());
        }
        dataQueue.offer(request);
        this.notifyAll();
    } else {
        throwException(this + " got closed.");
    }
}

Example 3 with RaftClientRequestProto

use of org.apache.ratis.shaded.proto.RaftProtos.RaftClientRequestProto in project incubator-ratis by apache.

the class AppendStreamer method reQueuePendingRequests.

private void reQueuePendingRequests(RaftPeerId newLeader) {
    if (isRunning()) {
        // resend all the pending requests
        while (!ackQueue.isEmpty()) {
            final RaftClientRequestProto oldRequest = ackQueue.pollLast();
            final RaftRpcRequestProto.Builder newRpc = RaftRpcRequestProto.newBuilder(oldRequest.getRpcRequest()).setReplyId(newLeader.toByteString());
            final RaftClientRequestProto newRequest = RaftClientRequestProto.newBuilder(oldRequest).setRpcRequest(newRpc).build();
            dataQueue.offerFirst(newRequest);
        }
    }
}

Example 4 with RaftClientRequestProto

use of org.apache.ratis.shaded.proto.RaftProtos.RaftClientRequestProto in project incubator-ratis by apache.

the class GrpcClientRpc method sendRequest.

private CompletableFuture<RaftClientReply> sendRequest(RaftClientRequest request, RaftClientProtocolClient proxy) throws IOException {
    final RaftClientRequestProto requestProto = toRaftClientRequestProto(request);
    final CompletableFuture<RaftClientReplyProto> replyFuture = new CompletableFuture<>();
    // create a new grpc stream for each non-async call.
    final StreamObserver<RaftClientRequestProto> requestObserver = proxy.appendWithTimeout(new StreamObserver<RaftClientReplyProto>() {

        @Override
        public void onNext(RaftClientReplyProto value) {
            replyFuture.complete(value);
        }

        @Override
        public void onError(Throwable t) {
            replyFuture.completeExceptionally(RaftGrpcUtil.unwrapIOException(t));
        }

        @Override
        public void onCompleted() {
            if (!replyFuture.isDone()) {
                replyFuture.completeExceptionally(new IOException(clientId + ": Stream completed but no reply for request " + request));
            }
        }
    });
    requestObserver.onNext(requestProto);
    requestObserver.onCompleted();
    return replyFuture.thenApply(ClientProtoUtils::toRaftClientReply);
}