瀏覽代碼

!79 sqlite code generator support bugfix
Merge pull request !79 from blueden/master

blueden 6 年之前
父節點
當前提交
84e35540f8

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

@@ -140,6 +140,8 @@ public class DataSourceConfig {
             return DbType.DB2;
         } else if (str.contains("mariadb")) {
             return DbType.MARIADB;
+        } else if (str.contains("sqlite")) {
+            return DbType.MARIADB;
         } else if (str.contains("h2")) {
             return DbType.H2;
         } else {
@@ -162,6 +164,9 @@ public class DataSourceConfig {
                 case DB2:
                     typeConvert = new DB2TypeConvert();
                     break;
+                case SQLITE:
+                    typeConvert = new SqliteTypeConvert();
+                    break;
                 case MARIADB:
                     typeConvert = new MySqlTypeConvert();
                     break;

+ 1 - 1
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/converts/SqliteTypeConvert.java

@@ -20,7 +20,7 @@ public class SqliteTypeConvert implements ITypeConvert {
             return DbColumnType.BOOLEAN;
         } else if (t.contains("int")) {
             return DbColumnType.INTEGER;
-        } else if (t.contains("text") || t.contains("char") || t.contains("enum") || t.contains("set")) {
+        } else if (t.contains("text") || t.contains("char") || t.contains("enum") ) {
             return DbColumnType.STRING;
         } else if (t.contains("decimal") || t.contains("numeric")) {
             return DbColumnType.BIG_DECIMAL;