com.jeesuite.confcenter.springboot.CCPropertySourceLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-config-client Show documentation
There is a newer version: 1.3.8
Show newest version
package com.jeesuite.confcenter.springboot;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.boot.env.PropertySourceLoader;
import org.springframework.core.env.PropertiesPropertySource;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PropertiesLoaderUtils;

import com.jeesuite.confcenter.ConfigcenterContext;

public class CCPropertySourceLoader implements PropertySourceLoader, DisposableBean {

	private final static Logger logger = LoggerFactory.getLogger("com.jeesuite");

	private ConfigcenterContext ccContext = ConfigcenterContext.getInstance();
	private String profiles = null;

	@Override
	public String[] getFileExtensions() {
		return new String[] { "properties" };
	}

	public PropertySource load(String name, Resource resource, String profile) throws IOException {

		logger.info("load PropertySource -> name:{},profile:{}", name, profile);
		if (profile == null) {
			Properties properties = PropertiesLoaderUtils.loadProperties(resource);
			if (profiles == null) {
				profiles = properties.getProperty("spring.profiles.active");
			} else {
				logger.info("spring.profiles.active = " + profiles + ",ignore load remote config");
			}
			// 如果指定了profile，则也不加载远程配置
			if (profiles == null && ccContext.getStatus() == null) {
				ccContext.init(properties, true);
				ccContext.mergeRemoteProperties(properties);
				ccContext.syncConfigToServer(properties, true);
			}

			if (!properties.isEmpty()) {
				return new PropertiesPropertySource(name, properties);
			}
		}
		return null;
	}
	
	public List> load(String name, Resource resource) throws IOException {
		Properties properties =  PropertiesLoaderUtils.loadProperties(resource);
		if (profiles == null) {
			profiles = properties.getProperty("spring.profiles.active");
		} else {
			logger.info("spring.profiles.active = " + profiles + ",ignore load remote config");
		}
		// 如果指定了profile，则也不加载远程配置
		if (profiles == null && ccContext.getStatus() == null) {
			ccContext.init(properties, true);
			ccContext.mergeRemoteProperties(properties);
			ccContext.syncConfigToServer(properties, true);
			PropertySource props = new PropertiesPropertySource(ConfigcenterContext.MANAGER_PROPERTY_SOURCE, properties);
			return Arrays.asList(props);
		}
		
		return new ArrayList<>();
	}

	@Override
	public void destroy() throws Exception {
		ccContext.close();
	}

}