美文网首页sofabolt
SOFABolt 源码分析11 - Config 配置管理的设计

SOFABolt 源码分析11 - Config 配置管理的设计

作者: 原水寒 | 来源:发表于2018-10-14 14:41 被阅读81次
    image.png

    如上图所示,SOFABolt 整个配置管理分为五部分:

    • 系统变量管理类:ConfigManager,配置多个实例共享的配置。
    • 用户变量管理类:ConfigContainer,可以为多个实例配置不同的配置。
    • 开关类:Switch,一些开关操作。
    • 用户可配置实例类:ConfigurableInstance,包含 ConfigContainer 和 Switch 实例,RpcServer 和 RpcClient 都是其子类。
      Rpc 独有的配置项和管理器:RpcConfig。

    参数配置方式通常有两种,一种是系统变量的设置,这是一种全局的配置;另外一种,就是用户通过 api (例如,通过构造器参数)进行配置,这样可以为不同的实例的参数配置不同的值。所以在 Netty 的可配置类中,通常会提供系统变量配置和构造器配置两种参数配置,当然参数本身还有原始的默认值。最终参数配置的优先级是:

    用户 api 配置变量(构造器参数变量)> 系统变量 > 原始的默认值

    一、系统变量管理类

    系统变量的设置通常由两种方式:

    • java 进程启动参数添加 -Dxxx = yyy
    • System.setProperty(String key, String value)

    SOFABolt 通过 Configs 类配置了一系列的配置项的 key 及其默认值;通过 ConfigManager 进行配置项的获取。源码如下:

    ========================== Configs ==========================
    /** TCP_NODELAY option */
    public static final String TCP_NODELAY  = "bolt.tcp.nodelay";
    public static final String TCP_NODELAY_DEFAULT = "true";
    
    ======================= ConfigManager =======================
    public static boolean tcp_nodelay() {
        return getBool(Configs.TCP_NODELAY, Configs.TCP_NODELAY_DEFAULT);
    }
    
    public static boolean getBool(String key, String defaultValue) {
        return Boolean.parseBoolean(System.getProperty(key, defaultValue));
    }
    

    二、用户变量管理类

    • ConfigType 定义了是服务端还是客户端的配置:CLIENT_SIDE 和 SERVER_SIDE;
    • ConfigItem 定义了配置项,目前只有两个:NETTY_BUFFER_LOW_WATER_MARK 和 NETTY_BUFFER_HIGH_WATER_MARK,Netty write buffer 的高低水位。
    • ConfigContainer 定义了设置和获取用户配置项的接口;DefaultConfigContainer 对其进行了默认实现

    疑问:关于 ConfigItem 的扩展,官方文档:“You can define new config items based on this if need.” 但是 ConfigItem 是 enum,是无法被继承的,应该怎样去扩展自己的配置呢?个人感觉 ConfigItem 和 Configs 有冗余的嫌疑,https://github.com/alipay/sofa-bolt/issues/101

    三、开关类

    • Switch 定义开关配置接口;
    • GlobalSwitch 定义了全局开关项;
    • ProtocolSwitch 定义了与协议相关的开关项,例如 CRC_SWITCH。
    ======================= GlobalSwitch =======================
    // 重连
    public static final int CONN_RECONNECT_SWITCH = 0;
    // 监控连接
    public static final int CONN_MONITOR_SWITCH = 1;
    // 服务端启动连接管理器
    public static final int SERVER_MANAGE_CONNECTION_SWITCH = 2;
    // 服务端同步停止
    public static final int SERVER_SYNC_STOP = 3;
    /** user settings */
    private BitSet userSettings  = new BitSet();
    
    /**
     * Init with system default value
     *   if settings exist by system property then use system property at first;
     *   if no settings exist by system property then use default value in {@link com.alipay.remoting.config.Configs}
     * All these settings can be overwrite by user api settings.
     */
    public GlobalSwitch() {
        if (ConfigManager.conn_reconnect_switch()) {
            userSettings.set(CONN_RECONNECT_SWITCH);
        } else {
            userSettings.clear(CONN_RECONNECT_SWITCH);
        }
    
        if (ConfigManager.conn_monitor_switch()) {
            userSettings.set(CONN_MONITOR_SWITCH);
        } else {
            userSettings.clear(CONN_MONITOR_SWITCH);
        }
    }
    
    ======================= ProtocolSwitch =======================
    // switche index
    public static final int CRC_SWITCH_INDEX = 0x000;
    // default value
    public static final boolean CRC_SWITCH_DEFAULT_VALUE = true;
    /** protocol switches */
    private BitSet bs = new BitSet();
    

    开关是使用 BitSet 来实现的,BitSet简要原理:
    BitSet默认底层数据结构是一个long[]数组,开始时长度为1,即只有long[0],而一个long有64bit。
    当BitSet.set(1)的时候,表示将long[0]的第二位设置为true,即0000 0000 ... 0010(64bit),则long[0]==2;
    当BitSet.get(1)的时候,第二位为1,则表示true;如果是0,则表示false;
    当BitSet.set(64)的时候,表示设置第65位,此时long[0]已经不够用了,扩容处long[1]来,进行存储。
    存储类似 {index:boolean} 键值对。

    四、用户可配置实例类

    NettyConfigure:提供了 Netty 配置的接口类;
    ConfigurableInstance:提供了可配置实例的接口类;
    AbstractConfigurableInstance:ConfigurableInstance 实现类,源码如下:

    public class AbstractConfigurableInstance implements ConfigurableInstance {
        private ConfigContainer configContainer = new DefaultConfigContainer();
        private GlobalSwitch    globalSwitch    = new GlobalSwitch();
        private ConfigType      configType;
    
        protected AbstractConfigurableInstance(ConfigType configType) {
            this.configType = configType;
        }
    
        // public 方法,用户可以直接获取该实例进行操作
        @Override
        public ConfigContainer conf() {
            return this.configContainer;
        }
    
         // public 方法,用户可以直接获取该实例进行操作
        @Override
        public GlobalSwitch switches() {
            return this.globalSwitch;
        }
    
        @Override
        public void initWriteBufferWaterMark(int low, int high) {
            this.configContainer.set(configType, ConfigItem.NETTY_BUFFER_LOW_WATER_MARK, low);
            this.configContainer.set(configType, ConfigItem.NETTY_BUFFER_HIGH_WATER_MARK, high);
        }
    
        @Override
        public int netty_buffer_low_watermark() {
            // 首先获取用户api配置
            if (null != configContainer
                && configContainer.contains(configType, ConfigItem.NETTY_BUFFER_LOW_WATER_MARK)) {
                return (Integer) configContainer
                    .get(configType, ConfigItem.NETTY_BUFFER_LOW_WATER_MARK);
            } else {
                // 然后在从系统变量进行获取
                return ConfigManager.netty_buffer_low_watermark();
            }
        }
    
        @Override
        public int netty_buffer_high_watermark() {
            if (null != configContainer
                && configContainer.contains(configType, ConfigItem.NETTY_BUFFER_HIGH_WATER_MARK)) {
                return (Integer) configContainer.get(configType,
                    ConfigItem.NETTY_BUFFER_HIGH_WATER_MARK);
            } else {
                return ConfigManager.netty_buffer_high_watermark();
            }
        }
    }
    

    AbstractConfigurableInstance 有两个极重要的子类 RpcServer 和 RpcClient。

    public RpcClient() {
        // 创建对象的时候就指定好了 ConfigType
        super(ConfigType.CLIENT_SIDE);
    }
    
    public void init() {
        ...
        // 连接监控开关是否打开
        if (switches().isOn(GlobalSwitch.CONN_MONITOR_SWITCH)) {
            ...
        }
        // 重连开关是否打开
        if (switches().isOn(GlobalSwitch.CONN_RECONNECT_SWITCH)) {
            ...
        }
    }
    
    /**
     * enable connection reconnect switch on
     * Notice: This api should be called before RpcClient#init()
     */
    public void enableReconnectSwitch() {
        this.switches().turnOn(GlobalSwitch.CONN_RECONNECT_SWITCH);
    }
    
    /**
     * disable connection reconnect switch off
     * Notice: This api should be called before RpcClient#init()
     */
    public void disableReconnectSwith() {
        this.switches().turnOff(GlobalSwitch.CONN_RECONNECT_SWITCH);
    }
    
    /**
     * is reconnect switch on
     * @return
     */
    public boolean isReconnectSwitchOn() {
        return this.switches().isOn(GlobalSwitch.CONN_RECONNECT_SWITCH);
    }
    

    最后,来看一下使用姿势:

    ======================= 系统变量 =======================
    System.setProperty(Configs.CONN_RECONNECT_SWITCH, "true");
    java -Dbolt.conn.reconnect.switch = true -jar xx.jar
    
    
    ======================= 用户变量 =======================
    RpcClient client = new RpcClient();
    // 1. 通用方式,目前 ConfigItem 只有 netty write buffer 的高低水位
    client.conf().set(ConfigType.CLIENT_SIDE, ConfigItem.NETTY_BUFFER_LOW_WATER_MARK, 80 * 1024);
    client.conf().set(ConfigType.CLIENT_SIDE, ConfigItem.NETTY_BUFFER_HIGH_WATER_MARK, 100 * 1024);
    
    // 2. 设置高低水位的预留方式
    client.initWriteBufferWaterMark(80 * 1024, 100 * 1024);
    client.init();
    
    ======================= 开关 =======================
    // 1. 通用方式
    client.switches().turnOn(GlobalSwitch.CONN_RECONNECT_SWITCH);
    client.switches().isOn(GlobalSwitch.CONN_RECONNECT_SWITCH);
    client.switches().turnOff(GlobalSwitch.CONN_RECONNECT_SWITCH);
    client.init();
    
    // 2. 预留设置
    client.enableReconnectSwitch();
    client.isReconnectSwitchOn();
    client.disableReconnectSwith();
    

    相关文章

      网友评论

        本文标题:SOFABolt 源码分析11 - Config 配置管理的设计

        本文链接:https://www.haomeiwen.com/subject/kvmzaftx.html