com.javaoffers.batis.modelhelper.utils.TableInfo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-model-common Show documentation
There is a newer version: 3.5.11.12
package com.javaoffers.batis.modelhelper.utils;

import org.springframework.util.Assert;

import java.lang.reflect.Field;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Description: table description
 * @Auther: create by cmj on 2022/5/4 00:00
 */
public class TableInfo {

    /**
     * table name
     */
    private String tableName;

    /**
     * K: table field (col Name)
     * V: Corresponding table field information, obtained by database query
     */
    private Map primaryColNames = new LinkedHashMap<>();

    /**
     * This property is typically used for inserts and updates
     * K: fields of the table (Col Name)
     * V: Corresponding table field information, obtained by database query
     */
    private Map colNames = new LinkedHashMap<>();

    /**
     * Store field information
     */
    private Set columnInfos = new LinkedHashSet<>();

    /**
     * This property is typically used for inserts and updates
     * K: col name of table
     * V: the corresponding table field, generated by the class attribute
     */
    private Map> originalColNameOfModelField = new LinkedHashMap<>();

    /**
     * Record the field information of the model class.
     * fieldName as alias name
     */
    private Map fieldNameAndFields = new HashMap<>();

    /**
     * This property is usually used when querying
     * K: attribute name (also alias)
     * V: the corresponding table field, generated by the class attribute, or sql-function field
     * for Sql-Fun example : key: sex ,  value: IF(sex = 1,'boy','girl').
     * lazy loading storage
     */
    private Map fieldNameMappingcolNameOfModel = new LinkedHashMap<>();

    /**
     * This property is usually used when querying
     * K: colName
     * V: Fields of model class
     * lazy loading storage
     */
    private Map> colNameMappingModelFields = new LinkedHashMap<>();

    /**
     * The lamda information will be filled during the running process.
     * This colNameOfGetter is only provided to Table Helper
     * K: The method name obtained by lamda
     * V: The corresponding table fields are generated by lamda (colName)
     * lazy loading storage
     */
    private Map colNameOfGetter = new ConcurrentHashMap<>();

    /**
     * col name mapping anno info
     */
    private Map colNameAnnoInfos = new HashMap<>();

    /**
     * fieldName mapping anno info
     */
    private Map fieldNameAnnoInfos = new HashMap<>();


    /***************************************************************/
    public String getTableName() {
        return tableName;
    }

    public Map getColNames() {
        return colNames;
    }

    public void putColNames(String colName, ColumnInfo columnInfo) {
        this.colNames.put(colName, columnInfo);
    }

    Set getColumnInfos() {
        return columnInfos;
    }

    public TableInfo(String tableName) {
        this.tableName = tableName;
    }

    public Map getFieldNameColNameOfModel() {
        return fieldNameMappingcolNameOfModel;
    }

    void putFieldNameColNameOfModel(String fieldName, String colName) {
        this.fieldNameMappingcolNameOfModel.put(fieldName, colName);
    }

    Map getMethodNameMappingFieldNameOfGetter() {
        return colNameOfGetter;
    }

    public Map> getColNameAndFieldOfModel() {
        return colNameMappingModelFields;
    }

    public boolean isAutoincrement(String colName) {
        ColumnInfo columnInfo = getColumnInfo(colName);
        return columnInfo.isAutoincrement();
    }

    public Object getDefaultValue(String colName) {
        return getColumnInfo(colName).getDefaultValue();
    }

    private ColumnInfo getColumnInfo(String colName) {
        ColumnInfo columnInfo = this.getColNames().get(colName);
        Assert.isTrue(columnInfo != null, "the " + colName + "col name is empty");
        return columnInfo;
    }

    public Map getPrimaryColNames() {
        return primaryColNames;
    }

    public Map> getOriginalColNameOfModelField() {
        return originalColNameOfModelField;
    }

    public boolean isSqlFun(String colName) {
        ColNameAnnoInfo colNameAnnoInfo = getColNameAnnoInfoNotCreate(colName);
        return colNameAnnoInfo.isSqlFun();
    }

    public boolean colNameIsExcludeColAll(String colName) {
        ColNameAnnoInfo colNameAnnoInfoNotCreate = getColNameAnnoInfoNotCreate(colName);
        return colNameAnnoInfoNotCreate.isExcludeColAll();
    }

    public Map getFieldNameAndField() {
        return this.fieldNameAndFields;
    }

    public boolean fieldNameIsExcludeColAll(String fieldName){
        FieldNameAnnoInfo fieldNameAnnoInfo = getFieldNameAnnoInfoNotCreate(fieldName);
        return fieldNameAnnoInfo.isExcludeColAll();

    }

    private FieldNameAnnoInfo getFieldNameAnnoInfoNotCreate(String fieldName) {
        return fieldNameAnnoInfos.getOrDefault(fieldName, new FieldNameAnnoInfo());
    }

    void putFieldNameExcludeColAll(String fieldName, boolean isExcludeColAll) {
        FieldNameAnnoInfo fieldNameAnnoInfo = getFieldNameAnnoInfoOrCreate(fieldName);
        fieldNameAnnoInfo.setExcludeColAll(isExcludeColAll);
    }

    FieldNameAnnoInfo getFieldNameAnnoInfoOrCreate(String fieldName) {
        FieldNameAnnoInfo fieldNameAnnoInfo = fieldNameAnnoInfos.get(fieldName);
        if(fieldNameAnnoInfo == null){
            fieldNameAnnoInfo = new FieldNameAnnoInfo();
            fieldNameAnnoInfo.setFieldName(fieldName);
        }
        fieldNameAnnoInfos.put(fieldName, fieldNameAnnoInfo);
        return fieldNameAnnoInfo;
    }

    Map> putColNameAndFieldOfModel(String colName, Field field) {
        List fields = colNameMappingModelFields.get(colName);
        if (fields == null) {
            fields = new LinkedList<>();
            colNameMappingModelFields.put(colName, fields);
        }
        fields.add(field);
        return colNameMappingModelFields;
    }

    ColNameAnnoInfo getColNameAnnoInfoOrCreate(String colName) {
        ColNameAnnoInfo colNameAnnoInfo = colNameAnnoInfos.get(colName);
        if (colNameAnnoInfo == null) {
            colNameAnnoInfo = new ColNameAnnoInfo();
            colNameAnnoInfo.setColName(colName);
        }
        colNameAnnoInfos.put(colName, colNameAnnoInfo);
        return colNameAnnoInfo;
    }

    ColNameAnnoInfo getColNameAnnoInfoNotCreate(String colName) {
        return colNameAnnoInfos.getOrDefault(colName, new ColNameAnnoInfo());
    }

    void putPrimaryColNames(String primaryCol, ColumnInfo columnInfo) {
        primaryColNames.put(primaryCol, columnInfo);
    }

    void putOriginalColNameAndFieldOfModelField(String colName, Field field) {
        List fields = this.originalColNameOfModelField.get(colName);
        if (fields == null) {
            fields = new ArrayList<>();
            this.originalColNameOfModelField.put(colName, fields);
        }
        fields.add(field);
    }

    void putSqlFun(String colName, boolean isFunFun) {
        ColNameAnnoInfo colNameAnnoInfo = getColNameAnnoInfoOrCreate(colName);
        colNameAnnoInfo.setSqlFun(isFunFun);
    }

    void putFieldNameAndField(String fieldName, Field field) {
        this.fieldNameAndFields.put(fieldName, field);
    }

    void putColNameExcludeColAll(String colName, boolean isFunGroup) {
        ColNameAnnoInfo colNameAnnoInfo = getColNameAnnoInfoOrCreate(colName);
        colNameAnnoInfo.setExcludeColAll(isFunGroup);
    }

    public void unmodifiable() {
        this.colNames = Collections.unmodifiableMap(colNames);
        this.fieldNameMappingcolNameOfModel = Collections.unmodifiableMap(fieldNameMappingcolNameOfModel);
        this.colNameMappingModelFields = Collections.unmodifiableMap(colNameMappingModelFields);
        this.primaryColNames = Collections.unmodifiableMap(primaryColNames);
        this.originalColNameOfModelField = Collections.unmodifiableMap(originalColNameOfModelField);
        this.fieldNameAndFields = Collections.unmodifiableMap(this.fieldNameAndFields);
    }


}