10 亿数据量只需要 100MB 内存,Redis 的位存储为什么这么牛?
来源:https://www.toutiao.com/i6767642839267410445
本文主要和大家分享一下 redis 的高级特性:bit 位操作。
力求让大家彻底学会使用 redis 的 bit 位操作并掌握其底层实现原理!主要包含以下内容:
redis 位操作命令示例
底层数据结构分析
为什么他的算法时间复杂度是 O(1)?
10 亿数据量需要多大的存储空间?
redis 位操作适合哪些应用场景?
文章内容较长,建议大家收藏后持续阅读,点击右上角关注,获取更多技术干货文章!
本文 redis 试验代码基于如下环境:
操作系统:Mac OS 64 位
版本:Redis 5.0.7 64 bit
运行模式:standalone mode
redis 位操作
reids 位操作也叫位数组操作、bitmap,它提供了 SETBIT、GETBIT、BITCOUNT、BITTOP 四个命令用于操作二进制位数组。
先来看一波基本操作示例:

SETBIT
语法:
即:命令 key 偏移量 0/1
setbit 命令用于写入位数组指定偏移量的二进制位设置值,偏移量从 0 开始计数,且只允许写入 1 或者 0,如果写入非 0 和 1 的值则写入失败:

GETBIT
语法:
即:命令 key 偏移量
gitbit 命令用于获取位数组指定偏移量上的二进制值:

BITCOUNT
语法:
即:命令 key
bitcount 命令用于获取指定 key 的位数组中值为 1 的二进制位的数量,之前我们写入了偏移量 0 的值为 1,偏移量 10 的值为 1,偏移量 8 的值为 0:

BITOP
语法:
即:命令 操作 结果目标 key key1 key2 ...
bitop 命令可以对多个位数组的 key 进行 and(按位与)、or(按位或)、xor(按位异或)运算,并将运算结果设置到 destkey 中:

底层数据结构分析
SDS 是 redis 中的一种数据结构,叫做简单动态字符串(Simple Dynamic String),并且它是一种二进制安全的,在大多数的情况下 redis 中的字符串都用 SDS 来存储。
SDS 的数据结构:
struct sdshdr {
#记录buff数组中已使用字节的数量
#也是SDS所保存字符串的长度
int len;
#记录buff数组中未使用字节的数量
int free;
#字节数组,字符串就存储在这个数组里
char buff[];
}
数据存储示例:

SDS 的优点:
时间复杂度为 O(1)
杜绝缓冲区溢出
减少修改字符串长度时候所需的内存重分配次数
二进制安全的 API 操作
兼容部分 C 字符串函数
redis 中的位数组采用的是 String 字符串数据格式来存储,而字符串对象使用的正是上文说的 SDS 简单动态字符串数据结构。

大家都知道的是一个字节用的是 8 个二进制位来存储的,也就是 8 个 0 或者 1,即一个字节可以存储十进制 0~127 的数字,也即包含了所有的数字、英文大小写字母以及标点符号。
1Byte=8bit
1KB=1024Byte
1MB=1024KB
1GB=1024MB
位数组在 redis 存储世界里,每一个字节也是 8 位,初始都是:
0 0 0 0 0 0 0 0
而位操作就是在对应的 offset 偏移量上设置 0 或者 1,比如将第 3 位设置为 1,即:
0 0 0 0 1 0 0 0
#对应redis操作即:
setbit key 3 1
在此基础上,如果要在偏移量为 13 的位置设置 1,即:
setbit key 13 1
#对应redis中的存储为:
0 0 1 0 | 0 0 0 0 | 0 0 0 0 | 1 0 0 0
时间复杂度
GETBIT 命令时间复杂度 O(1)
STEBIT 命令时间复杂度 O(1)
BITCOUNT 命令时间复杂度 O(n)
BITOP 命令时间复杂度 O(n)、O(n2)
我们来看 GETBIT 以及 SETBIT 命令的时间复杂度为什么是 O(1),当我们执行一个 SETBIT key 10086 1 的值的时候,reids 的计算方式如下:
获取到要写入位数组中的哪个字节:10086÷8=1260,需要写入到位数组的下标 1260 的字节
获取要写入到这个字节的第几位:10086 mod 8 = 6,需要写入到这个字节的下标为 6 即第 7 位上去。
通过这两种计算方式大家可以清晰的看到,位操作的 GETBIT 和 SETBIT 都是常量计算,因此它的时间复杂度为 O(1)。
而 BITCOUNT 命令需要对整个位数组的所有元素进行遍历算出值为 1 的有多少个,当然 redis 对于大数据了的 bit 执行 bitcount 命令会有一整套复杂的优化的算法,但是核心思路还是这个意思,无非是减少部分遍历查询次数。比如以 128 位为一次遍历,那么他的遍历次数就是所有的位数除以 128。
BITTOP 命令则是根据不同的操作有不同的执行方式。比如 AND 操作,则需要查看位值为 1 的即可。
存储空间计算
根据上面的介绍,相信大家已经知道了基于 redis 的位数组数据结构存储的数据占用内存大小是怎么计算的了。比如有 100 亿的数据,那么它需要的字节数组:
1000000000÷8÷1024÷1024≈119.21MB
也就是存储 10 亿的数据只需要 119MB 左右的内存空间,这对于现在动辄 16G、32G 集群版的 redis,完全没有问题。
需要注意的是,如果你的数据量不大,那就不要把起始偏移量搞的很大,这样也是占空间的,比如我们只需要存储几百条数据,但是其中的偏移量却很大,这就会造成了很大的内存空间浪费。
应用场景
实际项目开发中有很多业务都适合采用 redis 的 bit 来实现。
用户签到场景
每天的日期字符串作为一个 key,用户 Id 作为 offset,统计每天用户的签到情况,总的用户签到数
活跃用户数统计
用户日活、月活、留存率等均可以用 redis 位数组来存储,还是以每天的日期作为 key,用户活跃了就写入 offset 为用户 id 的位值 1。
同理月活也是如此。
用户是否在线以及总在线人数统计
同样是使用一个位数组,用户的 id 映射偏移量,在线标识为 1,下线标识为 0。即可实现用户上下线查询和总在线人数的统计
APP 内用户的全局消息提示小红点
现在大多数的 APP 里都有站内信的功能,当有消息的时候,则提示一个小红点,代表用户有新的消息。