io.github.bootystar.mybatisplus.generator.config.ConfigBase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of booty-mybatis-plus-generator Show documentation
The newest version!
package io.github.bootystar.mybatisplus.generator.config;


import com.baomidou.mybatisplus.generator.config.builder.CustomFile;
import com.baomidou.mybatisplus.generator.config.po.TableField;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import lombok.Data;
import lombok.Setter;
import org.apache.ibatis.type.JdbcType;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 配置基类
 * @author booty
 */
@Data
public abstract class ConfigBase implements IConfig {
    private static final Logger log = LoggerFactory.getLogger(ConfigBase.class);

    //--------------常量---------------
    protected final String shift3 = "#";
    protected final String shift4 = "$";
    protected final String shift5 = "%";
    protected final String shift8 = "*";
    protected final String shiftLeft = "{";
    protected final String shiftRight = "}";


    /**
     * 自定义文件
     */
    @Setter
    protected List customFiles;

    @Override
    public Map renderData(TableInfo tableInfo) {
        HashMap data = new HashMap<>();
        // 添加自定义字段
        try {
            for (Field field : this.getClass().getDeclaredFields()) {
                if (unAccessibleFiled(field)) continue;
                String name = field.getName();
                field.setAccessible(true);
                data.put(name, field.get(this));
            }
            for (Field field : ConfigBase.class.getDeclaredFields()) {
                if (unAccessibleFiled(field)) continue;
                String name = field.getName();
                field.setAccessible(true);
                if (!"customFiles".equals(field.getName())) {
                    data.put(name, field.get(this));
                }
            }
        } catch (IllegalAccessException e) {
            log.error("Generate Injection Field Error Please Report to Developer",e);
        }

        // 当前时间
        data.put("nowTime", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));

        // 时间类型列表
        List jdbcTimeTypes =
                Arrays.asList(
                        JdbcType.DATE,
                        JdbcType.TIME,
                        JdbcType.TIMESTAMP,
                        JdbcType.DATETIMEOFFSET,// SQL Server 2008
                        JdbcType.TIME_WITH_TIMEZONE,// JDBC 4.2 JDK8
                        JdbcType.TIMESTAMP_WITH_TIMEZONE // JDBC 4.2 JDK8
                );
        // 对应fields[i].metaInfo.jdbcType
        data.put("jdbcTimeTypes", jdbcTimeTypes);
        // 排序字段
        List fields = tableInfo.getFields();
        List existColumnNames = fields.stream().map(TableField::getColumnName).collect(Collectors.toList());
        if (orderColumnMap != null && orderColumnMap.size() > 0) {
            orderColumnMap.entrySet().stream()
                    .filter(e -> existColumnNames.contains(e.getKey()))
                    .map(e -> String.format("a.`%s`%s", e.getKey(), e.getValue() ? " desc" : ""))
                    .reduce((e1, e2) -> e1 + "," + e2)
                    .ifPresent(e -> data.put("orderBySql", e))
            ;

        }
        return data;
    }

    protected static boolean unAccessibleFiled(Field field) {
        field.setAccessible(true);
        int modifiers = field.getModifiers();
//        if (Modifier.isFinal(modifiers)){
//            return true;
//        }
        if (Modifier.isStatic(modifiers)) {
            return true;
        }
        if (Modifier.isNative(modifiers)) {
            return true;
        }
        return false;
    }


    @Override
    public List getCustomFiles() {
        return this.customFiles;
    }

    @Override
    public boolean getFileOverride() {
        return this.fileOverride;
    }



    //--------------返回结果相关配置---------------

    /**
     * DTO所在包
     */
    protected String DTOPackage = "dto";
    /**
     * VO所在包
     */
    protected String VOPackage = "vo";

    /**
     * 返回结果类所在包
     */
    protected String returnResultClassPackage;

    /**
     * 返回结果是否支持泛型
     */
    protected boolean returnResultGenericType;
    /**
     * 返回结果类
     */
    protected String returnResultClass;

    /**
     * 返回方法
     */
    protected String returnResultMethodName;

    // ------------------controller相关配置----------------

    /**
     * controller是否使用@RequestBody注解
     */
    protected boolean requestBody = true;

    /**
     * 是否添加参数校验
     */
    protected boolean enableValidated = true;
    /**
     * 是否添加跨域注解
     */
    protected boolean enableOrigins;

    /**
     * 所有请求都使用post方法
     */
    protected boolean allPost = false;
    /**
     * 请求基础url
     */
    protected String baseUrl;

    /**
     * 是否覆盖已有文件
     */
    protected boolean fileOverride;

    /**
     * VO是否生成ResultMap
     */
    protected boolean resultMapForVO;
    /**
     * VO是否生成字段注释
     */
    protected boolean fieldAnnotationOnVO;

    /**
     * 新增排除的字段
     */
    protected Collection insertExcludeFields;

    /**
     * 修改排除的字段
     */
    protected Collection updateExcludeFields;

    /**
     * 排序字段map
     * 字段名 -> 是否倒序
     */
    protected Map orderColumnMap;

    /**
     * java api包
     */
    protected String javaApiPackage = "javax";

    /**
     * 新增DTO
     */
    protected boolean generateInsert = true;
    /**
     * 更新DTO
     */
    protected boolean generateUpdate = true;
    /**
     * 生成删除方法
     */
    protected boolean generateDelete = true;
    /**
     * 查询DTO
     */
    protected boolean generateSelect = true;
    /**
     * 导入DTO
     */
    protected boolean generateImport = true;
    /**
     * 导出DTO
     */
    protected boolean generateExport = true;

    /**
     * rest样式
     */
    protected boolean restStyle = true;

    /**
     * 使用vo生出导出方法(不生成额外ExportDTO)
     */
    protected boolean exportOnVO;
    /**
     * 使用vo生出导入方法(不生成额外ImportDTO)
     */
    protected boolean importOnVO;

}