重构CheckMsg命令

Author: vintagewang

bin/checkmsg.sh

@@ -43,6 +43,7 @@
 import com.alibaba.rocketmq.tools.command.consumer.DeleteSubscriptionGroupCommand;
 import com.alibaba.rocketmq.tools.command.consumer.StartMonitoringSubCommand;
 import com.alibaba.rocketmq.tools.command.consumer.UpdateSubGroupSubCommand;
+import com.alibaba.rocketmq.tools.command.message.CheckMsgSubCommand;
 import com.alibaba.rocketmq.tools.command.message.PrintMessageSubCommand;
 import com.alibaba.rocketmq.tools.command.message.QueryMsgByIdSubCommand;
 import com.alibaba.rocketmq.tools.command.message.QueryMsgByKeySubCommand;
@@ -107,6 +108,7 @@ public class MQAdminStartup {
         subCommandList.add(new CleanExpiredCQSubCommand());
 
         subCommandList.add(new StartMonitoringSubCommand());
+        subCommandList.add(new CheckMsgSubCommand());
     }
 
 

rocketmq-tools/src/main/java/com/alibaba/rocketmq/tools/command/MQAdminStartup.java

@@ -0,0 +1,58 @@
+package com.alibaba.rocketmq.tools.command.message;
+
+import org.apache.commons.cli.CommandLine;
+import org.apache.commons.cli.Option;
+import org.apache.commons.cli.Options;
+
+import com.alibaba.rocketmq.remoting.RPCHook;
+import com.alibaba.rocketmq.tools.command.SubCommand;
+
+
+/**
+ * @auther lansheng.zj
+ */
+public class CheckMsgSubCommand implements SubCommand {
+    @Override
+    public String commandName() {
+        return "checkMsg";
+    }
+
+
+    @Override
+    public String commandDesc() {
+        return "Check Message Store";
+    }
+
+
+    @Override
+    public Options buildCommandlineOptions(Options options) {
+        Option opt = new Option("p", "cStorePath", true, "cStorePath");
+        opt.setRequired(true);
+        options.addOption(opt);
+
+        opt = new Option("s", "cSize ", true, "cSize");
+        opt.setRequired(true);
+        options.addOption(opt);
+
+        opt = new Option("l", "lStorePath ", true, "lStorePath");
+        opt.setRequired(true);
+        options.addOption(opt);
+
+        opt = new Option("z", "lSize ", true, "lSize");
+        opt.setRequired(true);
+        options.addOption(opt);
+
+        return options;
+    }
+
+
+    @Override
+    public void execute(CommandLine commandLine, Options options, RPCHook rpcHook) {
+        Store store = new Store(commandLine.getOptionValue("cStorePath").trim(), //
+            Integer.parseInt(commandLine.getOptionValue("cSize").trim()),//
+            commandLine.getOptionValue("lStorePath").trim(), //
+            Integer.parseInt(commandLine.getOptionValue("lSize").trim()));
+        store.load();
+        store.traval(false);
+    }
+}

rocketmq-tools/src/main/java/com/alibaba/rocketmq/tools/command/message/CheckMsgSubCommand.java

@@ -1,4 +1,4 @@
-package com.alibaba.rocketmq.tools.msgcheck;
+package com.alibaba.rocketmq.tools.command.message;
 
 import java.io.File;
 import java.nio.ByteBuffer;