site.sorghum.anno.db.DbTableContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of anno-admin Show documentation
The newest version!
package site.sorghum.anno.db;

import jakarta.inject.Inject;
import jakarta.inject.Named;
import site.sorghum.anno._metadata.AnEntity;
import site.sorghum.anno._metadata.AnField;
import site.sorghum.anno._metadata.EntityMetadataLoader;
import site.sorghum.anno._metadata.MetadataContext;
import site.sorghum.anno.anno.annotation.clazz.AnnoRemoveImpl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * Anno Clazz 缓存
 *
 * @author sorghum
 * @since 2023/05/20
 */
@Named
public class DbTableContext implements MetadataContext {

    @Inject
    private EntityMetadataLoader entityMetadataLoader;

    /**
     * Anno Clazz 缓存
     */
    private final Map> tableParamCache = new HashMap<>();

    /**
     * 获取缓存
     *
     * @param entityName 对象名
     * @return {@link Class}<{@link ?}>
     */
    public  TableParam getTableParam(String entityName) {
        TableParam tableParam = tableParamCache.get(entityName);
        // 复制一份，防止被修改
        TableParam returnParam = new TableParam<>();
        returnParam.setTableName(tableParam.getTableName());
        returnParam.setClazz(tableParam.getClazz());
        returnParam.setColumns(new ArrayList<>(tableParam.getColumns()));
        returnParam.setDbRemove(tableParam.getDbRemove());
        returnParam.setVirtualTable(tableParam.isVirtualTable());
        returnParam.setJoinTables(new ArrayList<>(tableParam.getJoinTables()));
        returnParam.setDbName(tableParam.getDbName());
        return returnParam;
    }

    /**
     * 获取 entity db 数据
     *
     * @param clazz 实体类
     */
    public  TableParam getTableParam(Class clazz) {
        String entityName = entityMetadataLoader.getEntityName(clazz);
        return getTableParam(entityName);
    }

    /**
     * 获取原始信息
     *
     * @param entityName 对象名
     * @return {@link Class}<{@link ?}>
     */
    public  TableParam getOriginalTableParam(String entityName) {
        return (TableParam) tableParamCache.get(entityName);
    }

    /**
     * 获取 entity db 原始信息
     *
     * @param clazz 实体类
     */
    public  TableParam getOriginalTableParam(Class clazz) {
        String entityName = entityMetadataLoader.getEntityName(clazz);
        return getTableParam(entityName);
    }

    @Override
    public void refresh(List allEntities) {
        tableParamCache.clear();

        for (AnEntity entity : allEntities) {
            TableParam tableParam = new TableParam<>();
            tableParam.setClazz(entity.getThisClass());
            tableParam.setTableName(entity.getTableName());
            List columns = entity.getDbAnFields().stream().map(AnField::getTableFieldName).collect(Collectors.toList());
            tableParam.setColumns(columns);
            AnnoRemoveImpl annoRemove = entity.getAnnoRemove();
            if (annoRemove.getRemoveType() == 0) {
                tableParam.setDbRemove(new DbRemove());
            } else {
                tableParam.setDbRemove(new DbRemove(true, annoRemove.getRemoveField(), annoRemove.getRemoveValue(), annoRemove.getNotRemoveValue()));
            }
            // 设置是否虚拟表
            tableParam.setVirtualTable(entity.isVirtualTable());
            tableParam.setDbName(entity.getDbName());
            tableParamCache.put(entity.getEntityName(), tableParam);
        }
    }
}