cn.hiboot.mcn.autoconfigure.config.ConfigProperties Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mcn-boot-autoconfigure Show documentation
MCN Boot Autoconfigure
There is a newer version: 3.3.1
Show newest version
package cn.hiboot.mcn.autoconfigure.config;

import cn.hiboot.mcn.autoconfigure.web.exception.error.ErrorPageController;
import cn.hiboot.mcn.core.util.McnUtils;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.support.PropertiesLoaderUtils;
import org.springframework.util.StreamUtils;
import org.springframework.web.util.HtmlUtils;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.Properties;

/**
 * 定位mcn默认配置所在位置
 *
 * @author DingHao
 * @since 2020/7/22 23:57
 */
public abstract class ConfigProperties {
    private static final String DATA_SOURCE = "DataSource";

    public static final String APP_BASE_PACKAGE = "app.base-package";

    public static final String MULTIPLE_DATASOURCE_PREFIX = "multiple.datasource";
    public static final String JPA_MULTIPLE_DATASOURCE_PREFIX = "jpa." + MULTIPLE_DATASOURCE_PREFIX;
    public static final String MYBATIS_MULTIPLE_DATASOURCE_PREFIX = "mybatis." + MULTIPLE_DATASOURCE_PREFIX;
    public static final String DYNAMIC_DATASOURCE_PREFIX = "dynamic.datasource";
    public static final String DAO_PACKAGE_NAME = "dao.package.name";
    public static final String BASE_PACKAGE_NAME = "base.package.name";


    private static String error_view;

    static {
        try {
            error_view = StreamUtils.copyToString(createResource("defaultErrorView.html", ErrorPageController.class).getInputStream(), StandardCharsets.UTF_8);
        } catch (IOException e) {
            //ignore
        }
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    public static Map loadConfig(ClassLoader classLoader,String location) {
        if (classLoader == null) {
            classLoader = ConfigProperties.class.getClassLoader();
        }
        Properties properties = new Properties();
        try {
            properties = PropertiesLoaderUtils.loadAllProperties(location, classLoader);
        } catch (IOException e) {
            //ignore
        }
        return (Map) properties;
    }

    public static ClassPathResource mcnDefault() {
        return createResource("mcn-default.properties", ConfigProperties.class);
    }

    public static ClassPathResource createResource(String file,Class clazz) {
        return new ClassPathResource(file, clazz);
    }

    public static String errorView(Map error, String basePath) {
        String status = error.get("status").toString();
        Object message = error.get("message");
        if(McnUtils.isNullOrEmpty(message)){
            message = error.get("error");
        }
        String msg = message == null ? "" : message.toString();
        return error_view.replace("{status}",htmlEscape(status)).replace("{msg}",htmlEscape(msg));
    }

    private static String htmlEscape(Object input) {
        return (input != null) ? HtmlUtils.htmlEscape(input.toString()) : null;
    }

    public static String getDataSourceBeanName(String dsName){
        return dsName + DATA_SOURCE;
    }
}