实现spring mvc 的java config 加载不同配置文件

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u014087707/article/details/79053203

需要配置一个spring 启动时的事件监听器.

SpringRootAppInitListener.java

配置类

    /**
     * spring 根容器启动时的监听事件
     *
     * @return ApplicationContextInitializer<?>[]
     */
    @Override
    protected ApplicationContextInitializer<?>[] getRootApplicationContextInitializers() {
        SpringRootAppInitListener springRootAppInitListener = new SpringRootAppInitListener();
        return new ApplicationContextInitializer[]{springRootAppInitListener};
    }

实现方式

public class SpringRootAppInitListener implements ApplicationContextInitializer {

    private static final Logger logger = LoggerFactory.getLogger(SpringRootAppInitListener.class);

    private static final String DEFAULT_CONFIG_PREFIX = "application";

    private static final String DEFAULT_CONFIG_SUFFIX = ".properties";

    private static final String DEFAULT_CONFIG_DELIMITE = "-";

    private static final String DEFAULT_CONFIG_KEY = "spring.active.profile";


    @Override
    public void initialize(ConfigurableApplicationContext applicationContext) {
        MutablePropertySources sources = applicationContext.getEnvironment().getPropertySources();
        //加载默认的配置文件
        PropertiesLoader propertiesLoader = new PropertiesLoader(DEFAULT_CONFIG_PREFIX + DEFAULT_CONFIG_SUFFIX);
        Properties defaultProperties = propertiesLoader.getProperties();
        //获取要加载的配置文件的后缀名称
        String suFix = defaultProperties.getProperty(DEFAULT_CONFIG_KEY);

        //加载配置文件
        Properties properties = null;
        if (!StringUtil.isEmpty(suFix)) {
            properties = propertiesLoader.loadClassPathProperties(DEFAULT_CONFIG_PREFIX + DEFAULT_CONFIG_DELIMITE + suFix + DEFAULT_CONFIG_SUFFIX);
        }
        //两个配置文件合并
        propertiesLoader.mergerProperties(defaultProperties, properties);
        //加载到spring 的配置中去
        sources.addFirst(new PropertiesPropertySource(DEFAULT_CONFIG_PREFIX, properties));
        logger.info("spring root init!");
    }
}

PropertiesLoader.java

public class PropertiesLoader {
    private static Logger logger = LoggerFactory.getLogger(PropertiesLoader.class);

    private static ResourceLoader resourceLoader = new DefaultResourceLoader();

    private final Properties properties;

    public PropertiesLoader(String... resourcesPaths) {
        properties = loadProperties(resourcesPaths);
    }

    public Properties getProperties() {
        return properties;
    }

    /**
     * 取出Property,但以System的Property优先,取不到返回空字符串.
     */
    private String getValue(String key) {
        String systemProperty = System.getProperty(key);
        if (systemProperty != null) {
            return systemProperty;
        }
        if (properties.containsKey(key)) {
            return properties.getProperty(key);
        }
        return "";
    }

    /**
     * 取出String类型的Property,但以System的Property优先,如果都为Null则抛出异常.
     */
    public String getProperty(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return value;
    }

    /**
     * 取出String类型的Property,但以System的Property优先.如果都为Null则返回Default值.
     */
    public String getProperty(String key, String defaultValue) {
        String value = getValue(key);
        return value != null ? value : defaultValue;
    }

    /**
     * 取出Integer类型的Property,但以System的Property优先.如果都为Null或内容错误则抛出异常.
     */
    public Integer getInteger(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return Integer.valueOf(value);
    }

    /**
     * 取出Integer类型的Property,但以System的Property优先.如果都为Null则返回Default值,如果内容错误则抛出异常
     */
    public Integer getInteger(String key, Integer defaultValue) {
        String value = getValue(key);
        return value != null ? Integer.valueOf(value) : defaultValue;
    }

    /**
     * 取出Double类型的Property,但以System的Property优先.如果都为Null或内容错误则抛出异常.
     */
    public Double getDouble(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return Double.valueOf(value);
    }

    /**
     * 取出Double类型的Property,但以System的Property优先.如果都为Null则返回Default值,如果内容错误则抛出异常
     */
    public Double getDouble(String key, Integer defaultValue) {
        String value = getValue(key);
        return value != null ? Double.valueOf(value) : defaultValue;
    }

    /**
     * 取出Boolean类型的Property,但以System的Property优先.如果都为Null抛出异常,如果内容不是true/false则返回false.
     */
    public Boolean getBoolean(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return Boolean.valueOf(value);
    }

    /**
     * 取出Boolean类型的Property,但以System的Property优先.如果都为Null则返回Default值,如果内容不为true/false则返回false.
     */
    public Boolean getBoolean(String key, boolean defaultValue) {
        String value = getValue(key);
        return value != null ? Boolean.valueOf(value) : defaultValue;
    }

    /**
     * 载入多个文件, 文件路径使用Spring Resource格式.
     */
    private Properties loadProperties(String... resourcesPaths) {
        Properties props = new Properties();

        for (String location : resourcesPaths) {

            InputStream is = null;
            try {
                Resource resource = resourceLoader.getResource(location);
                is = resource.getInputStream();
                props.load(is);
            } catch (IOException ex) {
                logger.info("Could not load properties from path:" + location + ", " + ex.getMessage());
            } finally {
                IOUtils.closeQuietly(is);
            }
        }
        return props;
    }

    /**
     * 载入多个文件 classpath 文件
     */
    public Properties loadClassPathProperties(String resourcesPaths) {
        Properties props = new Properties();
        Resource resource = new ClassPathResource(resourcesPaths);
        InputStream is = null;
        try {
            is = resource.getInputStream();
            props.load(is);
        } catch (IOException ex) {
            logger.info("Could not load properties from path:" + resourcesPaths + ", " + ex.getMessage());
        } finally {
            IOUtils.closeQuietly(is);
        }
        return props;
    }

    /**
     * 合并两个配置
     *
     * @param source source
     * @param dist   dist
     */
    public void mergerProperties(Properties source, Properties dist) {
        Enumeration enum1 = source.propertyNames();//得到配置文件的名字
        while (enum1.hasMoreElements()) {
            Object strKey = enum1.nextElement();
            Object value = source.get(strKey);
            dist.put(strKey, value);
        }
    }
}

配置方式

resources
    application.properties
    application-dev.properties
    application-prod.properties
    application-test.properties

配置方式

在application.properties 配置
    spring.active.profile=dev // 激活不同的环境

猜你喜欢

转载自blog.csdn.net/u014087707/article/details/79053203