springboot项目接入配置中心,实现@ConfigurationProperties的bean属性刷新方案

Java框架

浏览数:176

2019-3-22

AD:资源代下载服务

前言

  配置中心,通过key=value的形式存储环境变量。配置中心的属性做了修改,项目中可以通过配置中心的依赖(sdk)立即感知到。需要做的就是如何在属性发生变化时,改变带有@ConfigurationProperties的bean的相关属性。

配置中心

  在读配置中心源码的时候发现,里面维护了一个Environment,以及ZookeeperPropertySource。当配置中心属性发生变化的时候,清空ZookeeperPropertySource,并放入最新的属性值。

  

public class ZookeeperPropertySource extends EnumerablePropertySource<Properties>

  

  ZookeeperPropertySource重写了equals和hahscode方法,根据这两个方法可以判定配置中心是否修改了属性。

  

配置中心定义的属性变量

message.center.channels[0].type=HELIUYAN
message.center.channels[0].desc=和留言系统
message.center.channels[1].type=EC_BACKEND
message.center.channels[1].desc=电商后台
message.center.channels[2].type=BILL_FLOW
message.center.channels[2].desc=话费和流量提醒
message.center.channels[3].type=INTEGRATED_CASHIER
message.center.channels[3].desc=综合收银台

message.center.businesses[0].type=BIZ_EXP_REMINDER
message.center.businesses[0].desc=业务到期提醒
message.center.businesses[0].topic=message-center-biz-expiration-reminder-topic
message.center.businesses[1].type=RECHARGE_TRANSACTION_PUSH
message.center.businesses[1].desc=充值交易实时推送
message.center.businesses[1].topic=message-center-recharge-transaction-push-topic

message.center.businesses2Channels[BIZ_EXP_REMINDER]=EC_BACKEND
message.center.businesses2Channels[RECHARGE_TRANSACTION_PUSH]=INTEGRATED_CASHIER

message.center.bizTypeForMsgType[RECHARGE_TRANSACTION_PUSH]=data.type:pay-finish,data.type:rechr-finish,data.type:refund-finish

java属性配置映射类

import org.springframework.boot.context.properties.ConfigurationProperties;

import java.util.List;
import java.util.Map;
import java.util.Objects;

/**
 * @author hujunzheng
 * @create 2018-06-28 11:37
 **/
@ConfigurationProperties(prefix = "message.center")
public class MessageCenterConstants {

    private List<Business> businesses;

    private List<Channel> channels;

    private Map<String, String> businesses2Channels;

    private Map<String, String> bizTypeForMsgType;

    public void setBusinesses(List<Business> businesses) {
        this.businesses = businesses;
    }

    public void setChannels(List<Channel> channels) {
        this.channels = channels;
    }

    public List<Business> getBusinesses() {
        return businesses;
    }

    public List<Channel> getChannels() {
        return channels;
    }

    public Map<String, String> getBusinesses2Channels() {
        return businesses2Channels;
    }

    public void setBusinesses2Channels(Map<String, String> businesses2Channels) {
        this.businesses2Channels = businesses2Channels;
    }

    public Map<String, String> getBizTypeForMsgType() {
        return bizTypeForMsgType;
    }

    public void setBizTypeForMsgType(Map<String, String> bizTypeForMsgType) {
        this.bizTypeForMsgType = bizTypeForMsgType;
    }

    public static class Business implements Comparable<Business> {
        //业务类型
        private String type;
        //业务描述
        private String desc;
        //对应 kafka 的 topic
        private String topic;

        public String getType() {
            return type;
        }

        public void setType(String type) {
            this.type = type;
        }

        public String getDesc() {
            return desc;
        }

        public void setDesc(String desc) {
            this.desc = desc;
        }

        public String getTopic() {
            return topic;
        }

        public void setTopic(String topic) {
            this.topic = topic;
        }

        @Override
        public int compareTo(Business o) {
            if (type.compareTo(o.type) == 0 || topic.compareTo(o.topic) == 0) {
                return 0;
            }
            return Objects.hash(type, topic);
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            Business business = (Business) o;
            return Objects.equals(type, business.type) ||
                    Objects.equals(topic, business.topic);
        }

        @Override
        public int hashCode() {
            return Objects.hash(type, topic);
        }

        @Override
        public String toString() {
            return "Business{" +
                    "type='" + type + '\'' +
                    ", desc='" + desc + '\'' +
                    ", topic='" + topic + '\'' +
                    '}';
        }
    }

    public static class Channel implements Comparable<Channel> {

        //渠道类型
        private String type;
        //渠道描述
        private String desc;

        public String getType() {
            return type;
        }

        public void setType(String type) {
            this.type = type;
        }

        public String getDesc() {
            return desc;
        }

        public void setDesc(String desc) {
            this.desc = desc;
        }

        @Override
        public int compareTo(Channel o) {
            return this.type.compareTo(o.type);
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            Channel channel = (Channel) o;
            return Objects.equals(type, channel.type);
        }

        @Override
        public int hashCode() {
            return Objects.hash(type);
        }

        @Override
        public String toString() {
            return "Channel{" +
                    "type='" + type + '\'' +
                    ", desc='" + desc + '\'' +
                    '}';
        }
    }
}

属性刷新方案

@Bean
public MergedProperties kafkaMessageMergedProperties() {
    return ConfigCenterUtils.createToRefreshPropertiesBean(MergedProperties.class);
}

public static class MergedProperties {
    private Map<String, MessageCenterConstants.Business> businesses;
    private Map<String, MessageCenterConstants.Channel> channels;
    //业务映射渠道
    private Map<String, String> businesses2Channels;
    //消息类型映射业务类型
    private Map<String, String> msgType2BizType;

    public MergedProperties() throws GeneralException {
        this.refreshProperties();
    }

    private void refreshProperties() throws GeneralException {
//获取到配置中心最新的propertySource ZookeeperPropertySource propertySource = ConfigHelper.getZookeeperPropertySource(); MessageCenterConstants messageCenterConstants = null;
     //判断属性是否刷新 if (ConfigCenterUtils.propertySourceRefresh(propertySource)) {
       //将属性binding到带有@ConfigurationProperties注解的类中 messageCenterConstants = RelaxedConfigurationBinder .with(MessageCenterConstants.class) .setPropertySources(propertySource) .doBind(); }
     //以下是自定义处理,可忽略 if (!Objects.isNull(messageCenterConstants)) { //Business.type <-> Business this.setBusinesses(Maps.newHashMap( Maps.uniqueIndex(Sets.newHashSet(messageCenterConstants.getBusinesses()), business -> business.getType()) )); //Channel.type <-> Channel this.setChannels(Maps.newHashMap( Maps.uniqueIndex(Sets.newHashSet(messageCenterConstants.getChannels()), channel -> channel.getType()) )); //business <-> channels this.setBusinesses2Channels(messageCenterConstants.getBusinesses2Channels()); //消息类型映射业务类型 this.setMsgType2BizType( messageCenterConstants.getBizTypeForMsgType().entrySet() .stream().map(entry -> { Map<String, String> tmpMap = Maps.newHashMap(); if (StringUtils.isBlank(entry.getValue())) { return tmpMap; } Arrays.stream(entry.getValue().split(",")).forEach(value -> tmpMap.put(value, entry.getKey())); return tmpMap; }).flatMap(map -> map.entrySet().stream()).collect(Collectors.toMap(entry -> entry.getKey(), entry -> entry.getValue())) ); } }
   //刷新方法 private void catchRefreshProperties() { try { this.refreshProperties(); } catch (Exception e) { LOGGER.error("KafkaMessageConfig 配置中心属性刷新失败", e); } }
   //get方法上指定刷新属性 @ToRefresh(method = "catchRefreshProperties") public Map<String, MessageCenterConstants.Business> getBusinesses() { return businesses; } public void setBusinesses(Map<String, MessageCenterConstants.Business> businesses) { this.businesses = businesses; } @ToRefresh(method = "catchRefreshProperties") public Map<String, MessageCenterConstants.Channel> getChannels() { return channels; } public void setChannels(Map<String, MessageCenterConstants.Channel> channels) { this.channels = channels; } @ToRefresh(method = "catchRefreshProperties") public Map<String, String> getBusinesses2Channels() { return businesses2Channels; } public void setBusinesses2Channels(Map<String, String> businesses2Channels) { this.businesses2Channels = businesses2Channels; } @ToRefresh(method = "catchRefreshProperties") public Map<String, String> getMsgType2BizType() { return msgType2BizType; } public void setMsgType2BizType(Map<String, String> msgType2BizType) { this.msgType2BizType = msgType2BizType; } }

工具类

ConfigCenterUtils

import com.cmos.cfg.core.ConfigHelper;
import com.cmos.cfg.zookeeper.ZookeeperPropertySource;
import org.apache.commons.lang3.StringUtils;
import org.springframework.cglib.proxy.Enhancer;
import org.springframework.cglib.proxy.MethodInterceptor;
import org.springframework.cglib.proxy.MethodProxy;
import org.springframework.core.BridgeMethodResolver;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.util.ReflectionUtils;

import java.lang.reflect.Method;
import java.util.Objects;

/**
 * @author hujunzheng
 * @create 2018-07-04 15:45
 **/
public class ConfigCenterUtils {
    private static ZookeeperPropertySource propertySource = ConfigHelper.getZookeeperPropertySource();

  //判断配置中心属性是否刷新 public synchronized static boolean propertySourceRefresh(ZookeeperPropertySource newPropertySource) { if (propertySource.equals(newPropertySource)) { return false; } if (propertySource.hashCode() == newPropertySource.hashCode()) { return false; } propertySource = newPropertySource; return true; }
   //创建代理类,代理@ToRefresh注解的方法,调用相应的刷新方法 public static <T> T createToRefreshPropertiesBean(Class<T> clazz) { Enhancer enhancer = new Enhancer(); // 设置代理对象父类 enhancer.setSuperclass(clazz); // 设置增强 enhancer.setCallback(new MethodInterceptor() { @Override public Object intercept(Object target, Method method, Object[] args, MethodProxy methodProxy) throws Throwable { ToRefresh toRefresh = AnnotationUtils.findAnnotation(method, ToRefresh.class); if (Objects.isNull(toRefresh) || StringUtils.isBlank(toRefresh.method())) { return methodProxy.invokeSuper(target, args); } Method refreshMethod = ReflectionUtils.findMethod(target.getClass(), toRefresh.method()); if (Objects.isNull(refreshMethod)) { return methodProxy.invokeSuper(target, args); } refreshMethod = BridgeMethodResolver.findBridgedMethod(refreshMethod); refreshMethod.setAccessible(true); refreshMethod.invoke(target, null); return methodProxy.invokeSuper(target, args); } }); return (T) enhancer.create();// 创建代理对象 } }
import org.apache.commons.lang3.StringUtils;

import java.lang.annotation.Documented;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;

import static java.lang.annotation.ElementType.METHOD;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

/**
 * @author hujunzheng
 * @create 2018-07-06 9:59
 **/
@Target({METHOD})
@Retention(RUNTIME)
@Documented
public @interface ToRefresh {
    //刷新方法
    String method() default StringUtils.EMPTY;
}

RelaxedConfigurationBinder

  动态将propertysource绑定到带有@ConfigurationProperties注解的bean中

  参考:org.springframework.boot.context.properties.ConfigurationPropertiesBindingPostProcessor

import com.cmos.common.exception.GeneralException;
import org.springframework.boot.bind.PropertiesConfigurationFactory;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.core.convert.ConversionService;
import org.springframework.core.convert.support.DefaultConversionService;
import org.springframework.core.env.*;
import org.springframework.validation.Validator;
import org.springframework.validation.beanvalidation.SpringValidatorAdapter;

import javax.validation.Validation;

import static org.springframework.core.annotation.AnnotatedElementUtils.getMergedAnnotation;

/**
 * @author hujunzheng
 * @create 2018-07-03 18:01
 *
 * 不强依赖ConfigurationProperties,进行配置注入
 **/
public class RelaxedConfigurationBinder<T> {
    private final PropertiesConfigurationFactory<T> factory;

    public RelaxedConfigurationBinder(T object) {
        this(new PropertiesConfigurationFactory<>(object));
    }

    public RelaxedConfigurationBinder(Class<T> type) {
        this(new PropertiesConfigurationFactory<>(type));
    }

    public static <T> RelaxedConfigurationBinder<T> with(T object) {
        return new RelaxedConfigurationBinder<>(object);
    }

    public static <T> RelaxedConfigurationBinder<T> with(Class<T> type) {
        return new RelaxedConfigurationBinder<>(type);
    }

    public RelaxedConfigurationBinder(PropertiesConfigurationFactory<T> factory) {
        this.factory = factory;
        ConfigurationProperties properties = getMergedAnnotation(factory.getObjectType(), ConfigurationProperties.class);
        javax.validation.Validator validator = Validation.buildDefaultValidatorFactory().getValidator();
        factory.setValidator(new SpringValidatorAdapter(validator));
        factory.setConversionService(new DefaultConversionService());
        if (null != properties) {
            factory.setIgnoreNestedProperties(properties.ignoreNestedProperties());
            factory.setIgnoreInvalidFields(properties.ignoreInvalidFields());
            factory.setIgnoreUnknownFields(properties.ignoreUnknownFields());
            factory.setTargetName(properties.prefix());
            factory.setExceptionIfInvalid(properties.exceptionIfInvalid());
        }
    }

    public RelaxedConfigurationBinder<T> setTargetName(String targetName) {
        factory.setTargetName(targetName);
        return this;
    }

    public RelaxedConfigurationBinder<T> setPropertySources(PropertySource<?>... propertySources) {
        MutablePropertySources sources = new MutablePropertySources();
        for (PropertySource<?> propertySource : propertySources) {
            sources.addLast(propertySource);
        }
        factory.setPropertySources(sources);
        return this;
    }

    public RelaxedConfigurationBinder<T> setPropertySources(Environment environment) {
        factory.setPropertySources(((ConfigurableEnvironment) environment).getPropertySources());
        return this;
    }

    public RelaxedConfigurationBinder<T> setPropertySources(PropertySources propertySources) {
        factory.setPropertySources(propertySources);
        return this;
    }

    public RelaxedConfigurationBinder<T> setConversionService(ConversionService conversionService) {
        factory.setConversionService(conversionService);
        return this;
    }

    public RelaxedConfigurationBinder<T> setValidator(Validator validator) {
        factory.setValidator(validator);
        return this;
    }

    public RelaxedConfigurationBinder<T> setResolvePlaceholders(boolean resolvePlaceholders) {
        factory.setResolvePlaceholders(resolvePlaceholders);
        return this;
    }

    public T doBind() throws GeneralException {
        try {
            return factory.getObject();
        } catch (Exception ex) {
            throw new GeneralException("配置绑定失败!", ex);
        }
    }
}

 


版权声明:本文原创发表于博客园,作者为小眼儿。本文欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则视为侵权。