疯狂的狮子li
2021-03-15 9b4b9e0c744f43417a3c8b741f35090ab4e3c68f
ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTable.java
@@ -1,12 +1,13 @@
package com.ruoyi.generator.domain;
import java.util.List;
import javax.validation.Valid;
import javax.validation.constraints.NotBlank;
import org.apache.commons.lang3.ArrayUtils;
import cn.hutool.core.util.StrUtil;
import com.ruoyi.common.constant.GenConstants;
import com.ruoyi.common.core.domain.BaseEntity;
import com.ruoyi.common.utils.StringUtils;
import org.apache.commons.lang3.ArrayUtils;
import javax.validation.Valid;
import javax.validation.constraints.NotBlank;
import java.util.List;
/**
 * 业务表 gen_table
@@ -28,11 +29,17 @@
    @NotBlank(message = "表描述不能为空")
    private String tableComment;
    /** 关联父表的表名 */
    private String subTableName;
    /** 本表关联父表的外键名 */
    private String subTableFkName;
    /** 实体类名称(首字母大写) */
    @NotBlank(message = "实体类名称不能为空")
    private String className;
    /** 使用的模板(crud单表操作 tree树表操作) */
    /** 使用的模板(crud单表操作 tree树表操作 sub主子表操作) */
    private String tplCategory;
    /** 生成包路径 */
@@ -63,6 +70,9 @@
    /** 主键信息 */
    private GenTableColumn pkColumn;
    /** 子表信息 */
    private GenTable subTable;
    /** 表列信息 */
    @Valid
@@ -114,6 +124,26 @@
    public void setTableComment(String tableComment)
    {
        this.tableComment = tableComment;
    }
    public String getSubTableName()
    {
        return subTableName;
    }
    public void setSubTableName(String subTableName)
    {
        this.subTableName = subTableName;
    }
    public String getSubTableFkName()
    {
        return subTableFkName;
    }
    public void setSubTableFkName(String subTableFkName)
    {
        this.subTableFkName = subTableFkName;
    }
    public String getClassName()
@@ -216,6 +246,16 @@
        this.pkColumn = pkColumn;
    }
    public GenTable getSubTable()
    {
        return subTable;
    }
    public void setSubTable(GenTable subTable)
    {
        this.subTable = subTable;
    }
    public List<GenTableColumn> getColumns()
    {
        return columns;
@@ -286,6 +326,16 @@
        this.parentMenuName = parentMenuName;
    }
    public boolean isSub()
    {
        return isSub(this.tplCategory);
    }
    public static boolean isSub(String tplCategory)
    {
        return tplCategory != null && StrUtil.equals(GenConstants.TPL_SUB, tplCategory);
    }
    public boolean isTree()
    {
        return isTree(this.tplCategory);
@@ -293,7 +343,7 @@
    public static boolean isTree(String tplCategory)
    {
        return tplCategory != null && StringUtils.equals(GenConstants.TPL_TREE, tplCategory);
        return tplCategory != null && StrUtil.equals(GenConstants.TPL_TREE, tplCategory);
    }
    public boolean isCrud()
@@ -303,7 +353,7 @@
    public static boolean isCrud(String tplCategory)
    {
        return tplCategory != null && StringUtils.equals(GenConstants.TPL_CRUD, tplCategory);
        return tplCategory != null && StrUtil.equals(GenConstants.TPL_CRUD, tplCategory);
    }
    public boolean isSuperColumn(String javaField)
@@ -315,9 +365,9 @@
    {
        if (isTree(tplCategory))
        {
            return StringUtils.equalsAnyIgnoreCase(javaField,
            return StrUtil.equalsAnyIgnoreCase(javaField,
                    ArrayUtils.addAll(GenConstants.TREE_ENTITY, GenConstants.BASE_ENTITY));
        }
        return StringUtils.equalsAnyIgnoreCase(javaField, GenConstants.BASE_ENTITY);
        return StrUtil.equalsAnyIgnoreCase(javaField, GenConstants.BASE_ENTITY);
    }
}