Mybatis-Plus 使用操作_1

绝地灬酷狼 2022-12-15 14:20 80阅读 0赞

# 代码生成器 #

pom 文件

<dependency>
                <groupId>com.baomidou</groupId>
                <artifactId>mybatis-plus-boot-starter</artifactId>
                <version>3.4.0</version>
            </dependency>
    
            <dependency>
                <groupId>com.baomidou</groupId>
                <artifactId>mybatis-plus-generator</artifactId>
                <version>3.4.0</version>
            </dependency>
            
            <dependency>
                <groupId>org.apache.velocity</groupId>
                <artifactId>velocity-engine-core</artifactId>
                <version>2.0</version>
            </dependency>

代码

package 
    com.ak.admin;
    
    import com.baomidou.mybatisplus.annotation.FieldFill;
    import com.baomidou.mybatisplus.annotation.IdType;
    import com.baomidou.mybatisplus.generator.AutoGenerator;
    import com.baomidou.mybatisplus.generator.config.*;
    import com.baomidou.mybatisplus.generator.config.po.TableFill;
    import com.baomidou.mybatisplus.generator.config.rules.DateType;
    import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
    import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
    import org.springframework.boot.test.context.SpringBootTest;
    
    import java.util.ArrayList;
    
    @SpringBootTest
    public class MyTest {
    
    
        public static void main(String[] args) {
            // 代码生成器
            AutoGenerator mpg = new AutoGenerator();
    
            // 全局配置
            GlobalConfig gc = new GlobalConfig();
            String projectPath = System.getProperty("user.dir");
            gc.setOutputDir(projectPath + "/src/main/java");
            gc.setAuthor("joker");
            gc.setOpen(false);
            gc.setIdType(IdType.ASSIGN_UUID);
            gc.setDateType(DateType.ONLY_DATE);
            gc.setSwagger2(true);
            mpg.setGlobalConfig(gc);
    
            // 数据源配置
            DataSourceConfig dsc = new DataSourceConfig();
            dsc.setUrl("jdbc:mysql://rm-uf682626j42h0nn6ljo.mysql.rds.aliyuncs.com:3306/taobao?useUnicode=true&characterEncoding=utf-8&useSSL=true&serverTimezone=UTC");
            // dsc.setSchemaName("public");
            dsc.setDriverName("com.mysql.cj.jdbc.Driver");
            dsc.setUsername("joker");
            dsc.setPassword("1654697013wanWAN");
            mpg.setDataSource(dsc);
    
            // 包配置
            PackageConfig pc = new PackageConfig();
            pc.setModuleName("admin");
            pc.setParent("com.ak");
            pc.setMapper("mapper");
            pc.setService("service");
            pc.setEntity("entity");
            pc.setServiceImpl("serviceImpl");
            mpg.setPackageInfo(pc);
    //
    //        // 自定义配置
    //        InjectionConfig cfg = new InjectionConfig() {
    //            @Override
    //            public void initMap() {
    //                // to do nothing
    //            }
    //        };
    //
    //        // 如果模板引擎是 freemarker
    //        String templatePath = "/templates/mapper.xml.ftl";
    //        // 如果模板引擎是 velocity
    //        // String templatePath = "/templates/mapper.xml.vm";
    //
    //        // 自定义输出配置
    //        List<FileOutConfig> focList = new ArrayList<>();
    //        // 自定义配置会被优先输出
    //        focList.add(new FileOutConfig(templatePath) {
    //            @Override
    //            public String outputFile(TableInfo tableInfo) {
    //                // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
    //                return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()
    //                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
    //            }
    //        });
    //
    //        cfg.setFileOutConfigList(focList);
    //        mpg.setCfg(cfg);
    //
    //        // 配置模板
    //        TemplateConfig templateConfig = new TemplateConfig();
    //
    //        // 配置自定义输出模板
    //        //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
    //        // templateConfig.setEntity("templates/entity2.java");
    //        // templateConfig.setService();
    //        // templateConfig.setController();
    //
    //        templateConfig.setXml(null);
    //        mpg.setTemplate(templateConfig);
    
            // 策略配置
            StrategyConfig strategy = new StrategyConfig();
            strategy.setNaming(NamingStrategy.underline_to_camel);
            strategy.setColumnNaming(NamingStrategy.underline_to_camel);
         //   strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
            strategy.setEntityLombokModel(true);
            strategy.setRestControllerStyle(true);
            // 公共父类
          //  strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
            // 写于父类中的公共字段
          //  strategy.setSuperEntityColumns("id");
            strategy.setInclude("comment","goods","orderitem","orders","person","shoppingcart","store");
            strategy.setLogicDeleteFieldName("status");
            TableFill gmt_create=new TableFill("gmt_create", FieldFill.INSERT);
            TableFill gmt_modifty=new TableFill("gmt_modifty", FieldFill.INSERT_UPDATE);
    
            strategy.setControllerMappingHyphenStyle(true);
          //  strategy.setTablePrefix(pc.getModuleName() + "_");
            ArrayList<TableFill> objectArrayList=new ArrayList<>();
            objectArrayList.add(gmt_create);
            objectArrayList.add(gmt_modifty);
            strategy.setTableFillList(objectArrayList);
            strategy.setVersionFieldName("version");
            mpg.setStrategy(strategy);
         //   mpg.setTemplateEngine(new FreemarkerTemplateEngine());
            mpg.execute();
        }
    
    }