org.shoulder.batch.service.impl.ExportSupport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shoulder-batch Show documentation
Shoulder 扩展-批处理模块，提供批量数据导入、导出、异步校验、导入历史记录管理等能力。
There is a newer version: 0.8.1
package org.shoulder.batch.service.impl;

import com.fasterxml.jackson.core.type.TypeReference;
import org.shoulder.batch.model.ExportConfig;
import org.shoulder.core.context.AppContext;
import org.shoulder.core.i18.Translator;
import org.shoulder.core.log.Logger;
import org.shoulder.core.log.LoggerFactory;
import org.shoulder.core.util.ContextUtils;
import org.shoulder.core.util.JsonUtils;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 导出辅助
 * todo 使用 core 包的配置
 *
 * @author lym
 */
public class ExportSupport {

    private static final Logger log = LoggerFactory.getLogger(ExportSupport.class);

    private static final String LOCALIZE_FILE_PATH = "META-INF/export-localize.json";

    /**
     * 数据类型 - 导出配置 - 语言标识, 编码/分隔符映射
     */
    private static final Map LOCALIZE_CACHE = new HashMap<>();

    /**
     * 数据类型 - 导出配置
     * 使用者来调用 {@link #putConfig} 方法填充
     */
    private static final Map CONFIG_CACHE = new HashMap<>();

    static {
        try (InputStream inputStream = ExportSupport.class.getClassLoader().getResourceAsStream(LOCALIZE_FILE_PATH)) {
            List exportLocalizeList = JsonUtils.parseObject(inputStream, new TypeReference<>() {
            });
            for (ExportLocalize exportLocalize : exportLocalizeList) {
                LOCALIZE_CACHE.put(exportLocalize.getLanguageId(), exportLocalize);
            }
        } catch (IOException e) {
            log.warn("csv_localize.json read failed! csv localize disabled.", e);
        }
    }

    public static ExportLocalize getLocalizeByLanguageId(String languageId) {
        return LOCALIZE_CACHE.get(languageId);
    }


    public static ExportConfig getConfig(String csvId) {
        return CONFIG_CACHE.get(csvId);
    }

    public static void putConfig(String csvId, ExportConfig exportConfig) {
        CONFIG_CACHE.put(csvId, exportConfig);
    }


    public static ExportConfig getConfigWithLocale(String id) {
        ExportConfig exportConfig = getConfig(id);
        if (exportConfig == null) {
            return null;
        }
        localizeExportConfig(exportConfig);
        return exportConfig;
    }

    /**
     * 对exportConfig进行本地化
     *
     * @param exportConfig 导出配置
     */
    private static void localizeExportConfig(ExportConfig exportConfig) {
        // 编码 / 分隔符 本地化
        String languageId = AppContext.getLocale().getLanguage();
        ExportLocalize exportLocalize = getLocalizeByLanguageId(languageId);
        if (exportLocalize != null) {
            exportConfig.setEncode(exportLocalize.getEncoding());
            exportConfig.setSeparator(exportLocalize.getDelimiter().charAt(0));
        } else {
            log.info("can't find languageId {} in export-localize.json, fall back to default", languageId);
        }
        // 详情
        List headers = new ArrayList<>();
        for (String header : exportConfig.getHeadersI18n()) {
            headers.add(ContextUtils.getBean(Translator.class).getMessage(header));
        }
        for (ExportConfig.Column column : exportConfig.getColumns()) {
            column.setColumnName(ContextUtils.getBean(Translator.class).getMessage(column.getColumnNameI18n()));
        }
        exportConfig.setHeaders(headers);
    }

    public static class ExportLocalize {
        public ExportLocalize() {
        }

        public ExportLocalize(String languageId, String encoding, String delimiter) {
            this.languageId = languageId;
            this.encoding = encoding;
            this.delimiter = delimiter;
        }

        private String languageId;

        private String encoding;

        private String delimiter;

        public String getLanguageId() {
            return languageId;
        }

        public String getEncoding() {
            return encoding;
        }

        public String getDelimiter() {
            return delimiter;
        }

        public void setLanguageId(String languageId) {
            this.languageId = languageId;
        }

        public void setEncoding(String encoding) {
            this.encoding = encoding;
        }

        public void setDelimiter(String delimiter) {
            this.delimiter = delimiter;
        }
    }

}