com.itxiaoer.plugins.mybatis.DomainExtendPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of itxiaoer-mybatis-generator-plugin Show documentation
There is a newer version: 1.1.15
Show newest version
package com.itxiaoer.plugins.mybatis;

import com.itxiaoer.plugins.mybatis.util.JavaDocUtils;
import lombok.extern.slf4j.Slf4j;
import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.java.*;

import java.util.List;
import java.util.Objects;

/**
 * mybatis 代码生成扩展（javadoc）
 *
 * @author : liuyk
 */
@Slf4j
@SuppressWarnings("ALL")
public class DomainExtendPlugin extends PluginAdapter {

    @Override
    public boolean validate(List list) {
        return true;
    }

    @Override
    public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
        topLevelClass.addAnnotation("@SuppressWarnings(\"unused\")");

        List methods = topLevelClass.getMethods();
        for (Method method : methods) {
            JavaDocUtils.addJavaDoc(method);
        }

        List innerClasses = topLevelClass.getInnerClasses();

        for (InnerClass innerClass : innerClasses) {
            topLevelClass.addAnnotation("@SuppressWarnings(\"ALL\")");
        }

        List innerEnums = topLevelClass.getInnerEnums();

        for (InnerEnum innerEnum : innerEnums) {
            innerEnum.addAnnotation("@SuppressWarnings(\"ALL\")");
        }

        String author = this.getProperties().getProperty("author");
        topLevelClass.addJavaDocLine("/**");
        topLevelClass.addJavaDocLine("* " + introspectedTable.getRemarks());
        topLevelClass.addJavaDocLine("* ");
        topLevelClass.addJavaDocLine("* @author : " + (Objects.isNull(author) ? "liuyk" : author));
        topLevelClass.addJavaDocLine("*/");

        return super.modelBaseRecordClassGenerated(topLevelClass, introspectedTable);
    }

    @Override
    public boolean modelFieldGenerated(Field field, TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable, ModelClassType modelClassType) {

        String validate = properties.getProperty("validate");

        if (!Objects.isNull(validate) && !validate.isEmpty()) {

            // 生成校验规则
            String javaProperty = introspectedColumn.getJavaProperty();
            boolean nullable = introspectedColumn.isNullable();
            int length = introspectedColumn.getLength();
            int scale = introspectedColumn.getScale();


            topLevelClass.addImportedType("javax.validation.constraints.Size");
            topLevelClass.addImportedType("javax.validation.constraints.NotBlank");
            topLevelClass.addImportedType("javax.validation.constraints.Digits");
            topLevelClass.addImportedType("javax.validation.constraints.NotEmpty");
            topLevelClass.addImportedType("net.tsingyun.commons.core.Constants");
            topLevelClass.addImportedType("com.fasterxml.jackson.annotation.JsonFormat");

            if (!Objects.equals("id", javaProperty)) {
                // 不能为空


                String jdbcTypeName = introspectedColumn.getJdbcTypeName();
                if (Objects.equals(jdbcTypeName, "BIGINT") || Objects.equals(jdbcTypeName, "INTEGER")) {
                    if (!nullable) {
                        field.addAnnotation("@NotEmpty(message = \"" + introspectedColumn.getRemarks() + "不能为空\")");
                    }
                    field.addAnnotation(" @Digits(integer = " + length + ", fraction = " + scale + ", message = \"" + introspectedColumn.getRemarks() + "只能为数字\")");
                }

                if (Objects.equals("DECIMAL", jdbcTypeName)) {
                    if (!nullable) {
                        field.addAnnotation("@NotEmpty(message = \"" + introspectedColumn.getRemarks() + "不能为空\")");
                    }
                    field.addAnnotation(" @Digits(integer = " + length + ", fraction = " + scale + ", message = \"" + introspectedColumn.getRemarks() + "只能为数字\")");

                }

                if (Objects.equals("VARCHAR", jdbcTypeName)) {
                    if (!nullable) {
                        field.addAnnotation("@NotBlank(message = \"" + introspectedColumn.getRemarks() + "不能为空\")");
                    }
                    field.addAnnotation("@Size(max = " + length + ", message = \"" + introspectedColumn.getRemarks() + "不能超过" + length + "位\")");

                }
                if (Objects.equals("DATE", jdbcTypeName)) {
                    if (!nullable) {
                        field.addAnnotation("@NotEmpty(message = \"" + introspectedColumn.getRemarks() + "不能为空\")");
                    }
                    field.addAnnotation("@JsonFormat(shape = JsonFormat.Shape.STRING, pattern = Constants.DATE_PATTERN, timezone = \"GMT+8\")");
                }

                if (Objects.equals("DATETIME", jdbcTypeName)) {
                    if (!nullable) {
                        field.addAnnotation("@NotEmpty(message = \"" + introspectedColumn.getRemarks() + "不能为空\")");
                    }
                    field.addAnnotation("@JsonFormat(shape = JsonFormat.Shape.STRING, pattern = Constants.DATE_TIME_PATTERN, timezone = \"GMT+8\")");

                }
            }
        }
        return super.modelFieldGenerated(field, topLevelClass, introspectedColumn, introspectedTable, modelClassType);
    }


}