package com.dg.core.db.manual.mapper.util;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import org.springframework.beans.factory.annotation.Value;

import java.io.File;
import java.util.Collections;

public class MapperGenerator {
    @Value("spring.datasource.url")
    String url = "jdbc:mysql://10.8.0.74:3306/huacheng_smart_life?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&allowMultiQueries=true&serverTimezone=GMT%2B8";
    @Value("spring.datasource.username")
    String username = "root";
    @Value("spring.datasource.password")
    String password = "nahanMysqlPwd&&RemoteRemote2019nahan";

    public void gen() {
        String userdir = System.getProperty("user.dir");//获取当前工作目录
        String outputDir = userdir + File.separator + "src" + File.separator + "main";

        FastAutoGenerator.create(url, username, password)
                .globalConfig(builder -> {
                    builder.author("fengjin") // 设置作者
//                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir(outputDir + File.separator + "java"); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.dg.core.db.gen") // 设置父包名
//                            .moduleName("system") // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, outputDir +
                                    File.separator + "resources" + File.separator + "mapper")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
//                    builder.addInclude("");
                    builder.addInclude("automessage_smart_consulting") // 设置需要生成的表名
                            .addTablePrefix("t_", "c_"); // 设置过滤表前缀
                })
                .templateConfig(builder -> {
                    builder.controller("");
                    builder.service("");
                    builder.serviceImpl("");
                })
                .execute();
    }

    public static void main(String[] args) {
        MapperGenerator m = new MapperGenerator();
        m.gen();
    }
}