1. 程式人生 > >【原始碼】RocketMQ如何實現獲取指定訊息

【原始碼】RocketMQ如何實現獲取指定訊息

概要

訊息查詢是什麼?

訊息查詢就是根據使用者提供的msgId從MQ中取出該訊息

RocketMQ如果有多個節點如何查詢?

問題:RocketMQ分散式結構中,資料分散在各個節點,即便是同一Topic的資料,也未必都在一個broker上。客戶端怎麼知道資料該去哪個節點上查?

猜想1:逐個訪問broker節點查詢資料

猜想2:有某種資料中心存在,該中心知道所有訊息儲存的位置,只要向該中心查詢即可得到訊息具體位置,進而取得訊息內容

實際:

1.訊息Id中含有訊息所在的broker的地址資訊(IP\Port)以及該訊息在CommitLog中的偏移量。

2.客戶端實現會從msgId字串中解析出broker地址,向指定broker節查詢訊息。

問題:CommitLog檔案有多個,只有偏移量估計不能確定在哪個檔案吧?

實際:單個Broker節點內offset是全域性唯一的,不是每個CommitLog檔案的偏移量都是從0開始的。單個節點內所有CommitLog檔案共用一套偏移量,每個檔案的檔名為其第一個訊息的偏移量。所以可以根據偏移量和檔名確定CommitLog檔案。

原始碼閱讀

0.使用方式

MessageExt  msg = consumer.viewMessage(msgId);

1.訊息ID解析

這個瞭解下就可以了

public class MessageId {
    private SocketAddress address;
    private long offset;

    public MessageId(SocketAddress address, long offset) {
        this.address = address;
        this.offset = offset;
    }

    //get-set
}

//from MQAdminImpl.java
public MessageExt viewMessage(
    String msgId) throws RemotingException, MQBrokerException, InterruptedException, MQClientException {

    MessageId messageId = null;
    try {
        //從msgId字串中解析出address和offset
        //address = ip:port
        //offset為訊息在CommitLog檔案中的偏移量
        messageId = MessageDecoder.decodeMessageId(msgId);
    } catch (Exception e) {
        throw new MQClientException(ResponseCode.NO_MESSAGE, "query message by id finished, but no message.");
    }
    return this.mQClientFactory.getMQClientAPIImpl().viewMessage(RemotingUtil.socketAddress2String(messageId.getAddress()),
        messageId.getOffset(), timeoutMillis);
}

//from MessageDecoder.java
public static MessageId decodeMessageId(final String msgId) throws UnknownHostException {
    SocketAddress address;
    long offset;
    //ipv4和ipv6的區別
    //如果msgId總長度超過32字元,則為ipv6
    int ipLength = msgId.length() == 32 ? 4 * 2 : 16 * 2;

    byte[] ip = UtilAll.string2bytes(msgId.substring(0, ipLength));
    byte[] port = UtilAll.string2bytes(msgId.substring(ipLength, ipLength + 8));
    ByteBuffer bb = ByteBuffer.wrap(port);
    int portInt = bb.getInt(0);
    address = new InetSocketAddress(InetAddress.getByAddress(ip), portInt);

    // offset
    byte[] data = UtilAll.string2bytes(msgId.substring(ipLength + 8, ipLength + 8 + 16));
    bb = ByteBuffer.wrap(data);
    offset = bb.getLong(0);

    return new MessageId(address, offset);
}

2.長連線客戶端RPC實現

要發請求首先得先建立連線,這裡方法可以看到建立連線相關的操作。值得注意的是,第一次訪問的時候可能連線還沒建立,建立連線需要消耗一段時間。程式碼中對這個時間也做了判斷,如果連線建立完成後,發現已經超時,則不再發出請求。目的應該是儘可能減少請求執行緒的阻塞時間。

//from NettyRemotingClient.java
@Override
public RemotingCommand invokeSync(String addr, final RemotingCommand request, long timeoutMillis)
    throws InterruptedException, RemotingConnectException, RemotingSendRequestException, RemotingTimeoutException {
    long beginStartTime = System.currentTimeMillis();
    //這裡會先檢查有無該地址的通道,有則返回,無則建立
    final Channel channel = this.getAndCreateChannel(addr);
    if (channel != null && channel.isActive()) {
        try {
            //前置鉤子
            doBeforeRpcHooks(addr, request); 
            //判斷通道建立完成時是否已到達超時時間,如果超時直接丟擲異常。不發請求
            long costTime = System.currentTimeMillis() - beginStartTime;
            if (timeoutMillis < costTime) {
                throw new RemotingTimeoutException("invokeSync call timeout");
            }
            //同步呼叫
            RemotingCommand response = this.invokeSyncImpl(channel, request, timeoutMillis - costTime);
            //後置鉤子
            doAfterRpcHooks(RemotingHelper.parseChannelRemoteAddr(channel), request, response); //後置鉤子
            return response;
        } catch (RemotingSendRequestException e) {
            log.warn("invokeSync: send request exception, so close the channel[{}]", addr);
            this.closeChannel(addr, channel);
            throw e;
        } catch (RemotingTimeoutException e) {
            if (nettyClientConfig.isClientCloseSocketIfTimeout()) {
                this.closeChannel(addr, channel);
                log.warn("invokeSync: close socket because of timeout, {}ms, {}", timeoutMillis, addr);
            }
            log.warn("invokeSync: wait response timeout exception, the channel[{}]", addr);
            throw e;
        }
    } else {
        this.closeChannel(addr, channel);
        throw new RemotingConnectException(addr);
    }
}

下一步看看它的同步呼叫做了什麼處理。注意到它會構建一個Future物件加入待響應池,發出請求報文後就掛起執行緒,然後等待喚醒(waitResponse內部使用CountDownLatch等待)。

//from NettyRemotingAbstract.java
public RemotingCommand invokeSyncImpl(final Channel channel, final RemotingCommand request, final long timeoutMillis) throws InterruptedException, RemotingSendRequestException, RemotingTimeoutException { //請求id final int opaque = request.getOpaque(); try { //請求存根 final ResponseFuture responseFuture = new ResponseFuture(channel, opaque, timeoutMillis, null, null); //加入待響應的請求池 this.responseTable.put(opaque, responseFuture); final SocketAddress addr = channel.remoteAddress(); //將請求發出,成功發出時更新狀態 channel.writeAndFlush(request).addListener(new ChannelFutureListener() { @Override public void operationComplete(ChannelFuture f) throws Exception { if (f.isSuccess()) { //若成功發出,更新請求狀態為“已發出” responseFuture.setSendRequestOK(true); return; } else { responseFuture.setSendRequestOK(false); } //若發出失敗,則從池中移除(沒用了,釋放資源) responseTable.remove(opaque); responseFuture.setCause(f.cause()); //putResponse的時候會喚醒等待的執行緒 responseFuture.putResponse(null); log.warn("send a request command to channel <" + addr + "> failed."); } }); //只等待一段時間,不會一直等下去 //若正常響應,則收到響應後,此執行緒會被喚醒,繼續執行下去 //若超時,則到達該時間後執行緒甦醒,繼續執行 RemotingCommand responseCommand = responseFuture.waitResponse(timeoutMillis); if (null == responseCommand) { if (responseFuture.isSendRequestOK()) { throw new RemotingTimeoutException(RemotingHelper.parseSocketAddressAddr(addr), timeoutMillis, responseFuture.getCause()); } else { throw new RemotingSendRequestException(RemotingHelper.parseSocketAddressAddr(addr), responseFuture.getCause()); } } return responseCommand; } finally { //正常響應完成時,將future釋放(正常邏輯) //超時時,將future釋放。這個請求已經作廢了,後面如果再收到響應,就可以直接丟棄了(由於找不到相關的響應鉤子,就不處理了) this.responseTable.remove(opaque); } }

好,我們再來看看收到報文的時候是怎麼處理的。我們都瞭解JDK中的Future的原理,大概就是將這個任務提交給其他執行緒處理,該執行緒處理完畢後會將結果寫入到Future物件中,寫入時如果有執行緒在等待該結果,則喚醒這些執行緒。這裡也差不多,只不過執行執行緒在服務端,服務執行完畢後會將結果通過長連線傳送給客戶端,客戶端收到後根據報文中的ID資訊從待響應池中找到Future物件,然後就是類似的處理了。

class NettyClientHandler extends SimpleChannelInboundHandler<RemotingCommand> {

    //底層解碼完畢得到RemotingCommand的報文
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, RemotingCommand msg) throws Exception {
        processMessageReceived(ctx, msg);
    }
}

public void processMessageReceived(ChannelHandlerContext ctx, RemotingCommand msg) throws Exception {
    final RemotingCommand cmd = msg;
    if (cmd != null) {
        //判斷型別
        switch (cmd.getType()) {
            case REQUEST_COMMAND:
                processRequestCommand(ctx, cmd);
                break;
            case RESPONSE_COMMAND:
                processResponseCommand(ctx, cmd);
                break;
            default:
                break;
        }
    }
}

public void processResponseCommand(ChannelHandlerContext ctx, RemotingCommand cmd) {
    //取得訊息id
    final int opaque = cmd.getOpaque();
    //從待響應池中取得對應請求
    final ResponseFuture responseFuture = responseTable.get(opaque);
    if (responseFuture != null) {
        //將響應值注入到ResponseFuture物件中,等待執行緒可從這個物件獲取結果
        responseFuture.setResponseCommand(cmd);
        //請求已處理完畢,釋放該請求
        responseTable.remove(opaque);

        //如果有回撥函式的話則回撥(由當前執行緒處理)
        if (responseFuture.getInvokeCallback() != null) {
            executeInvokeCallback(responseFuture);
        } else {
            //沒有的話,則喚醒等待執行緒(由等待執行緒做處理)
            responseFuture.putResponse(cmd);
            responseFuture.release();
        }
    } else {
        log.warn("receive response, but not matched any request, " + RemotingHelper.parseChannelRemoteAddr(ctx.channel()));
        log.warn(cmd.toString());
    }
}

總結一下,客戶端的處理時序大概是這樣的:

結構大概是這樣的:

3.服務端的處理

//todo 服務端待補充CommitLog檔案對映相關內容

class NettyServerHandler extends SimpleChannelInboundHandler<RemotingCommand> {

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, RemotingCommand msg) throws Exception {
        processMessageReceived(ctx, msg);
    }
}

//from NettyRemotingAbscract.java
public void processMessageReceived(ChannelHandlerContext ctx, RemotingCommand msg) throws Exception {
    final RemotingCommand cmd = msg;
    if (cmd != null) {
        switch (cmd.getType()) {
            case REQUEST_COMMAND: //服務端走這裡
                processRequestCommand(ctx, cmd);
                break;
            case RESPONSE_COMMAND:
                processResponseCommand(ctx, cmd);
                break;
            default:
                break;
        }
    }
}

//from NettyRemotingAbscract.java
public void processRequestCommand(final ChannelHandlerContext ctx, final RemotingCommand cmd) {
    //檢視有無該請求code相關的處理器
    final Pair<NettyRequestProcessor, ExecutorService> matched = this.processorTable.get(cmd.getCode());
    //如果沒有,則使用預設處理器(可能沒有預設處理器)
    final Pair<NettyRequestProcessor, ExecutorService> pair = null == matched ? this.defaultRequestProcessor : matched;
    final int opaque = cmd.getOpaque();

    if (pair != null) {
        Runnable run = new Runnable() {
            @Override
            public void run() {
                try {
                    doBeforeRpcHooks(RemotingHelper.parseChannelRemoteAddr(ctx.channel()), cmd);
                    final RemotingResponseCallback callback = new RemotingResponseCallback() {
                        @Override
                        public void callback(RemotingCommand response) {
                            doAfterRpcHooks(RemotingHelper.parseChannelRemoteAddr(ctx.channel()), cmd, response);
                            if (!cmd.isOnewayRPC()) {
                                if (response != null) { //不為null,則由本類將響應值寫會給請求方
                                    response.setOpaque(opaque);
                                    response.markResponseType();
                                    try {
                                        ctx.writeAndFlush(response);
                                    } catch (Throwable e) {
                                        log.error("process request over, but response failed", e);
                                        log.error(cmd.toString());
                                        log.error(response.toString());
                                    }
                                } else { //為null,意味著processor內部已經將響應處理了,這裡無需再處理。
                                }
                            }
                        }
                    };
                    if (pair.getObject1() instanceof AsyncNettyRequestProcessor) {//QueryMessageProcessor為非同步處理器
                        AsyncNettyRequestProcessor processor = (AsyncNettyRequestProcessor)pair.getObject1();
                        processor.asyncProcessRequest(ctx, cmd, callback);
                    } else { 
                        NettyRequestProcessor processor = pair.getObject1();
                        RemotingCommand response = processor.processRequest(ctx, cmd);
                        doAfterRpcHooks(RemotingHelper.parseChannelRemoteAddr(ctx.channel()), cmd, response);
                        callback.callback(response);
                    }
                } catch (Throwable e) {
                    log.error("process request exception", e);
                    log.error(cmd.toString());

                    if (!cmd.isOnewayRPC()) {
                        final RemotingCommand response = RemotingCommand.createResponseCommand(RemotingSysResponseCode.SYSTEM_ERROR,
                            RemotingHelper.exceptionSimpleDesc(e));
                        response.setOpaque(opaque);
                        ctx.writeAndFlush(response);
                    }
                }
            }
        };

        if (pair.getObject1().rejectRequest()) {
            final RemotingCommand response = RemotingCommand.createResponseCommand(RemotingSysResponseCode.SYSTEM_BUSY,
                "[REJECTREQUEST]system busy, start flow control for a while");
            response.setOpaque(opaque);
            ctx.writeAndFlush(response);
            return;
        }

        try {
            final RequestTask requestTask = new RequestTask(run, ctx.channel(), cmd);
            pair.getObject2().submit(requestTask);
        } catch (RejectedExecutionException e) {
            if ((System.currentTimeMillis() % 10000) == 0) {
                log.warn(RemotingHelper.parseChannelRemoteAddr(ctx.channel())
                    + ", too many requests and system thread pool busy, RejectedExecutionException "
                    + pair.getObject2().toString()
                    + " request code: " + cmd.getCode());
            }

            if (!cmd.isOnewayRPC()) {
                final RemotingCommand response = RemotingCommand.createResponseCommand(RemotingSysResponseCode.SYSTEM_BUSY,
                    "[OVERLOAD]system busy, start flow control for a while");
                response.setOpaque(opaque);
                ctx.writeAndFlush(response);
            }
        }
    } else {
        String error = " request type " + cmd.getCode() + " not supported";
        final RemotingCommand response =
            RemotingCommand.createResponseCommand(RemotingSysResponseCode.REQUEST_CODE_NOT_SUPPORTED, error);
        response.setOpaque(opaque);
        ctx.writeAndFlush(response);
        log.error(RemotingHelper.parseChannelRemoteAddr(ctx.channel()) + error);
    }
}

//from QueryMessageProcesor.java
@Override
public RemotingCommand processRequest(ChannelHandlerContext ctx, RemotingCommand request)
    throws RemotingCommandException {
    switch (request.getCode()) {
        case RequestCode.QUERY_MESSAGE:
            return this.queryMessage(ctx, request);
        case RequestCode.VIEW_MESSAGE_BY_ID: //通過msgId查詢訊息
            return this.viewMessageById(ctx, request);
        default:
            break;
    }

    return null;
}

public RemotingCommand viewMessageById(ChannelHandlerContext ctx, RemotingCommand request)
    throws RemotingCommandException {
    final RemotingCommand response = RemotingCommand.createResponseCommand(null);
    final ViewMessageRequestHeader requestHeader =
        (ViewMessageRequestHeader) request.decodeCommandCustomHeader(ViewMessageRequestHeader.class);

    response.setOpaque(request.getOpaque());

    //getMessagetStore得到當前對映到記憶體中的CommitLog檔案,然後根據偏移量取得資料
    final SelectMappedBufferResult selectMappedBufferResult =
        this.brokerController.getMessageStore().selectOneMessageByOffset(requestHeader.getOffset());
    if (selectMappedBufferResult != null) {
        response.setCode(ResponseCode.SUCCESS);
        response.setRemark(null);

        //將響應通過socket寫回給客戶端
        try {
            //response物件的資料作為header
            //訊息內容作為body
            FileRegion fileRegion =
                new OneMessageTransfer(response.encodeHeader(selectMappedBufferResult.getSize()),
                    selectMappedBufferResult);
            ctx.channel().writeAndFlush(fileRegion).addListener(new ChannelFutureListener() {
                @Override
                public void operationComplete(ChannelFuture future) throws Exception {
                    selectMappedBufferResult.release();
                    if (!future.isSuccess()) {
                        log.error("Transfer one message from page cache failed, ", future.cause());
                    }
                }
            });
        } catch (Throwable e) {
            log.error("", e);
            selectMappedBufferResult.release();
        }

        return null; //如果有值,則直接寫回給請求方。這裡返回null是不需要由外層處理響應。
    } else {
        response.setCode(ResponseCode.SYSTEM_ERROR);
        response.setRemark("can not find message by the offset, " + requestHeader.getOffset());
    }

    return response;
}

&n