v4.1.0

改进dml生成功能

Author: 杨利兵

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

@@ -16,8 +16,23 @@
 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 DmlParser extends BaseParser {
+    @Override
+    protected Options options() {
+        Options options = super.options();
+
+        Option target = Option.builder()
+                .longOpt("target")
+                .hasArg()
+                .desc("指定生成目标，可选 mysql/mssql/sqlite，默认为mysql")
+                .build();
+        options.addOption(target);
+        return options;
+    }
+
     @Override
     protected boolean headerHelp() {
         System.out.println("生成初始数据SQL脚本，用于向数据添加初始数据。");

src/main/java/com/yanglb/codegen/core/translator/BaseSqlTranslator.java

@@ -17,8 +17,6 @@
 
 import java.util.List;
 
-import com.yanglb.codegen.exceptions.CodeGenException;
-
 public class BaseSqlTranslator<T> extends BaseTranslator<List<T>> {
 	// 列引号字符，如：SQL Server为[], MySQL为`等
 	protected String sqlColumnStart = "\"";

src/main/java/com/yanglb/codegen/core/translator/impl/DdlSqliteTranslatorImpl.java

@@ -28,6 +28,11 @@
 
 public class DdlSqliteTranslatorImpl extends BaseDdlTranslator {
 
+	public DdlSqliteTranslatorImpl() {
+		this.sqlColumnStart = "\"";
+		this.sqlColumnEnd = "\"";
+	}
+
 	@Override
 	protected void onBeforeTranslate() throws CodeGenException {
 		super.onBeforeTranslate();

src/main/java/com/yanglb/codegen/core/translator/impl/DmlTranslatorImpl.java

@@ -41,6 +41,22 @@ protected void onBeforeTranslate() throws CodeGenException {
 			fileName = fileName.substring(0, index);
 		}
 		this.writableModel.setFileName(fileName);
+
+		String target = parameterModel.getOptions().getOptionValue("target", "mysql");
+		switch (target) {
+			case "mysql":
+				sqlColumnStart = "`";
+				sqlColumnEnd = "`";
+				break;
+			case "mssql":
+				sqlColumnStart = "[";
+				sqlColumnEnd = "[";
+				break;
+			case "sqlite":
+				sqlColumnStart = "\"";
+				sqlColumnEnd = "\"";
+				break;
+		}
 	}
 
 	@Override

src/main/java/com/yanglb/codegen/utils/Infos.java

@@ -17,7 +17,7 @@
 
 public class Infos {
 	public static final String Name       = "Code Generator";
-	public static final String Version    = "4.0.3";
+	public static final String Version    = "4.1.0";
 	public static final String Copyright  = "Copyright 2015-2020 yanglb.com All Rights Reserved.";
 	public static final String Author     = "me@yanglb.com";
 	public static final String Website    = "https://yanglb.com";