当Java枚举遇到位掩码，还能这么玩?

客官°小女子只卖身不卖艺 2022-10-12 01:50 120阅读 0赞

![0D5F937FE12312312D81F17F.jpg][]

程序员的成长之路

互联网/程序员/技术/资料共享

[关注][Link 1]

阅读本文大概需要 4.2 分钟。

来自：https://juejin.cn/post/6971567246774632462

相信大家都用过 Fastjson，阿里的一个开源 JSON 库，在阿里系的开源项目里应用的非常广泛。虽然有时候也被戏称“沉的快”，但 Fastjson 从功能丰富度、易用性、源码设计角度来看，都是一款十分优秀的工具库。

在使用 Fastjson 时，经常会配置一些枚举参数，比如日期格式、格式化输出、NULL值格式啊等等，就像下面这样配置：

String jsonStr = JSON.toJSONString(obj, 
                          SerializerFeature.WriteDateUseDateFormat,
                          SerializerFeature.PrettyFormat,
                          SerializerFeature.WriteNullStringAsEmpty);
    
    // JSON.toJSONString
    public static String toJSONString(Object object, SerializerFeature... features);

这种配置方式用起来那是相当爽，要什么输出配置，利用 JAVA 的可变长度参数（varargs）直接在方法后面动态追加就行；要是再加上`import static`，那么连 SerializerFeature 都不用写了。

不过细想一下，这种动态传的参数数组，Fastjson 在接受后怎么知道我们具体传了哪些参数？接受时遍历数组，每次 equals 对比吗？比如这样：

// 写三个 for 循环的原因是大概率下，不同 feature 处理时机不同，所以不能在一个 for 循环内处理
    
    for (SerializerFeature feature : features) {
        if(feature.equals(SerializerFeature.WriteDateUseDateFormat)){
            // solve WriteDateUseDateFormat
        }
    }
    
    for (SerializerFeature feature : features) {
        if(feature.equals(SerializerFeature.PrettyFormat)){
            // solve PrettyFormat
        }
    }
    
    for (SerializerFeature feature : features) {
        if(feature.equals(SerializerFeature.WriteNullStringAsEmpty)){
            // solve WriteNullStringAsEmpty
        }
    }

这样也太不“优雅”了，每次还需要遍历，白白浪费性能！或者只用一个循环，弄几个变量存储这几个 boolean 值呢：

boolean writeDateUseDateFormatEnable = false;
    boolean PrettyFormatEnable = false;
    boolean WriteNullStringAsEmptyEnable = false;
    
    for (SerializerFeature feature : features) {
        if(feature.equals(SerializerFeature.WriteDateUseDateFormat)){
        writeDateUseDateFormatEnable = true;
        }
        if(feature.equals(SerializerFeature.PrettyFormat)){
        PrettyFormatEnable = true;
        }
        if(feature.equals(SerializerFeature.WriteNullStringAsEmpty)){
        WriteNullStringAsEmptyEnable = true;
        }
    }

这样比上面要好一点了，但还是需要循环判断，而且对每个 Feature 都要额外增加一个变量来存储，一样不太“优雅”。

在 Fastjson 中使用了一种很巧妙的方式来处理这个动态的枚举参数

## 枚举中的序数（ordinal） ##

在正式介绍之前，需要先了解枚举中的一个概念- 序数（ordinal），每个枚举类都会有一个 ordinal 属性，这个ordinal 代表的是当前枚举值在枚举类中的序号。比如下面这个枚举类里，F\_A /F\_B /F\_C /F\_D 四个枚举值的序数依次为 0/1/2/3

public enum Feature {
      F_A, // ordinal 0
      F_B, // ordinal 1
      F_C, // ordinal 2
      F_D, // ordinal 3
      ;
    }

通过 ordinal() 方法，就可以获取枚举实例的序数值，比如`Feature._F_A_.ordinal()`

## Fastjson 中的妙用 ##

了解了枚举序数之后，现在来看看 Fastjson 中是怎么个玩法。在`SerializerFeature` 的源码中有一个 \*\*mask（掩码） \*\*，这个 mask 的值为 `1 << ordinal`

### 枚举中的位掩码 - Mask ###

public enum SerializerFeature {
       WriteDateUseDateFormat,
       PrettyFormat,
       WriteNullStringAsEmpty
        ...
    
        SerializerFeature(){
            mask = (1 << ordinal());
        }
    
        public final int mask;
    
        ...
    }

在位运算中掩码 mask 的作用一般是为了保持/更改/删除某（些）位的值，有张图非常形象（这个图可以简单的理解为，白色像素代表1，黑色像素代表0，按为与后，为1的像素位才会显示）：

![e1b6fdc777117a8d03239dd4c0baec08.png][]

那在 `SerializerFeature` 中，WriteDateUseDateFormat, PrettyFormat, WriteNullStringAsEmpty 三个值得序数分别为 0/1/2，左移后他们对应的二进制位如下：

0 0 0 1 WriteDateUseDateFormat
    0 0 1 0 PrettyFormat
    0 1 0 0 WriteNullStringAsEmpty
    
    ...
    1 0 0 0

这里的做法很巧妙，用 1 左移序数个位，就可以得到一个序数位为 1 的数字，比如序数为 1 ，那么第0位就是1，序数为3，那么第4为就是1，以此类推，这样枚举中每个值的 mask 里为 1 的位都会不同

### 多个配置的处理 ###

单看这个位掩码还是觉得没啥用，来看看实战吧。现在定义一个初始为 0 的 features 变量，用来存储所有的 feature

int features = 0;

利用位或（OR）对 features 和 mask 进行运算

features |= SerializerFeature.WriteDateUseDateFormat.getMask();
    
          0 0 0 0 [input(features)]
    (|)   0 0 0 1 [mask(feature mask)]
    -------------
          0 0 0 1 [output(new features)]

位或运算后的 features 为 `0 0 0 1`，第 0 位上变成了 1 ，代表第 1 位的枚举值（WriteDateUseDateFormat）被启用了，接着继续对 PrettyFormat 也执行位或，

features |= SerializerFeature.PrettyFormat.getMask();

0 0 0 1 [input(features)]
    (|)   0 0 1 0 [mask(feature mask)]
    -------------
          0 0 1 1 [output(new features)]

此时 features 为 `0 0 1 1`，第 2 位 上也变成了 1，代表第 2 位的枚举值（PrettyFormat）也被启用了

### 判断是否配置 ###

有了 features 的值，还是需要一个简单的判断方法，来检查某个枚举值是否被设置：

public static boolean isEnabled(int features, SerializerFeature feature) {
        return (features & feature.mask) != 0;
    }

用 features 和 某个 Feature 的掩码做位与后，就可以得出一个某位为 1 的数字。位与运算中只有上下两个位都为 1 ，返回的位才会为 1，那么只要返回的结果位内包含任何一个 1 ，这个数就不会为 0 ；所以只要这个结果不为 0 ，就可以说明这个 Feature 已经被设置了。

0 0 1 1 [input(features)]
    (&)   0 0 1 0 [mask(PrettyFormat)]
    -------------
          0 0 1 0 [output(new features)]

比如上面这个例子中，当前 features 为 `0 0 1 1`，和 PrettyFormat 的 mask 做位与后，就可以得出 `0 0 1 0` ，结果不为 0 ，所以 PrettyFormat 已经被设置了

### 完整示例 ###

// 存储所有配置的 Feature
    int features = 0;
    
    // 每添加一个 Feature， 就拿 features 和 当前 Feature 的掩码做位或运算
    features |= SerializerFeature.WriteDateUseDateFormat.getMask();
    features |= SerializerFeature.PrettyFormat.getMask();
    
    // 再通过位与运算的结果，就可以判断某个 Feature 是否配置
    boolean writeDateUseDateFormatEnabled = SerializerFeature.isEnabled(features,SerializerFeature.WriteDateUseDateFormat);
    boolean prettyFormatEnabled = SerializerFeature.isEnabled(features,SerializerFeature.PrettyFormat);
    boolean writeNullStringAsEmpty = SerializerFeature.isEnabled(features,SerializerFeature.WriteNullStringAsEmpty);
    
    System.out.println("writeDateUseDateFormatEnabled: "+writeDateUseDateFormatEnabled);
    System.out.println("prettyFormatEnabled: "+prettyFormatEnabled);
    System.out.println("writeNullStringAsEmpty: "+writeNullStringAsEmpty);
    
    //output
    writeDateUseDateFormatEnabled: true
    prettyFormatEnabled: true
    writeNullStringAsEmpty: false

## 总结 ##

不止是 Fastjson，Jackson 中对 Feature 的处理也是基于这个枚举序数+位掩码的逻辑，二者实现一模一样，算是一种主流的做法吧。

<END>

**推荐阅读：**

[别去外包！][Link 2]

[MySQL Online DDL 原理和踩坑][MySQL Online DDL]

最近面试BAT，整理一份面试资料《Java面试BATJ通关手册》，覆盖了Java核心技术、JVM、Java并发、SSM、微服务、数据库、数据结构等等。

**获取方式：点个「****在看****」，点击上方小卡片，进入公众号后回复「****面试题****」领取，更多内容陆续奉上。**

朕已阅 ![7946e9f74e18d290530991d12a2578d1.gif][]

[0D5F937FE12312312D81F17F.jpg]: /images/20221005/56f67ff49775405cbfa8fd1ae08fe13b.png
[Link 1]: https://mp.weixin.qq.com/s?__biz=MzUyNDkzNzczNQ%3D%3D&chksm=fa24f0bacd5379ac93d74ffebac826d2401fd494f82e900e5102a8bf968f108e4ec72a4ad548&idx=3&lang=zh_CN&mid=2247487442&scene=21&sn=61e758d4bf9f15a3419fd028146469a3&token=1854383306#wechat_redirect
[e1b6fdc777117a8d03239dd4c0baec08.png]: /images/20221005/b61b186c3f9149c084b3ffe15d044247.png
[Link 2]: http://mp.weixin.qq.com/s?__biz=MzUyNDkzNzczNQ%3D%3D&chksm=fa275cbbcd50d5adfa967c4a6034579f70a521510055c34a23735468341965afaa82c4fa59fe&idx=1&mid=2247508947&scene=21&sn=7622feaff76eecc1cedb6c002b5fc5d5#wechat_redirect
[MySQL Online DDL]: http://mp.weixin.qq.com/s?__biz=MzUyNDkzNzczNQ%3D%3D&chksm=fa275cbbcd50d5ad0ee6bb62891b5c65af87aca50277bc6fe744bc9f5b6ff81d00948f21a149&idx=2&mid=2247508947&scene=21&sn=4a61695fe1269750c2fd3f25fc11b5ac#wechat_redirect
[7946e9f74e18d290530991d12a2578d1.gif]: /images/20221005/d641b6ecaecf417b99fedc59b5eab999.png