输出msg时添加 --lang 参数，用来选择要输出的语言

Author: yanglb

src/main/java/com/yanglb/codegen/core/generator/impl/MsgGeneratorImpl.java

@@ -29,6 +29,7 @@
 import com.yanglb.codegen.utils.Resources;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 
 
@@ -49,7 +50,7 @@ protected void onGeneration() throws CodeGenException {
         List<String> langList = new ArrayList<>();
         TableModel tableModel = list.get(0);
         for (String key : tableModel.getColumns()) {
-            if (!"id".equals(key)) {
+            if (needOutput(key)) {
                 langList.add(key);
 
                 settingMap.put("MsgLang", key);
@@ -69,4 +70,12 @@ protected void onGeneration() throws CodeGenException {
             }
         }
     }
+
+    private boolean needOutput(String key) {
+        if ("id".equals(key)) return false;
+        if ("default".equals(key)) return true;
+        String[] lang = parameterModel.getOptions().getOptionValues("lang");
+        if (lang == null || lang.length == 0) return true;
+        return Arrays.asList(lang).contains(key);
+    }
 }

src/main/java/com/yanglb/codegen/core/parser/impl/MsgParser.java

@@ -16,8 +16,26 @@
 package com.yanglb.codegen.core.parser.impl;
 
 import com.yanglb.codegen.core.parser.BaseParser;
+import org.apache.commons.cli.Option;
+import org.apache.commons.cli.Options;
 
 public class MsgParser extends BaseParser {
+    @Override
+    protected Options options() {
+        Options options = super.options();
+
+        Option lang = Option.builder()
+                .longOpt("lang")
+                .argName("lang")
+                .desc("待处理语言列表，无需填写默认语言(default)，默认为全部")
+                .hasArg(true)
+                .hasArgs()
+                .build();
+        options.addOption(lang);
+
+        return options;
+    }
+
     @Override
     protected boolean headerHelp() {
         System.out.println("生成多语言资源信息");