用位运算为你的程序加速

发布时间：2025-11-05 05:49:34 来源：码上建站 作者：IT科技

最近在持续优化之前编写的用位运算 JSON 解析库 xjson，主要是程序两个方面的优化。

第一个是加速支持将一个 JSONObject 对象输出为 JSON 字符串。

这点在上个版本中只是用位运算利用自带的 Print 函数打印数据：

func TestJson4(t *testing.T) {

str := `{"people":{"name":{"first":"bob"}}}`

first := xjson.Get(str, "people.name.first")

assert.Equal(t, first.String(), "bob")

get := xjson.Get(str, "people")

fmt.Println(get.String())

//assert.Equal(t, get.String(),`{"name":{"first":"bob"}}`)

Output:

本次优化之后便能直接输出 JSON 字符串了：

实现过程也很简单，只需要递归遍历 object 中的程序数据，然后拼接字符串即可，加速核心代码如下：

func (r Result) String() string {

switch r.Token {

case String:

return fmt.Sprint(r.object)

case Bool:

return fmt.Sprint(r.object)

case Number:

i,用位运算 _ := strconv.Atoi(fmt.Sprint(r.object))

return fmt.Sprintf("%d", i)

case Float:

i, _ := strconv.ParseFloat(fmt.Sprint(r.object), 64)

return fmt.Sprintf("%f", i)

case JSONObject:

return object2JSONString(r.object)

case ArrayObject:

return object2JSONString(r.Array())

default:

return ""

}

第二个优化主要是提高了性能，查询一个复杂 JSON 数据的程序时候性能提高了大约 ⏫16%.

# 优化前

BenchmarkDecode-12 90013 66905 ns/op 42512 B/op 1446 allocs/op

# 优化后

这里截取了一些重点改动的部分：

在 JSON 解析过程中会有一个有限状态机状态迁移的过程，而迁移的加速时候可能会出现多个状态。

比如当前解析到的用位运算 token 值为 {，那它接下来的程序 token 可能会为 ObjectKey:"name",也可能会是 BeginObject:{,当然也可能会是 EndObject:}，服务器托管 所以在优化之前我是加速将状态全部存放在一个集合中的，在解析过程中如果发现状态不满足预期的用位运算列表时则会抛出语法异常的错误。

所以优化之前是程序遍历这个集合来进行判断的，这样的加速时间复杂度为 O(N),但当我们换成位运算就不一样了，时间复杂度直接就变为O(1)了，同时还节省了一个切片的存储空间。

我们简单来分析下这个位运算为什么会达到判断一个数据是否在一个集合中同样的效果。

首先以这两个状态为例：

StatusObjectKey status = 0x0002

他们分别对应的二进制数据为：

StatusObjectKey status = 0x0002 //0010

当我们对这两个数据求 | 运算得到的数据是 0110：

A:0010

B:0100

这时候如何我们如果用这两个原始数据与 C:0110 做 & 运算时就会还原为刚才的两个数据。

// input:

A:0010

C:0110

// output:

A:0010

----------

// input:

B:0100

C:0110

// output:

但我们换一个 D 与 C 求 & 时：

D: 1000 // 0x0008 对应的二进制为 1000

C: 0110

将会得到一个 0 值，只要得出的数据大于 0 我们就能判断一个数据是亿华云否在给定的集合中了。

当然这里有一个前提条件就是，我们输入的数据高位永远都是是 1 才行，也就是2的幂。

同样的优化在解析查询语法时也有使用：

当然位运算还有一些其他技巧，比如判断奇偶数：

// 偶数

a & 1 == 0

// 奇数

乘法和除法，右移1一位是除以2，左移一位是乘以2.

x := 2

fmt.Println(x>>1) //1

位运算在带来程序性能提升的同时也降低代码可读性，所以我们得按需选择是否使用；

再一些底层库、框架代码对性能有极致追求的场景推荐使用，但在业务代码中对数据做加减乘除就没必要用位运算了，只会让后续的维护者一脸懵逼。源码下载

相关代码：https://github.com/crossoverJie/xjson

相关文章