springboot中使用Mybatis_plus

代码生成器

package com.goodwe.daomain;

import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * 代码生成器
 */
public class Generator {
        /**
         * <p>
         * 读取控制台内容
         * </p>
         */
        public static String scanner(String tip) {
                Scanner scanner = new Scanner(System.in);
                StringBuilder help = new StringBuilder();
                help.append("请输入" + tip + "：");
                System.out.println(help.toString());
                if (scanner.hasNext()) {
                        String ipt = scanner.next();
                        if (null!=ipt&&ipt.length()>0) {
                                return ipt;
                        }
                }
                throw new MybatisPlusException("请输入正确的" + tip + "！");
        }

        public static void main(String[] args) {
                // 代码生成器
                AutoGenerator mpg = new AutoGenerator();

                // 全局配置
                GlobalConfig gc = new GlobalConfig();
                String projectPath = System.getProperty("user.dir");
                gc.setOutputDir(projectPath + "/daomain/src/main/java");
                gc.setAuthor("goodwe");
                gc.setOpen(false);
                gc.setSwagger2(true);
                //gc.setIdType(IdType.AUTO);
                gc.setFileOverride(true);   //覆盖文件
                mpg.setGlobalConfig(gc);

                // 数据源配置
                DataSourceConfig dsc = new DataSourceConfig();
                dsc.setUrl("jdbc:mysql://101.132.67.199:3306/classroom_management");
                dsc.setDriverName("com.mysql.cj.jdbc.Driver");
                dsc.setUsername("root");
                dsc.setPassword("123456");
                mpg.setDataSource(dsc);

                // 包配置
                PackageConfig pc = new PackageConfig();
                pc.setModuleName("plus");
                pc.setParent("com.goodwe.daomain");
                mpg.setPackageInfo(pc);

                // 自定义配置
                InjectionConfig cfg = new InjectionConfig() {
                        @Override
                        public void initMap() {
                                // to do nothing
                        }
                };
                List<FileOutConfig> focList = new ArrayList<>();
                focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {
                        @Override
                        public String outputFile(TableInfo tableInfo) {
                                // 自定义输入文件名称
                                return projectPath + "/daomain/src/main/resources/mapper/" + pc.getModuleName()
                                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
                        }
                });
                cfg.setFileOutConfigList(focList);
                mpg.setCfg(cfg);
                mpg.setTemplate(new TemplateConfig().setXml(null).setController(null));

                // 策略配置
                StrategyConfig strategy = new StrategyConfig();
                strategy.setNaming(NamingStrategy.underline_to_camel);
                strategy.setColumnNaming(NamingStrategy.underline_to_camel);
                strategy.setInclude(scanner("表名"));
                strategy.setControllerMappingHyphenStyle(true);
               // strategy.setTablePrefix("data_"); //去除表前缀
                mpg.setStrategy(strategy);
                mpg.setTemplateEngine(new FreemarkerTemplateEngine());
                mpg.execute();
        }
}

 

依赖文件:

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>com.goodwe</groupId>
        <artifactId>finaltest</artifactId>
        <version>0.0.1-SNAPSHOT</version>
    </parent>
    <groupId>com.goodwe</groupId>
    <artifactId>daomain</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>daomain</name>
    <description>Demo project for Spring Boot</description>

    <properties>
        <java.version>1.8</java.version>
    </properties>

    <dependencies>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.15</version>
        </dependency>
        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>4.5.0</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-core</artifactId>
            <version>3.1.0</version>
            <scope>compile</scope>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.1.0</version>
            <scope>compile</scope>
        </dependency>
        <dependency>
            <groupId>org.testng</groupId>
            <artifactId>testng</artifactId>
            <version>RELEASE</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.freemarker</groupId>
            <artifactId>freemarker</artifactId>
            <version>2.3.28</version>
        </dependency>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.12</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-test</artifactId>
            <version>5.1.7.RELEASE</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-test</artifactId>
            <scope>test</scope>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>

</project>

转载于:https://my.oschina.net/u/4100033/blog/3041258