com.github.dreamroute.excel.helper.cache.CacheFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of excel-helper Show documentation
excel export base on apache poi 3.17
The newest version!
package com.github.dreamroute.excel.helper.cache;

import com.github.dreamroute.excel.helper.annotation.CellProps;
import com.github.dreamroute.excel.helper.annotation.HeaderProps;
import com.github.dreamroute.excel.helper.util.ClassAssistant;
import com.github.dreamroute.excel.helper.util.HeaderInfo;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.MapUtils;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Row;

import java.lang.reflect.Field;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * use to cache pojo's props
 * 
 * @author [email protected]
 *
 */
public class CacheFactory {

    private CacheFactory() {}

    private static final ConcurrentHashMap, String> SHEET_NAME_MAP = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, List> FIELDS_MAP = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, List> HEADER_VALUES = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, Integer[]> COLUMN_WIDTH = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, CellType[]> CELL_TYPE = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, HeaderProps[]> HEADER_PROPS = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, CellProps[]> CELL_PROPS = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, Map> HEADER_INFO = new ConcurrentHashMap<>();
    private static final ConcurrentHashMap, List> FORMULA_VALUES = new ConcurrentHashMap<>();

    public static String findSheetName(Class cls) {
        return SHEET_NAME_MAP.computeIfAbsent(cls, ClassAssistant::getSheetName);
    }

    public static List findHeaderValues(Class cls) {
        return HEADER_VALUES.computeIfAbsent(cls, ClassAssistant::getHeaderValues);
    }

    public static List findFields(Class cls) {
        return FIELDS_MAP.computeIfAbsent(cls, ClassAssistant::getAllFields);
    }

    public static CellType[] findCellType(Class cls) {
        return CELL_TYPE.computeIfAbsent(cls, ClassAssistant::getCellType);
    }

    public static Integer[] findColumnWidth(Class cls) {
        return COLUMN_WIDTH.computeIfAbsent(cls, ClassAssistant::getColumnWidth);
    }

    public static HeaderProps[] findHeaderProps(Class cls) {
        return HEADER_PROPS.computeIfAbsent(cls, ClassAssistant::getHeaderProps);
    }

    public static CellProps[] findCellProps(Class cls) {
        CellProps[] cps = CELL_PROPS.get(cls);
        if (cps == null || cps.length == 0) {
            cps = ClassAssistant.getCellProps(cls);
            CELL_PROPS.put(cls, cps);
        }
        return cps;
    }

    public static Map findHeaderInfo(Class cls, Row header) {
        Map headerInfo = HEADER_INFO.get(cls);
        if (MapUtils.isEmpty(headerInfo)) {
            headerInfo = ClassAssistant.getHeaderInfo(cls, header);
            HEADER_INFO.put(cls, headerInfo);
        }
        return headerInfo;
    }
    public static List findFormulaValues(Class cls) {
        List formulaValues = FORMULA_VALUES.get(cls);
        if (CollectionUtils.isEmpty(formulaValues)) {
            formulaValues = ClassAssistant.getFormulaValues(cls);
            FORMULA_VALUES.put(cls, formulaValues);
        }
        return formulaValues;
    }

}