/** * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You under the Apache License, Version 2.0 * (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.alibaba.rocketmq.tools.command.message; import com.alibaba.rocketmq.client.exception.MQBrokerException; import com.alibaba.rocketmq.client.exception.MQClientException; import com.alibaba.rocketmq.common.UtilAll; import com.alibaba.rocketmq.common.message.MessageExt; import com.alibaba.rocketmq.common.protocol.body.ConsumeMessageDirectlyResult; import com.alibaba.rocketmq.remoting.RPCHook; import com.alibaba.rocketmq.remoting.common.RemotingHelper; import com.alibaba.rocketmq.remoting.exception.RemotingException; import com.alibaba.rocketmq.tools.admin.DefaultMQAdminExt; import com.alibaba.rocketmq.tools.admin.api.MessageTrack; import com.alibaba.rocketmq.tools.command.MQAdminStartup; import com.alibaba.rocketmq.tools.command.SubCommand; import org.apache.commons.cli.CommandLine; import org.apache.commons.cli.Option; import org.apache.commons.cli.Options; import java.io.DataOutputStream; import java.io.File; import java.io.FileOutputStream; import java.io.IOException; import java.util.List; /** * @author shijia.wxr<[email protected]> */ public class QueryMsgByIdSubCommand implements SubCommand { @Override public String commandName() { return "queryMsgById"; } @Override public String commandDesc() { return "Query Message by Id"; } @Override public Options buildCommandlineOptions(Options options) { Option opt = new Option("i", "msgId", true, "Message Id"); opt.setRequired(true); options.addOption(opt); opt = new Option("g", "consumerGroup", true, "consumer group name"); opt.setRequired(false); options.addOption(opt); opt = new Option("d", "clientId", true, "The consumer's client id"); opt.setRequired(false); options.addOption(opt); return options; } public static void queryById(final DefaultMQAdminExt admin, final String msgId) throws MQClientException, RemotingException, MQBrokerException, InterruptedException, IOException { MessageExt msg = admin.viewMessage(msgId); String bodyTmpFilePath = createBodyFile(msg); System.out.printf("%-20s %s\n",// "Topic:",// msg.getTopic()// ); System.out.printf("%-20s %s\n",// "Tags:",// "[" + msg.getTags() + "]"// ); System.out.printf("%-20s %s\n",// "Keys:",// "[" + msg.getKeys() + "]"// ); System.out.printf("%-20s %d\n",// "Queue ID:",// msg.getQueueId()// ); System.out.printf("%-20s %d\n",// "Queue Offset:",// msg.getQueueOffset()// ); System.out.printf("%-20s %d\n",// "CommitLog Offset:",// msg.getCommitLogOffset()// ); System.out.printf("%-20s %d\n",// "Reconsume Times:",// msg.getReconsumeTimes()// ); System.out.printf("%-20s %s\n",// "Born Timestamp:",// UtilAll.timeMillisToHumanString2(msg.getBornTimestamp())// ); System.out.printf("%-20s %s\n",// "Store Timestamp:",// UtilAll.timeMillisToHumanString2(msg.getStoreTimestamp())// ); System.out.printf("%-20s %s\n",// "Born Host:",// RemotingHelper.parseSocketAddressAddr(msg.getBornHost())// ); System.out.printf("%-20s %s\n",// "Store Host:",// RemotingHelper.parseSocketAddressAddr(msg.getStoreHost())// ); System.out.printf("%-20s %d\n",// "System Flag:",// msg.getSysFlag()// ); System.out.printf("%-20s %s\n",// "Properties:",// msg.getProperties() != null ? msg.getProperties().toString() : ""// ); System.out.printf("%-20s %s\n",// "Message Body Path:",// bodyTmpFilePath// ); try { List<MessageTrack> mtdList = admin.messageTrackDetail(msg); if (mtdList.isEmpty()) { System.out.println("\n\nWARN: No Consumer"); } else { System.out.println("\n\n"); for (MessageTrack mt : mtdList) { System.out.println(mt); } } } catch (Exception e) { e.printStackTrace(); } } @Override public void execute(CommandLine commandLine, Options options, RPCHook rpcHook) { DefaultMQAdminExt defaultMQAdminExt = new DefaultMQAdminExt(rpcHook); defaultMQAdminExt.setInstanceName(Long.toString(System.currentTimeMillis())); try { defaultMQAdminExt.start(); final String msgId = commandLine.getOptionValue('i').trim(); if (commandLine.hasOption('g') && commandLine.hasOption('d')) { final String consumerGroup = commandLine.getOptionValue('g').trim(); final String clientId = commandLine.getOptionValue('d').trim(); ConsumeMessageDirectlyResult result = defaultMQAdminExt.consumeMessageDirectly(consumerGroup, clientId, msgId); System.out.println(result); } else { queryById(defaultMQAdminExt, msgId); } } catch (Exception e) { e.printStackTrace(); } finally { defaultMQAdminExt.shutdown(); } } private static String createBodyFile(MessageExt msg) throws IOException { DataOutputStream dos = null; try { String bodyTmpFilePath = "/tmp/rocketmq/msgbodys"; File file = new File(bodyTmpFilePath); if (!file.exists()) { file.mkdirs(); } bodyTmpFilePath = bodyTmpFilePath + "/" + msg.getMsgId(); dos = new DataOutputStream(new FileOutputStream(bodyTmpFilePath)); dos.write(msg.getBody()); return bodyTmpFilePath; } finally { if (dos != null) dos.close(); } } public static void main(String[] args) { MQAdminStartup.main(new String[] { new QueryMsgByIdSubCommand().commandName(), // "-n", "127.0.0.1:9876", // "-g", "CID_110", // "-d", "127.0.0.1@73376", // "-i", "0A654A3400002ABD00000011C3555205" // }); } }