From 780245f8c26611e567d47db102fc5728aa179772 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=BD=A9=E8=BE=95=E9=BE=99=E5=84=BF?= Date: Wed, 13 Mar 2024 14:03:06 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E7=94=9F=E6=88=90=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/huangge1199/blog/config/Codegen.java | 75 +++++++++++++++++++ 1 file changed, 75 insertions(+) create mode 100644 src/main/java/com/huangge1199/blog/config/Codegen.java diff --git a/src/main/java/com/huangge1199/blog/config/Codegen.java b/src/main/java/com/huangge1199/blog/config/Codegen.java new file mode 100644 index 0000000..e57b604 --- /dev/null +++ b/src/main/java/com/huangge1199/blog/config/Codegen.java @@ -0,0 +1,75 @@ +package com.huangge1199.blog.config; + +import com.mybatisflex.codegen.Generator; +import com.mybatisflex.codegen.config.GlobalConfig; +import com.mybatisflex.core.service.IService; +import com.mybatisflex.spring.service.impl.ServiceImpl; +import com.zaxxer.hikari.HikariDataSource; + +/** + * @author hyy + * @Classname Codegen + * @Description 代码生成器 + * @Date 2024/3/13 13:31:21 + */ +public class Codegen { + + public static void main(String[] args) { + // 配置数据源 + HikariDataSource dataSource = new HikariDataSource(); + dataSource.setJdbcUrl("jdbc:mysql://192.168.0.197:3306/blog?characterEncoding=utf-8"); + dataSource.setUsername("root"); + dataSource.setPassword("huangge1199"); + + // 创建配置内容,两种风格都可以。 + GlobalConfig globalConfig = createGlobalConfigUseStyle(); + + // 通过 datasource 和 globalConfig 创建代码生成器 + Generator generator = new Generator(dataSource, globalConfig); + + // 生成代码 + generator.generate(); + } + + public static GlobalConfig createGlobalConfigUseStyle() { + // 创建配置内容 + GlobalConfig globalConfig = new GlobalConfig(); + + // 设置根包 + globalConfig.getPackageConfig() + .setBasePackage("com.huangge1199.blog"); + + // 设置表前缀和只生成哪些表,setGenerateTable 未配置时,生成所有表 + globalConfig.getStrategyConfig() + .setTablePrefix("tb_"); + + // 设置生成 entity 并启用 Lombok + globalConfig.enableEntity() + .setWithLombok(true) + .setJdkVersion(17); + + // 设置生成 mapper + globalConfig.enableMapper() + .setClassSuffix("Mapper"); + + // 设置生成 service + globalConfig.getServiceConfig() + .setClassSuffix("Service") + .setSuperClass(IService.class); + + // 设置生成 serviceImpl + globalConfig.getServiceImplConfig() + .setClassSuffix("ServiceImpl") + .setSuperClass(ServiceImpl.class); + + // 设置生成 controller + globalConfig.getControllerConfig() + .setClassSuffix("Controller"); + + // 设置生成 mapperXml + globalConfig.getMapperXmlConfig() + .setFileSuffix("Mapper"); + + return globalConfig; + } +}