美文网首页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