Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.jeesuite.confcenter.ConfigcenterContext Maven / Gradle / Ivy
package com.jeesuite.confcenter;
import java.security.PrivateKey;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Objects;
import java.util.Properties;
import java.util.Set;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.Validate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.MutablePropertySources;
import org.springframework.core.env.PropertiesPropertySource;
import org.springframework.core.env.StandardEnvironment;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import com.jeesuite.common.crypt.RSA;
import com.jeesuite.common.http.HttpRequestEntity;
import com.jeesuite.common.http.HttpResponseEntity;
import com.jeesuite.common.http.HttpUtils;
import com.jeesuite.common.json.JsonUtils;
import com.jeesuite.common.util.NodeNameHolder;
import com.jeesuite.common.util.ResourceUtils;
import com.jeesuite.common.util.SimpleCryptUtils;
import com.jeesuite.confcenter.listener.HttpConfigChangeListener;
import com.jeesuite.confcenter.listener.ZkConfigChangeListener;
import com.jeesuite.spring.InstanceFactory;
import com.jeesuite.spring.helper.EnvironmentHelper;
public class ConfigcenterContext {
private final static Logger logger = LoggerFactory.getLogger("com.jeesuite");
private static ConfigcenterContext instance = new ConfigcenterContext();
public static final String MANAGER_PROPERTY_SOURCE = "configcenter";
private PrivateKey rsaPrivateKey;
private static final String OVERRIDE_PLACEHOLER = "[Override]";
private static final String DES_PREFIX = "{Cipher}";
private static final String RSA_PREFIX = "{Cipher:RSA}";
private static final String PLACEHOLDER_PREFIX = "${";
private static final String PLACEHOLDER_SUFFIX = "}";
private Boolean remoteEnabled;
private final String nodeId = NodeNameHolder.getNodeId();
private String[] apiBaseUrls;
private String app;
private String env;
private String version;
private String secret;
private boolean remoteFirst = false;
private boolean isSpringboot;
private int syncIntervalSeconds = 90;
private String pingUri = "/api/ping";
private ConfigChangeListener configChangeListener;
private List configChangeHanlders;
private ConfigStatus status;
public enum ConfigStatus{
INITED,FETCHED,UPLOAED
}
private ConfigcenterContext() {}
public synchronized void init(Properties properties,boolean isSpringboot) {
logger.info("client.nodeId:",nodeId);
ResourceUtils.merge(properties);
System.setProperty("client.nodeId", nodeId);
System.setProperty("springboot", String.valueOf(isSpringboot));
this.isSpringboot = isSpringboot;
String defaultAppName = getValue("spring.application.name");
app = getValue("jeesuite.configcenter.appName",defaultAppName);
if(remoteEnabled == null)remoteEnabled = Boolean.parseBoolean(getValue("jeesuite.configcenter.enabled","true"));
if(!remoteEnabled)return;
env = getValue("jeesuite.configcenter.profile","dev");
Validate.notBlank(env,"[jeesuite.configcenter.profile] is required");
setApiBaseUrl(getValue("jeesuite.configcenter.base.url"));
version = getValue("jeesuite.configcenter.version","0.0.0");
syncIntervalSeconds = ResourceUtils.getInt("jeesuite.configcenter.sync-interval-seconds", 90);
System.out.println(String.format("\n=====Configcenter config=====\nappName:%s\nenv:%s\nversion:%s\nremoteEnabled:%s\napiBaseUrls:%s\n=====Configcenter config=====", app,env,version,remoteEnabled,JsonUtils.toJson(apiBaseUrls)));
initRSAPrivateKey(null);
status = ConfigStatus.INITED;
}
private void initRSAPrivateKey(Properties properties) {
if(rsaPrivateKey != null)return;
String location = getValue("jeesuite.configcenter.encrypt-keyStore-location");
String storeType = getValue("jeesuite.configcenter.encrypt-keyStore-type", "JCEKS");
String storePass = getValue("jeesuite.configcenter.encrypt-keyStore-password");
String alias = getValue("jeesuite.configcenter.encrypt-keyStore-alias");
String keyPass = getValue("jeesuite.configcenter.encrypt-keyStore-keyPassword", storePass);
if(properties != null){
location = properties.getProperty("jeesuite.configcenter.encrypt-keyStore-location",location);
storeType = properties.getProperty("jeesuite.configcenter.encrypt-keyStore-type",storeType);
storePass = properties.getProperty("jeesuite.configcenter.encrypt-keyStore-password",storePass);
alias = properties.getProperty("jeesuite.configcenter.encrypt-keyStore-alias",alias);
keyPass = properties.getProperty("jeesuite.configcenter.encrypt-keyStore-keyPassword",keyPass);
}
//pass 支持DES加密
if(storePass != null)storePass = decodeEncryptIfRequire(storePass).toString();
if(keyPass != null)keyPass = decodeEncryptIfRequire(keyPass).toString();
System.out.println(String.format("\n=====RSA config=====\nlocation:%s\ntype:%s\npassword:%s\nalias:%s\nkeyPassword:%s\n=====RSA config=====",
location,storeType,hideSensitive("storePass", storePass),alias,hideSensitive("keyPass", keyPass)));
if(StringUtils.isAnyBlank(location,storePass,alias,keyPass))return;
try {
System.out.println("begin to init RSA private key...");
if(location.toLowerCase().startsWith("classpath")){
Resource resource = new ClassPathResource(location.substring(location.indexOf(":") + 1));
location = resource.getFile().getAbsolutePath();
}
rsaPrivateKey = RSA.loadPrivateKeyFromKeyStore(location, alias, storeType, storePass, keyPass);
System.out.println("init RSA private key OK!");
} catch (Exception e) {
System.err.println("load RSA private key error,location:"+location + ",error:" + e.getMessage());
}
}
public static ConfigcenterContext getInstance() {
return instance;
}
public boolean isRemoteEnabled() {
return remoteEnabled;
}
public void setRemoteEnabled(boolean remoteEnabled) {
this.remoteEnabled = remoteEnabled;
}
public String[] getApiBaseUrls() {
return apiBaseUrls;
}
public void setApiBaseUrl(String apiBaseUrl) {
Validate.notBlank(apiBaseUrl,"[jeesuite.configcenter.base.url] is required");
String[] urls = apiBaseUrl.split(",|;");
this.apiBaseUrls = new String[urls.length];
for (int i = 0; i < urls.length; i++) {
if(urls[i].endsWith("/")){
this.apiBaseUrls[i] = urls[i].substring(0, urls[i].length() - 1);
}else{
this.apiBaseUrls[i] = urls[i];
}
}
}
public String getApp() {
return app;
}
public String getEnv() {
return env;
}
public String getVersion() {
return version;
}
public String getSecret() {
return secret;
}
public boolean isRemoteFirst() {
return remoteFirst;
}
public int getSyncIntervalSeconds() {
return syncIntervalSeconds;
}
public String getNodeId() {
return nodeId;
}
public boolean isSpringboot() {
return isSpringboot;
}
public ConfigStatus getStatus() {
return status;
}
public void mergeRemoteProperties(Properties properties){
Properties remoteProperties = getAllRemoteProperties();
if(remoteProperties != null){
//合并属性
Set> entrySet = remoteProperties.entrySet();
for (Entry entry : entrySet) {
//本地配置优先
if(isRemoteFirst() == false && properties.containsKey(entry.getKey())){
logger.info("config[{}] exists in location,skip~",entry.getKey());
continue;
}
String value = entry.getValue().toString();
if(OVERRIDE_PLACEHOLER.equals(value))continue;
properties.setProperty(entry.getKey().toString(), value);
}
}
//替换本地变量占位符
Set> entrySet = properties.entrySet();
for (Entry entry : entrySet) {
String key = entry.getKey().toString();
String value = entry.getValue().toString();
if(value.contains(PLACEHOLDER_PREFIX)){
value = setReplaceHolderRefValue(properties,key,value);
properties.setProperty(key, value);
}
ResourceUtils.add(key, value);
}
}
private Properties getAllRemoteProperties(){
if(!remoteEnabled)return null;
Properties properties = new Properties();
Map map = fetchConfigFromServer(2);
if(map == null){
throw new RuntimeException("fetch remote config error!");
}
//DES解密密匙
secret = Objects.toString(map.remove("jeesuite.configcenter.encrypt-secret"),null);
remoteFirst = Boolean.parseBoolean(Objects.toString(map.remove("jeesuite.configcenter.remote-config-first"),"false"));
//如果rsa配置在配置中心，先初始化rsa私钥
properties.putAll(map);
initRSAPrivateKey(properties);
properties.clear();
Set keys = map.keySet();
for (String key : keys) {
Object value = decodeEncryptIfRequire(map.get(key));
properties.put(key, value);
}
status = ConfigStatus.FETCHED;
return properties;
}
@SuppressWarnings("unchecked")
private Map fetchConfigFromServer(int retry){
if(retry == 0)return null;
Map result = null;
String errorMsg = null;
for (String apiBaseUrl : apiBaseUrls) {
String url = String.format("%s/api/fetch_all_configs?appName=%s&env=%s&version=%s", apiBaseUrl,app,env,version);
System.out.println("fetch configs url:" + url);
String jsonString = null;
HttpResponseEntity response = HttpUtils.get(url);
if(response.isSuccessed()){
jsonString = response.getBody();
result = JsonUtils.toObject(jsonString, Map.class);
if(result.containsKey("code")){
errorMsg = result.get("msg").toString();
System.out.println("fetch error:"+errorMsg);
result = null;
}else{
break;
}
}
}
if(result == null){
result = fetchConfigFromServer(--retry);
}
return result;
}
public void syncConfigToServer(Properties properties,boolean first){
if(!remoteEnabled)return;
if(status.equals(ConfigStatus.INITED))return;
String syncType = properties.getProperty("jeesuite.configcenter.sync-type");
List sortKeys = new ArrayList<>();
Map params = new HashMap<>();
params.put("nodeId", nodeId);
params.put("appName", app);
params.put("env", env);
params.put("version", version);
params.put("springboot", String.valueOf(isSpringboot));
params.put("syncIntervalSeconds", String.valueOf(syncIntervalSeconds));
params.put("syncType", syncType);
String serverPort = ServerEnvUtils.getServerPort();
if(StringUtils.isNumeric(serverPort)){
params.put("serverport", serverPort);
}
params.put("serverip", ServerEnvUtils.getServerIpAddr());
Set> entrySet = properties.entrySet();
for (Entry entry : entrySet) {
String key = entry.getKey().toString();
String value = entry.getValue().toString();
params.put(key, hideSensitive(key, value));
sortKeys.add(key);
}
if(first){
Collections.sort(sortKeys);
System.out.println("==================final config list start==================");
for (String key : sortKeys) {
System.out.println(String.format("%s = %s", key,params.get(key) ));
}
System.out.println("==================final config list end====================");
//register listener
registerListener(syncType);
}else{
String serverip = EnvironmentHelper.getProperty("spring.cloud.client.ipAddress");
if(StringUtils.isNotBlank(serverip)){
params.put("serverip", serverip);
}
}
for (String apiBaseUrl : apiBaseUrls) {
String url = apiBaseUrl + "/api/notify_final_config";
logger.info("syncConfigToServer,url:" + url);
HttpResponseEntity responseEntity = HttpUtils.postJson(url, JsonUtils.toJson(params),HttpUtils.DEFAULT_CHARSET);
if(responseEntity.isSuccessed()){
logger.info("syncConfigToServer[{}] Ok",url);
}else{
logger.warn("syncConfigToServer[{}] error",url);
}
}
status = ConfigStatus.UPLOAED;
}
private void registerListener(String syncType) {
if("zookeeper".equals(syncType)){
String zkServers = getValue("jeesuite.configcenter.sync-zk-servers");
Validate.notBlank(zkServers,"config[jeesuite.configcenter.sync-zk-servers] is required for syncType [zookeepr]");
configChangeListener = new ZkConfigChangeListener(zkServers);
}else{
configChangeListener = new HttpConfigChangeListener();
}
configChangeListener.register(this);
}
public synchronized void updateConfig(Map updateConfig){
if(!updateConfig.isEmpty()){
Set keySet = updateConfig.keySet();
for (String key : keySet) {
String oldValue = ResourceUtils.getProperty(key);
ResourceUtils.add(key, decodeEncryptIfRequire(updateConfig.get(key)).toString());
StandardEnvironment environment = InstanceFactory.getInstance(StandardEnvironment.class);
MutablePropertySources propertySources = environment.getPropertySources();
MapPropertySource source = null;
synchronized (propertySources) {
if(!propertySources.contains(MANAGER_PROPERTY_SOURCE)){
source = new MapPropertySource(MANAGER_PROPERTY_SOURCE, new LinkedHashMap());
environment.getPropertySources().addFirst(source);
}else{
source = (MapPropertySource) propertySources.get(MANAGER_PROPERTY_SOURCE);
}
}
Map map = (Map) source.getSource();
Properties properties = new Properties();
properties.putAll(map);
properties.putAll(updateConfig);
propertySources.replace(source.getName(), new PropertiesPropertySource(source.getName(), properties));
logger.info("Config [{}] Change,oldValue:{},newValue:{}",key,oldValue,updateConfig.get(key));
}
if(configChangeHanlders == null){
configChangeHanlders = new ArrayList<>();
Map interfaces = InstanceFactory.getInstanceProvider().getInterfaces(ConfigChangeHanlder.class);
if(interfaces != null){
configChangeHanlders.addAll(interfaces.values());
}
}
for (ConfigChangeHanlder hander : configChangeHanlders) {
try {
hander.onConfigChanged(updateConfig);
logger.info("invoke {}.onConfigChanged successed!",hander.getClass().getName());
} catch (Exception e) {
e.printStackTrace();
logger.warn("invoke {}.onConfigChanged error,msg:{}",hander.getClass().getName(),e.getMessage());
}
}
}
}
public boolean pingCcServer(String pingUrl,int retry){
boolean result = false;
try {
System.out.println("pingCcServer ,retry:"+retry);
result = HttpUtils.get(pingUrl,HttpRequestEntity.create().connectTimeout(2000).readTimeout(2000)).isSuccessed();
} catch (Exception e) {}
if(retry == 0)return false;
if(!result){
try {Thread.sleep(1500);} catch (Exception e) {}
return pingCcServer(pingUrl,--retry);
}
return result;
}
public void close(){
configChangeListener.unRegister();
}
private String setReplaceHolderRefValue(Properties properties, String key, String value) {
String[] segments = value.split("\\$\\{");
String seg;
StringBuilder finalValue = new StringBuilder();
for (int i = 0; i < segments.length; i++) {
seg = StringUtils.trimToNull(segments[i]);
if(StringUtils.isBlank(seg))continue;
if(seg.contains(PLACEHOLDER_SUFFIX)){
String refKey = seg.substring(0, seg.indexOf(PLACEHOLDER_SUFFIX)).trim();
//其他非${}的占位符如：{{host}}
String withBraceString = null;
if(seg.contains("{")){
withBraceString = seg.substring(seg.indexOf(PLACEHOLDER_SUFFIX)+1);
}
//如果包含默认值，如：${host:127.0.0.1}
String orginKey = refKey;
if(refKey.contains(":")){
refKey = refKey.split(":")[0];
}
String refValue = properties.getProperty(refKey);
//
if(StringUtils.isNotBlank(refValue) && refValue.contains(PLACEHOLDER_PREFIX)){
//TODO 避免多层嵌套死循环
//refValue = setReplaceHolderRefValue(properties, refKey, refValue);
String subRefKey = refValue.replace(PLACEHOLDER_PREFIX, "").replace(PLACEHOLDER_SUFFIX, "");
refValue = properties.getProperty(subRefKey);
}
if(StringUtils.isBlank(refValue)){
refValue = System.getProperty(refKey);
}
if(StringUtils.isBlank(refValue)){
refValue = PLACEHOLDER_PREFIX + orginKey + PLACEHOLDER_SUFFIX;
}
finalValue.append(refValue);
if(withBraceString != null){
finalValue.append(withBraceString);
}else{
String[] segments2 = seg.split("\\}");
if(segments2.length == 2){
finalValue.append(segments2[1]);
}
}
}else{
finalValue.append(seg);
}
}
properties.put(key, finalValue.toString());
return finalValue.toString();
}
private Object decodeEncryptIfRequire(Object data) {
if (data.toString().startsWith(RSA_PREFIX)) {
Validate.notNull(rsaPrivateKey,"[rsaPrivateKey] not initialized!");
data = data.toString().replace(RSA_PREFIX, "");
return RSA.decrypt(rsaPrivateKey, data.toString());
} else if (data.toString().startsWith(DES_PREFIX)) {
Validate.notBlank(secret,"config[jeesuite.configcenter.encrypt-secret] is required");
data = data.toString().replace(DES_PREFIX, "");
return SimpleCryptUtils.decrypt(secret, data.toString());
}
return data;
}
List sensitiveKeys = new ArrayList<>(Arrays.asList("pass","key","secret","token","credentials"));
private String hideSensitive(String key,String orign){
if(StringUtils.isAnyBlank(key,orign))return "";
boolean is = false;
for (String k : sensitiveKeys) {
if(is = key.toLowerCase().contains(k))break;
}
int length = orign.length();
if(is && length > 1)return orign.substring(0, length/2).concat("****");
return orign;
}
private String getValue(String key){
return getValue(key,null);
}
private String getValue(String key,String defVal){
//jvm 启动参数优先
String value = System.getProperty(key);
if(StringUtils.isNotBlank(value))return value;
value = StringUtils.trimToNull(ResourceUtils.getProperty(key,defVal));
if(StringUtils.isNotBlank(value)){
if(value.startsWith(PLACEHOLDER_PREFIX)){
String refKey = value.substring(2, value.length() - 1).trim();
value = ResourceUtils.getProperty(refKey);
}
}
return value;
}
}
