Mybatis一键生成类文件

太过爱你忘了你带给我的痛 2022-10-09 14:26 132阅读 0赞

package com.zyq.util;
    
    import com.baomidou.mybatisplus.generator.AutoGenerator;
    import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
    import com.baomidou.mybatisplus.generator.config.GlobalConfig;
    import com.baomidou.mybatisplus.generator.config.PackageConfig;
    import com.baomidou.mybatisplus.generator.config.StrategyConfig;
    import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
    
    /**
     * 这是一键生成对应controller, mapper, service, serviceImpl 的类<br>
     * 
     * @author ZhangYuanqiang
     * @since 2021年4月13日
     */
    public class CreateClassUtil {
    
        public static final String table_name = "folder";
    
        public static void main(String[] args) {
            AutoGenerator mpg = new AutoGenerator();
            // 1、全局配置
            GlobalConfig gc = new GlobalConfig();
            // 获取项目路径
            String projectPath = System.getProperty("user.dir");
            // 生成路径(一般都是生成在此项目的src/main/java下面)
            gc.setOutputDir(projectPath + "/src/main/java");
            gc.setAuthor("ZhangYuanqiang"); // 设置作者
            gc.setOpen(false);
            // 第二次生成会把第一次生成的覆盖掉
            gc.setFileOverride(false);
            // 生成的service接口名字首字母是否为I，这样设置就没有
            gc.setServiceName("%sService");
            // 生成resultMap
            gc.setBaseResultMap(true);
            mpg.setGlobalConfig(gc);
    
            // 2、数据源配置
            DataSourceConfig dsc = new DataSourceConfig();
            dsc.setUrl("jdbc:mysql://127.0.0.1:3306/demo");
            dsc.setDriverName("com.mysql.jdbc.Driver");
            dsc.setUsername("root");
            dsc.setPassword("abc123");
            mpg.setDataSource(dsc);
    
            // 3、包配置
            PackageConfig pc = new PackageConfig();
            pc.setModuleName("");
            pc.setParent("com.zyq");
            mpg.setPackageInfo(pc);
    
            // 4、策略配置
            StrategyConfig strategy = new StrategyConfig();
            strategy.setNaming(NamingStrategy.underline_to_camel);
            strategy.setColumnNaming(NamingStrategy.underline_to_camel);
            strategy.setSuperControllerClass("com.zyq.controller.BaseController");
            strategy.setSuperEntityClass("com.zyq.bean.BaseBean");
            // 表名前缀
            // strategy.setTablePrefix("t_");
            // 使用lombok
            strategy.setEntityLombokModel(true);
            // 逆向工程使用的表 如果要生成多个,这里可以传入String[]
            strategy.setInclude(table_name);
            mpg.setStrategy(strategy);
    
            // 5、执行
            mpg.execute();
            System.out.println("执行完成！！");
        }
    
    }