Examples with RpcMessage io.seata.core.protocol.RpcMessage used on opensource projects

Example 11 with RpcMessage

use of io.seata.core.protocol.RpcMessage in project seata by seata.

the class AbstractNettyRemoting method buildResponseMessage.

protected RpcMessage buildResponseMessage(RpcMessage rpcMessage, Object msg, byte messageType) {
    RpcMessage rpcMsg = new RpcMessage();
    rpcMsg.setMessageType(messageType);
    // same with request
    rpcMsg.setCodec(rpcMessage.getCodec());
    rpcMsg.setCompressor(rpcMessage.getCompressor());
    rpcMsg.setBody(msg);
    rpcMsg.setId(rpcMessage.getId());
    return rpcMsg;
}

Example 12 with RpcMessage

use of io.seata.core.protocol.RpcMessage in project seata by seata.

the class AbstractNettyRemoting method sendSync.

/**
 * rpc sync request
 * Obtain the return result through MessageFuture blocking.
 *
 * @param channel       netty channel
 * @param rpcMessage    rpc message
 * @param timeoutMillis rpc communication timeout
 * @return response message
 * @throws TimeoutException
 */
protected Object sendSync(Channel channel, RpcMessage rpcMessage, long timeoutMillis) throws TimeoutException {
    if (timeoutMillis <= 0) {
        throw new FrameworkException("timeout should more than 0ms");
    }
    if (channel == null) {
        LOGGER.warn("sendSync nothing, caused by null channel.");
        return null;
    }
    MessageFuture messageFuture = new MessageFuture();
    messageFuture.setRequestMessage(rpcMessage);
    messageFuture.setTimeout(timeoutMillis);
    futures.put(rpcMessage.getId(), messageFuture);
    channelWritableCheck(channel, rpcMessage.getBody());
    String remoteAddr = ChannelUtil.getAddressFromChannel(channel);
    doBeforeRpcHooks(remoteAddr, rpcMessage);
    channel.writeAndFlush(rpcMessage).addListener((ChannelFutureListener) future -> {
        if (!future.isSuccess()) {
            MessageFuture messageFuture1 = futures.remove(rpcMessage.getId());
            if (messageFuture1 != null) {
                messageFuture1.setResultMessage(future.cause());
            }
            destroyChannel(future.channel());
        }
    });
    try {
        Object result = messageFuture.get(timeoutMillis, TimeUnit.MILLISECONDS);
        doAfterRpcHooks(remoteAddr, rpcMessage, result);
        return result;
    } catch (Exception exx) {
        LOGGER.error("wait response error:{},ip:{},request:{}", exx.getMessage(), channel.remoteAddress(), rpcMessage.getBody());
        if (exx instanceof TimeoutException) {
            throw (TimeoutException) exx;
        } else {
            throw new RuntimeException(exx);
        }
    }
}

Example 13 with RpcMessage

use of io.seata.core.protocol.RpcMessage in project seata by seata.

the class AbstractNettyRemotingServer method sendAsyncRequest.

@Override
public void sendAsyncRequest(Channel channel, Object msg) {
    if (channel == null) {
        throw new RuntimeException("client is not connected");
    }
    RpcMessage rpcMessage = buildRequestMessage(msg, ProtocolConstants.MSGTYPE_RESQUEST_ONEWAY);
    super.sendAsync(channel, rpcMessage);
}

Example 14 with RpcMessage

use of io.seata.core.protocol.RpcMessage in project seata by seata.

the class AbstractNettyRemotingClient method sendSyncRequest.

@Override
public Object sendSyncRequest(Object msg) throws TimeoutException {
    String serverAddress = loadBalance(getTransactionServiceGroup(), msg);
    int timeoutMillis = NettyClientConfig.getRpcRequestTimeout();
    RpcMessage rpcMessage = buildRequestMessage(msg, ProtocolConstants.MSGTYPE_RESQUEST_SYNC);
    // put message into basketMap, @see MergedSendRunnable
    if (NettyClientConfig.isEnableClientBatchSendRequest()) {
        // send batch message is sync request, needs to create messageFuture and put it in futures.
        MessageFuture messageFuture = new MessageFuture();
        messageFuture.setRequestMessage(rpcMessage);
        messageFuture.setTimeout(timeoutMillis);
        futures.put(rpcMessage.getId(), messageFuture);
        // put message into basketMap
        BlockingQueue<RpcMessage> basket = CollectionUtils.computeIfAbsent(basketMap, serverAddress, key -> new LinkedBlockingQueue<>());
        if (!basket.offer(rpcMessage)) {
            LOGGER.error("put message into basketMap offer failed, serverAddress:{},rpcMessage:{}", serverAddress, rpcMessage);
            return null;
        }
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug("offer message: {}", rpcMessage.getBody());
        }
        if (!isSending) {
            synchronized (mergeLock) {
                mergeLock.notifyAll();
            }
        }
        try {
            return messageFuture.get(timeoutMillis, TimeUnit.MILLISECONDS);
        } catch (Exception exx) {
            LOGGER.error("wait response error:{},ip:{},request:{}", exx.getMessage(), serverAddress, rpcMessage.getBody());
            if (exx instanceof TimeoutException) {
                throw (TimeoutException) exx;
            } else {
                throw new RuntimeException(exx);
            }
        }
    } else {
        Channel channel = clientChannelManager.acquireChannel(serverAddress);
        return super.sendSync(channel, rpcMessage, timeoutMillis);
    }
}

Example 15 with RpcMessage

use of io.seata.core.protocol.RpcMessage in project seata by seata.

the class AbstractNettyRemotingClient method sendAsyncRequest.

@Override
public void sendAsyncRequest(Channel channel, Object msg) {
    if (channel == null) {
        LOGGER.warn("sendAsyncRequest nothing, caused by null channel.");
        return;
    }
    RpcMessage rpcMessage = buildRequestMessage(msg, msg instanceof HeartbeatMessage ? ProtocolConstants.MSGTYPE_HEARTBEAT_REQUEST : ProtocolConstants.MSGTYPE_RESQUEST_ONEWAY);
    if (rpcMessage.getBody() instanceof MergeMessage) {
        mergeMsgMap.put(rpcMessage.getId(), (MergeMessage) rpcMessage.getBody());
    }
    super.sendAsync(channel, rpcMessage);
}