基于BIT数组实现全局功能开关-阿里云开发者社区

基于BIT数组实现全局功能开关

2022-05-13 236

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 某一天巧合打开了sofa-bolt项目，查找部分源码，看到了项目中使用bit数组实现功能开关的特性，感觉这种方式可以借鉴，于是写下这篇文章。

前提

某一天巧合打开了sofa-bolt项目，查找部分源码，看到了项目中使用bit数组实现功能开关的特性，感觉这种方式可以借鉴，于是写下这篇文章。

原理

bit数组的布局如下：

微信截图_20220513210221.png

由于每个bit都可以表示1或者0，刚好对应于开关的ON和OFF。只需要定义好每个开关所在的bit数组下标和开关的状态（ON = 1或者OFF = 0），通过判断不同开关下标所在的bit即可判断开关的状态：

微信截图_20220513210227.png

优点：节省空间，理论上只需要占用最多2n位的内存（n为开关的数量，这里考虑扩容，扩容让bit数组长度为原来的2倍）
缺点：暂时没发现

实现

JDK中的bit数组可以直接使用BitSet。首先定义开关定义SwitchDef：

public class SwitchConst {
    public static final boolean ON = true;
    public static final boolean OFF = false;
}
@RequiredArgsConstructor
@Getter
public enum SwitchDef {
    /**
     * 启用HTTPS
     */
    ENABLE_HTTPS(0, SwitchConst.ON, "启用HTTPS"),
    /**
     * 启用异步
     */
    ENABLE_ASYNC(1, SwitchConst.OFF, "启用异步"),
    ;
    /**
     * 下标
     */
    private final int index;
    /**
     * 默认状态
     */
    private final boolean defaultStatus;
    /**
     * 描述
     */
    private final String description;
    @Override
    public String toString() {
        return String.format("SwitchDef(name=%s,description=%s)", name(), description);
    }
}
复制代码

接着定义开关接口Switch：

public interface Switch {
    /**
     * 启用
     *
     * @param switchDef switchDef
     */
    void turnOn(SwitchDef switchDef);
    /**
     * 关闭
     *
     * @param switchDef switchDef
     */
    void turnOff(SwitchDef switchDef);
    /**
     * 判断状态
     *
     * @param switchDef switchDef
     * @return boolean
     */
    boolean status(SwitchDef switchDef);
}
复制代码

最后编写开关实现BitSetSwitch和客户端代码：

public enum BitSetSwitch implements Switch {
    /**
     * 单例
     */
    X;
    BitSetSwitch() {
        init();
    }
    private final BitSet switches = new BitSet();
    @Override
    public void turnOn(SwitchDef switchDef) {
        switches.set(switchDef.getIndex(), SwitchConst.ON);
    }
    @Override
    public void turnOff(SwitchDef switchDef) {
        switches.clear(switchDef.getIndex());
    }
    @Override
    public boolean status(SwitchDef switchDef) {
        return switches.get(switchDef.getIndex());
    }
    private void init() {
        Stream.of(SwitchDef.values()).forEach(item -> switches.set(item.getIndex(), item.isDefaultStatus()));
    }
    public static void main(String[] args) {
        Switch s = BitSetSwitch.X;
        s.turnOn(SwitchDef.ENABLE_HTTPS);
        s.turnOff(SwitchDef.ENABLE_ASYNC);
        System.out.printf("开关[%s],状态:%s%n", SwitchDef.ENABLE_HTTPS, s.status(SwitchDef.ENABLE_HTTPS));
        System.out.printf("开关[%s],状态:%s%n", SwitchDef.ENABLE_ASYNC, s.status(SwitchDef.ENABLE_ASYNC));
    }
}
复制代码

执行该main方法后控制台输出如下：

开关[SwitchDef(name=ENABLE_HTTPS,description=启用HTTPS)],状态:true
开关[SwitchDef(name=ENABLE_ASYNC,description=启用异步)],状态:false
复制代码

仿真场景（伪代码）如下：

Switch s = BitSetSwitch.X;
String uri = "www.throwx.cn";
String schema = "http";
if (s.turnOn(SwitchDef.ENABLE_HTTPS)){
    schema = "https";
}
HttpClint ch = new DefaultHttpClient();
if (s.turnOn(SwitchDef.ENABLE_ASYNC)){
    ch = new AsyncHttpClient();
}
Result r = ch.executeRequest(schema + uri);
......
复制代码