Merge pull request #1306 from halower/3.0

MP代码生成器添加达梦数据库生成支持

Author: qmdx

build.gradle

@@ -66,6 +66,7 @@ ext {
         "sqlserver"             : "com.microsoft.sqlserver:sqljdbc4:4.0",
         "postgresql"            : "org.postgresql:postgresql:9.4.1212",
         "oracle"                : fileTree(dir: 'libs', includes: ['ojdbc-11.2.0.3-jdk16.jar']),
+        "dm"                    : fileTree(dir: 'libs', includes: ["jdbcDriver-18.jar"]),
         "h2"                    : "com.h2database:h2:1.4.197",
         "mysql"                 : "mysql:mysql-connector-java:8.0.15",
         "sqlite"                : "org.xerial:sqlite-jdbc:3.27.2.1",

libs/jdbcDriver-18.jar

@@ -8,6 +8,7 @@ dependencies {
     testCompile "${lib.sqlserver}"
     testCompile "${lib.postgresql}"
     testCompile lib.oracle as ConfigurableFileTree
+    testCompile lib.dm as ConfigurableFileTree
     testCompile "${lib.h2}"
     testCompile "${lib.mysql}"
     testCompile "${lib.sqlite}"

mybatis-plus-generator/build.gradle

@@ -93,6 +93,9 @@ public IDbQuery getDbQuery() {
                 case SQLITE:
                     dbQuery = new SqliteQuery();
                     break;
+                case DM:
+                    dbQuery = new DMQuery();
+                    break;
                 default:
                     // 默认 MYSQL
                     dbQuery = new MySqlQuery();
@@ -167,6 +170,9 @@ public ITypeConvert getTypeConvert() {
                 case SQLITE:
                     typeConvert = new SqliteTypeConvert();
                     break;
+                case DM:
+                    typeConvert = new DmTypeConvert();
+                    break;
                 case MARIADB:
                     typeConvert = new MySqlTypeConvert();
                     break;

mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/DataSourceConfig.java

@@ -556,7 +556,11 @@ private TableInfo convertTableFields(TableInfo tableInfo, StrategyConfig config)
             } else if (DbType.ORACLE == dbType) {
                 tableName = tableName.toUpperCase();
                 tableFieldsSql = String.format(tableFieldsSql.replace("#schema", dataSourceConfig.getSchemaName()), tableName);
-            } else if (DbType.H2 == dbType) {
+            } else if (DbType.DM == dbType) {
+                tableName = tableName.toUpperCase();
+                tableFieldsSql = String.format(tableFieldsSql, tableName);
+            }
+            else if (DbType.H2 == dbType) {
                 tableName = tableName.toUpperCase();
                 try (PreparedStatement pkQueryStmt = connection.prepareStatement(String.format(H2Query.PK_QUERY_SQL, tableName));
                      ResultSet pkResults = pkQueryStmt.executeQuery()) {

mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/builder/ConfigBuilder.java

@@ -26,7 +26,7 @@
  * @author zhanyao
  * @since 2018-05-16
  */
-public class DB2TypeConvert implements ITypeConvert {
+public class  DB2TypeConvert implements ITypeConvert {
 
     @Override
     public IColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {

mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/converts/DB2TypeConvert.java

@@ -0,0 +1,138 @@
+/*
+ * Copyright (c) 2011-2020, baomidou ([email protected]).
+ * <p>
+ * Licensed 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
+ * <p>
+ * https://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * 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.baomidou.mybatisplus.generator.config.converts;
+
+import com.baomidou.mybatisplus.generator.config.GlobalConfig;
+import com.baomidou.mybatisplus.generator.config.ITypeConvert;
+import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
+import com.baomidou.mybatisplus.generator.config.rules.IColumnType;
+
+/**
+ * DM 字段类型转换
+ *
+ * @author halower
+ * @since 2019-06-27
+ */
+public class DmTypeConvert implements ITypeConvert {
+
+    @Override
+    public IColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
+        String t = fieldType.toLowerCase();
+        DbColumnType dbColumnType = new BasicDataTypeHandler().getDbColumnType(t);
+        return dbColumnType;
+    }
+
+     interface Handler {
+         DbColumnType getDbColumnType(String fieldType);
+    }
+
+    /**
+     * 常规数据类型
+     */
+    class BasicDataTypeHandler implements Handler {
+        @Override
+        public DbColumnType getDbColumnType(String fieldType) {
+            //字符数据类型: CHAR,CHARACTER,VARCHAR
+            if(fieldType.contains("char")) {
+                return DbColumnType.STRING;
+            }
+            /**
+             *  数值数据类型:
+             *  NUMERIC,DECIMAL,DEC,MONEY,BIT,BOOL,BOOLEAN,INTEGER,INT,BIGINT
+             *  TINYINT,BYTE,SMALLINT,BINARY,VARBINARY
+             */
+            else if (fieldType.contains("numeric")
+                || fieldType.contains("decimal")
+                || fieldType.contains("dec")
+                || fieldType.contains("money")) {
+                return DbColumnType.BIG_DECIMAL;
+            }
+            else if(fieldType.contains("bit")
+                || fieldType.contains("bool")
+                || fieldType.contains("boolean")){
+                return DbColumnType.BOOLEAN;
+            }
+            else if(fieldType.contains("integer") || fieldType.contains("int")){
+                return DbColumnType.INTEGER;
+            }
+            else if(fieldType.contains("bigint") ){
+                return DbColumnType.BIG_INTEGER;
+            }
+            else if(fieldType.contains("tinyint")
+                || fieldType.contains("byte")
+                || fieldType.contains("smallint")
+                ){
+                return DbColumnType.INTEGER;
+            }
+            else if (fieldType.contains("binary")
+                || fieldType.contains("varbinary")
+                ) {
+                return DbColumnType.BYTE_ARRAY;
+            }
+            /**
+             * 近似数值数据类型:
+             * FLOAT
+             */
+            else if (fieldType.contains("float")) {
+                return DbColumnType.FLOAT;
+            }
+            /**
+             * DOUBLE, DOUBLE PRECISION,REAL
+             */
+            else if (fieldType.contains("double") || fieldType.contains("real")) {
+                return DbColumnType.DOUBLE;
+            }
+           return new DateTimeDataTypeHandler().getDbColumnType(fieldType);
+        }
+    }
+
+    /**
+     *  日期时间数据类型
+     */
+    class DateTimeDataTypeHandler implements Handler{
+
+        @Override
+        public DbColumnType getDbColumnType(String fieldType) {
+           if (fieldType.contains("date")
+               || fieldType.contains("time")
+               || fieldType.contains("timestamp")) {
+                return DbColumnType.DATE;
+            }
+            return new MultimediaDataTypeHandler().getDbColumnType(fieldType);
+        }
+    }
+
+    /**
+     *  多媒体数据类型
+     *  TEXT,LONGVARCHAR,CLOB,BLOB,IMAGE
+     */
+
+    class MultimediaDataTypeHandler implements Handler{
+        @Override
+        public DbColumnType getDbColumnType(String fieldType) {
+            if (fieldType.contains("text")|| fieldType.contains("longvarchar")) {
+                return DbColumnType.STRING;
+            } else if (fieldType.contains("clob")) {
+                return DbColumnType.CLOB;
+            } else if (fieldType.contains("blob")) {
+                return DbColumnType.BLOB;
+            } else if (fieldType.contains("image")) {
+                return DbColumnType.BYTE_ARRAY;
+            }
+            return DbColumnType.STRING;
+        }
+    }
+}

mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/converts/DmTypeConvert.java

@@ -0,0 +1,68 @@
+package com.baomidou.mybatisplus.generator.config.querys;
+
+import com.baomidou.mybatisplus.annotation.DbType;
+
+/**
+ * DM 表数据查询
+ *
+ * @author halower
+ * @since 2019-06-27
+ */
+public class DMQuery  extends AbstractDbQuery{
+    @Override
+    public DbType dbType() {
+        return DbType.DM;
+    }
+
+    @Override
+    public String tablesSql() {
+        return "SELECT DISTINCT T1.TABLE_NAME,T2.COMMENTS AS TABLE_COMMENT FROM USER_TAB_COLUMNS T1 " +
+            "INNER JOIN USER_TAB_COMMENTS T2 ON T1.TABLE_NAME = T2.TABLE_NAME";
+    }
+
+    @Override
+    public String tableFieldsSql() {
+        return
+            "SELECT T2.COLUMN_NAME,T1.COMMENTS,T2.DATA_TYPE," +
+            "CASE WHEN CONSTRAINT_TYPE='P' THEN 'PRI' END AS KEY " +
+            "FROM  USER_COL_COMMENTS T1,USER_TAB_COLUMNS T2," +
+              "( SELECT T4.TABLE_NAME ,T4.COLUMN_NAME,T5.CONSTRAINT_TYPE " +
+               " FROM USER_CONS_COLUMNS T4,USER_CONSTRAINTS T5 " +
+               "WHERE T4.CONSTRAINT_NAME = T5.CONSTRAINT_NAME AND T5.CONSTRAINT_TYPE = 'P') T3 " +
+            "WHERE T1.TABLE_NAME = T2.TABLE_NAME AND "+
+            "T1.COLUMN_NAME=T2.COLUMN_NAME AND "+
+            "T1.COLUMN_NAME=T2.COLUMN_NAME AND "+
+            "T1.TABLE_NAME=T3.TABLE_NAME AND "+
+            "T1.COLUMN_NAME=T3.COLUMN_NAME AND "+
+             "T1.TABLE_NAME = '%s'";
+    }
+
+    @Override
+    public String tableName() {
+        return "TABLE_NAME";
+    }
+    @Override
+    public String tableComment() {
+        return "TABLE_COMMENT";
+    }
+
+    @Override
+    public String fieldName() {
+        return "COLUMN_NAME";
+    }
+
+    @Override
+    public String fieldType() {
+        return "DATA_TYPE";
+    }
+
+    @Override
+    public String fieldComment() {
+        return "COMMENTS";
+    }
+
+    @Override
+    public String fieldKey() {
+        return "KEY";
+    }
+}