Kaynağa Gözat

mybatis-plus代码生成修改

skyline 1 yıl önce
ebeveyn
işleme
d5273d07c6

+ 20 - 1
admin/src/main/java/com/kym/admin/utils/MybatisPlusGeneratorForAdmin.java

@@ -2,9 +2,15 @@ package com.kym.admin.utils;
 
 import com.baomidou.mybatisplus.generator.FastAutoGenerator;
 import com.baomidou.mybatisplus.generator.config.OutputFile;
+import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
 import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
 import com.kym.entity.BaseEntity;
+import com.kym.mapper.mybatisplus.MyBaseMapper;
+import com.kym.service.mybatisplus.MyBaseService;
+import com.kym.service.mybatisplus.MyBaseServiceImpl;
+import org.apache.poi.ss.formula.functions.T;
 
+import java.sql.Types;
 import java.util.Map;
 
 /**
@@ -16,12 +22,22 @@ public class MybatisPlusGeneratorForAdmin {
 
     public static void main(String[] args) {
         // 指定数据源
-        FastAutoGenerator.create("jdbc:mysql://121.40.98.15:3307/charge_admin", "kym", "kx7#zFvS$$%uVDkZPk")
+        FastAutoGenerator.create("jdbc:mysql://121.40.98.15:3307/charge_admin", "root", "KuaiyuMan/*-")
                 .globalConfig(builder -> {
                     builder.author("skyline")
                             // 指定输出目录
                             .disableOpenDir();
                 })
+                .dataSourceConfig(builder ->
+                        builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
+                            int typeCode = metaInfo.getJdbcType().TYPE_CODE;
+                            if (typeCode == Types.TINYINT) {
+                                // 自定义类型转换
+                                return DbColumnType.INTEGER;
+                            }
+                            return typeRegistry.getColumnType(metaInfo);
+                        })
+                )
                 .packageConfig(builder -> {
                     // 设置父包名
                     builder
@@ -73,12 +89,15 @@ public class MybatisPlusGeneratorForAdmin {
 
                     // service配置
                     builder.serviceBuilder()
+                            .superServiceClass(MyBaseService.class)
+                            .superServiceImplClass(MyBaseServiceImpl.class)
                             .enableFileOverride()
                             .formatServiceFileName("%sService")
                             .formatServiceImplFileName("%sServiceImpl");
 
                     // mapper配置
                     builder.mapperBuilder()
+                            .superClass(MyBaseMapper.class)
                             .enableFileOverride()
                             .enableBaseResultMap()
                             .enableBaseColumnList()

+ 19 - 1
miniapp/src/main/java/com/kym/miniapp/utils/MybatisPlusGeneratorForMiniApp.java

@@ -2,9 +2,14 @@ package com.kym.miniapp.utils;
 
 import com.baomidou.mybatisplus.generator.FastAutoGenerator;
 import com.baomidou.mybatisplus.generator.config.OutputFile;
+import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
 import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
 import com.kym.entity.BaseEntity;
+import com.kym.mapper.mybatisplus.MyBaseMapper;
+import com.kym.service.mybatisplus.MyBaseService;
+import com.kym.service.mybatisplus.MyBaseServiceImpl;
 
+import java.sql.Types;
 import java.util.Map;
 
 /**
@@ -22,6 +27,16 @@ public class MybatisPlusGeneratorForMiniApp {
                             // 指定输出目录
                             .disableOpenDir();
                 })
+                .dataSourceConfig(builder ->
+                        builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
+                            int typeCode = metaInfo.getJdbcType().TYPE_CODE;
+                            if (typeCode == Types.TINYINT) {
+                                // 自定义类型转换
+                                return DbColumnType.INTEGER;
+                            }
+                            return typeRegistry.getColumnType(metaInfo);
+                        })
+                )
                 .packageConfig(builder -> {
                     // 设置父包名
                     builder
@@ -63,7 +78,7 @@ public class MybatisPlusGeneratorForMiniApp {
                     // entity配置
                     builder.entityBuilder()
                             .superClass(BaseEntity.class)
-                            .addIgnoreColumns("id","create_time","update_time")
+                            .addIgnoreColumns("id", "create_time", "update_time")
                             .enableFileOverride()
                             .enableLombok();
 
@@ -77,12 +92,15 @@ public class MybatisPlusGeneratorForMiniApp {
 
                     // service配置
                     builder.serviceBuilder()
+                            .superServiceClass(MyBaseService.class)
+                            .superServiceImplClass(MyBaseServiceImpl.class)
                             .enableFileOverride()
                             .formatServiceFileName("%sService")
                             .formatServiceImplFileName("%sServiceImpl");
 
                     // mapper配置
                     builder.mapperBuilder()
+                            .superClass(MyBaseMapper.class)
                             .enableFileOverride()
                             .enableBaseResultMap()
                             .enableBaseColumnList()