Java tutorial
/** * 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 * */ public class QueryMsgByUniqueKeySubCommand implements SubCommand { @Override public String commandName() { return "queryMsgByUniqueKey"; } @Override public String commandDesc() { return "Query Message by Unique key"; } @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); opt = new Option("t", "topic", true, "The topic of msg"); opt.setRequired(true); options.addOption(opt); return options; } public static void queryById(final DefaultMQAdminExt admin, final String topic, final String msgId) throws MQClientException, RemotingException, MQBrokerException, InterruptedException, IOException { MessageExt msg = admin.viewMessage(topic, 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(); final String topic = commandLine.getOptionValue('t').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, topic, msgId); System.out.println(result); } else { queryById(defaultMQAdminExt, topic, 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 QueryMsgByUniqueKeySubCommand().commandName(), // "-n", "127.0.0.1:9876", // "-g", "CID_110", // "-d", "127.0.0.1@73376", // "-i", "0A654A3400002ABD00000011C3555205" // }); } }