应用 5:层峦叠嶂 —— 布隆过滤器

Redis 中的布隆过滤器

Redis 官方提供的布隆过滤器到了 Redis 4.0 提供了插件功能之后才正式登场。布隆过滤器作为一个插件加载到 Redis Server 中,给 Redis 提供了强大的布隆去重功能。

下面我们来体验一下 Redis 4.0 的布隆过滤器,为了省去繁琐安装过程,我们直接用 Docker 吧。

  1. > docker pull redislabs/rebloom # 拉取镜像
  2. > docker run -p 6379:6379 redislabs/rebloom # 运行容器
  3. > redis-cli # 连接容器中的 redis 服务

如果上面三条指令执行没有问题,下面就可以体验布隆过滤器了。

布隆过滤器基本使用

布隆过滤器有二个基本指令,bf.add 添加元素,bf.exists 查询元素是否存在,它的用法和 set 集合的 sadd 和 sismember 差不多。注意 bf.add 只能一次添加一个元素,如果想要一次添加多个,就需要用到 bf.madd 指令。同样如果需要一次查询多个元素是否存在,就需要用到 bf.mexists 指令。

  1. 127.0.0.1:6379> bf.add codehole user1
  2. (integer) 1
  3. 127.0.0.1:6379> bf.add codehole user2
  4. (integer) 1
  5. 127.0.0.1:6379> bf.add codehole user3
  6. (integer) 1
  7. 127.0.0.1:6379> bf.exists codehole user1
  8. (integer) 1
  9. 127.0.0.1:6379> bf.exists codehole user2
  10. (integer) 1
  11. 127.0.0.1:6379> bf.exists codehole user3
  12. (integer) 1
  13. 127.0.0.1:6379> bf.exists codehole user4
  14. (integer) 0
  15. 127.0.0.1:6379> bf.madd codehole user4 user5 user6
  16. 1) (integer) 1
  17. 2) (integer) 1
  18. 3) (integer) 1
  19. 127.0.0.1:6379> bf.mexists codehole user4 user5 user6 user7
  20. 1) (integer) 1
  21. 2) (integer) 1
  22. 3) (integer) 1
  23. 4) (integer) 0

似乎很准确啊,一个都没误判。下面我们用 Python 脚本加入很多元素,看看加到第几个元素的时候,布隆过滤器会出现误判。

  1. # coding: utf-8
  2. import redis
  3. client = redis.StrictRedis()
  4. client.delete("codehole")
  5. for i in range(100000):
  6. client.execute_command("bf.add", "codehole", "user%d" % i)
  7. ret = client.execute_command("bf.exists", "codehole", "user%d" % i)
  8. if ret == 0:
  9. print i
  10. break

Java 客户端 Jedis-2.x 没有提供指令扩展机制,所以你无法直接使用 Jedis 来访问 Redis Module 提供的 bf.xxx 指令。RedisLabs 提供了一个单独的包 JReBloom,但是它是基于 Jedis-3.0,Jedis-3.0 这个包目前还没有进入 release,没有进入 maven 的中央仓库,需要在 Github 上下载。在使用上很不方便,如果怕麻烦,还可以使用 lettuce,它是另一个 Redis 的客户端,相比 Jedis 而言,它很早就支持了指令扩展。

  1. public class BloomTest {
  2. public static void main(String[] args) {
  3. Client client = new Client();
  4. client.delete("codehole");
  5. for (int i = 0; i < 100000; i++) {
  6. client.add("codehole", "user" + i);
  7. boolean ret = client.exists("codehole", "user" + i);
  8. if (!ret) {
  9. System.out.println(i);
  10. break;
  11. }
  12. }
  13. client.close();
  14. }
  15. }

执行上面的代码后,你会张大了嘴巴发现居然没有输出,塞进去了 100000 个元素,还是没有误判,这是怎么回事?如果你不死心的话,可以将数字再加一个 0 试试,你会发现依然没有误判。

原因就在于布隆过滤器对于已经见过的元素肯定不会误判,它只会误判那些没见过的元素。所以我们要稍微改一下上面的脚本,使用 bf.exists 去查找没见过的元素,看看它是不是以为自己见过了。

  1. # coding: utf-8
  2. import redis
  3. client = redis.StrictRedis()
  4. client.delete("codehole")
  5. for i in range(100000):
  6. client.execute_command("bf.add", "codehole", "user%d" % i)
  7. # 注意 i+1,这个是当前布隆过滤器没见过的
  8. ret = client.execute_command("bf.exists", "codehole", "user%d" % (i+1))
  9. if ret == 1:
  10. print i
  11. break

Java 版:

  1. public class BloomTest {
  2. public static void main(String[] args) {
  3. Client client = new Client();
  4. client.delete("codehole");
  5. for (int i = 0; i < 100000; i++) {
  6. client.add("codehole", "user" + i);
  7. boolean ret = client.exists("codehole", "user" + (i + 1));
  8. if (ret) {
  9. System.out.println(i);
  10. break;
  11. }
  12. }
  13. client.close();
  14. }
  15. }

运行后,我们看到了输出是 214,也就是到第 214 的时候,它出现了误判。

那如何来测量误判率呢?我们先随机出一堆字符串,然后切分为 2 组,将其中一组塞入布隆过滤器,然后再判断另外一组的字符串存在与否,取误判的个数和字符串总量一半的百分比作为误判率。

  1. # coding: utf-8
  2. import redis
  3. import random
  4. client = redis.StrictRedis()
  5. CHARS = ''.join([chr(ord('a') + i) for i in range(26)])
  6. def random_string(n):
  7. chars = []
  8. for i in range(n):
  9. idx = random.randint(0, len(CHARS) - 1)
  10. chars.append(CHARS[idx])
  11. return ''.join(chars)
  12. users = list(set([random_string(64) for i in range(100000)]))
  13. print 'total users', len(users)
  14. users_train = users[:len(users)/2]
  15. users_test = users[len(users)/2:]
  16. client.delete("codehole")
  17. falses = 0
  18. for user in users_train:
  19. client.execute_command("bf.add", "codehole", user)
  20. print 'all trained'
  21. for user in users_test:
  22. ret = client.execute_command("bf.exists", "codehole", user)
  23. if ret == 1:
  24. falses += 1
  25. print falses, len(users_test)

Java 版:

  1. public class BloomTest {
  2. private String chars;
  3. {
  4. StringBuilder builder = new StringBuilder();
  5. for (int i = 0; i < 26; i++) {
  6. builder.append((char) ('a' + i));
  7. }
  8. chars = builder.toString();
  9. }
  10. private String randomString(int n) {
  11. StringBuilder builder = new StringBuilder();
  12. for (int i = 0; i < n; i++) {
  13. int idx = ThreadLocalRandom.current().nextInt(chars.length());
  14. builder.append(chars.charAt(idx));
  15. }
  16. return builder.toString();
  17. }
  18. private List<String> randomUsers(int n) {
  19. List<String> users = new ArrayList<>();
  20. for (int i = 0; i < 100000; i++) {
  21. users.add(randomString(64));
  22. }
  23. return users;
  24. }
  25. public static void main(String[] args) {
  26. BloomTest bloomer = new BloomTest();
  27. List<String> users = bloomer.randomUsers(100000);
  28. List<String> usersTrain = users.subList(0, users.size() / 2);
  29. List<String> usersTest = users.subList(users.size() / 2, users.size());
  30. Client client = new Client();
  31. client.delete("codehole");
  32. for (String user : usersTrain) {
  33. client.add("codehole", user);
  34. }
  35. int falses = 0;
  36. for (String user : usersTest) {
  37. boolean ret = client.exists("codehole", user);
  38. if (ret) {
  39. falses++;
  40. }
  41. }
  42. System.out.printf("%d %d\n", falses, usersTest.size());
  43. client.close();
  44. }
  45. }

运行一下,等待大约一分钟,输出:

  1. total users 100000
  2. all trained
  3. 628 50000

可以看到误判率大约 1% 多点。你也许会问这个误判率还是有点高啊,有没有办法降低一点?答案是有的。

我们上面使用的布隆过滤器只是默认参数的布隆过滤器,它在我们第一次 add 的时候自动创建。Redis 其实还提供了自定义参数的布隆过滤器,需要我们在 add 之前使用bf.reserve指令显式创建。如果对应的 key 已经存在,bf.reserve会报错。bf.reserve有三个参数,分别是 key, error_rateinitial_size。错误率越低,需要的空间越大。initial_size参数表示预计放入的元素数量,当实际数量超出这个数值时,误判率会上升。

所以需要提前设置一个较大的数值避免超出导致误判率升高。如果不使用 bf.reserve,默认的error_rate是 0.01,默认的initial_size是 100。

接下来我们使用 bf.reserve 改造一下上面的脚本:

  1. # coding: utf-8
  2. import redis
  3. import random
  4. client = redis.StrictRedis()
  5. CHARS = ''.join([chr(ord('a') + i) for i in range(26)])
  6. def random_string(n):
  7. chars = []
  8. for i in range(n):
  9. idx = random.randint(0, len(CHARS) - 1)
  10. chars.append(CHARS[idx])
  11. return ''.join(chars)
  12. users = list(set([random_string(64) for i in range(100000)]))
  13. print 'total users', len(users)
  14. users_train = users[:len(users)/2]
  15. users_test = users[len(users)/2:]
  16. falses = 0
  17. client.delete("codehole")
  18. # 增加了下面这一句
  19. client.execute_command("bf.reserve", "codehole", 0.001, 50000)
  20. for user in users_train:
  21. client.execute_command("bf.add", "codehole", user)
  22. print 'all trained'
  23. for user in users_test:
  24. ret = client.execute_command("bf.exists", "codehole", user)
  25. if ret == 1:
  26. falses += 1
  27. print falses, len(users_test)

Java 版本:

  1. public class BloomTest {
  2. private String chars;
  3. {
  4. StringBuilder builder = new StringBuilder();
  5. for (int i = 0; i < 26; i++) {
  6. builder.append((char) ('a' + i));
  7. }
  8. chars = builder.toString();
  9. }
  10. private String randomString(int n) {
  11. StringBuilder builder = new StringBuilder();
  12. for (int i = 0; i < n; i++) {
  13. int idx = ThreadLocalRandom.current().nextInt(chars.length());
  14. builder.append(chars.charAt(idx));
  15. }
  16. return builder.toString();
  17. }
  18. private List<String> randomUsers(int n) {
  19. List<String> users = new ArrayList<>();
  20. for (int i = 0; i < 100000; i++) {
  21. users.add(randomString(64));
  22. }
  23. return users;
  24. }
  25. public static void main(String[] args) {
  26. BloomTest bloomer = new BloomTest();
  27. List<String> users = bloomer.randomUsers(100000);
  28. List<String> usersTrain = users.subList(0, users.size() / 2);
  29. List<String> usersTest = users.subList(users.size() / 2, users.size());
  30. Client client = new Client();
  31. client.delete("codehole");
  32. // 对应 bf.reserve 指令
  33. client.createFilter("codehole", 50000, 0.001);
  34. for (String user : usersTrain) {
  35. client.add("codehole", user);
  36. }
  37. int falses = 0;
  38. for (String user : usersTest) {
  39. boolean ret = client.exists("codehole", user);
  40. if (ret) {
  41. falses++;
  42. }
  43. }
  44. System.out.printf("%d %d\n", falses, usersTest.size());
  45. client.close();
  46. }
  47. }

运行一下,等待约 1 分钟,输出如下:

  1. total users 100000
  2. all trained
  3. 6 50000

我们看到了误判率大约 0.012%,比预计的 0.1% 低很多,不过布隆的概率是有误差的,只要不比预计误判率高太多,都是正常现象。

注意事项

布隆过滤器的initial_size估计的过大,会浪费存储空间,估计的过小,就会影响准确率,用户在使用之前一定要尽可能地精确估计好元素数量,还需要加上一定的冗余空间以避免实际元素可能会意外高出估计值很多。

布隆过滤器的error_rate越小,需要的存储空间就越大,对于不需要过于精确的场合,error_rate设置稍大一点也无伤大雅。比如在新闻去重上而言,误判率高一点只会让小部分文章不能让合适的人看到,文章的整体阅读量不会因为这点误判率就带来巨大的改变。

布隆过滤器的原理

学会了布隆过滤器的使用,下面有必要把原理解释一下,不然读者还会继续蒙在鼓里

应用5:Redis 中的布隆过滤器 - 图1

每个布隆过滤器对应到 Redis 的数据结构里面就是一个大型的位数组和几个不一样的无偏 hash 函数。所谓无偏就是能够把元素的 hash 值算得比较均匀。

向布隆过滤器中添加 key 时,会使用多个 hash 函数对 key 进行 hash 算得一个整数索引值然后对位数组长度进行取模运算得到一个位置,每个 hash 函数都会算得一个不同的位置。再把位数组的这几个位置都置为 1 就完成了 add 操作。

向布隆过滤器询问 key 是否存在时,跟 add 一样,也会把 hash 的几个位置都算出来,看看位数组中这几个位置是否都为 1,只要有一个位为 0,那么说明布隆过滤器中这个 key 不存在。如果都是 1,这并不能说明这个 key 就一定存在,只是极有可能存在,因为这些位被置为 1 可能是因为其它的 key 存在所致。如果这个位数组比较稀疏,判断正确的概率就会很大,如果这个位数组比较拥挤,判断正确的概率就会降低。具体的概率计算公式比较复杂,感兴趣可以阅读扩展阅读,非常烧脑,不建议读者细看。

使用时不要让实际元素远大于初始化大小,当实际元素开始超出初始化大小时,应该对布隆过滤器进行重建,重新分配一个 size 更大的过滤器,再将所有的历史元素批量 add 进去 (这就要求我们在其它的存储器中记录所有的历史元素)。因为 error_rate 不会因为数量超出就急剧增加,这就给我们重建过滤器提供了较为宽松的时间。

空间占用估计

布隆过滤器的空间占用有一个简单的计算公式,但是推导比较繁琐,这里就省去推导过程了,直接引出计算公式,感兴趣的读者可以点击「扩展阅读」深入理解公式的推导过程。

布隆过滤器有两个参数,第一个是预计元素的数量 n,第二个是错误率 f。公式根据这两个输入得到两个输出,第一个输出是位数组的长度 l,也就是需要的存储空间大小 (bit),第二个输出是 hash 函数的最佳数量 k。hash 函数的数量也会直接影响到错误率,最佳的数量会有最低的错误率。

  1. k=0.7*(l/n) # 约等于
  2. f=0.6185^(l/n) # ^ 表示次方计算,也就是 math.pow

从公式中可以看出

  1. 位数组相对越长 (l/n),错误率 f 越低,这个和直观上理解是一致的
  2. 位数组相对越长 (l/n),hash 函数需要的最佳数量也越多,影响计算效率
  3. 当一个元素平均需要 1 个字节 (8bit) 的指纹空间时 (l/n=8),错误率大约为 2%
  4. 错误率为 10%,一个元素需要的平均指纹空间为 4.792 个 bit,大约为 5bit
  5. 错误率为 1%,一个元素需要的平均指纹空间为 9.585 个 bit,大约为 10bit
  6. 错误率为 0.1%,一个元素需要的平均指纹空间为 14.377 个 bit,大约为 15bit

你也许会想,如果一个元素需要占据 15 个 bit,那相对 set 集合的空间优势是不是就没有那么明显了?这里需要明确的是,set 中会存储每个元素的内容,而布隆过滤器仅仅存储元素的指纹。元素的内容大小就是字符串的长度,它一般会有多个字节,甚至是几十个上百个字节,每个元素本身还需要一个指针被 set 集合来引用,这个指针又会占去 4 个字节或 8 个字节,取决于系统是 32bit 还是 64bit。而指纹空间只有接近 2 个字节,所以布隆过滤器的空间优势还是非常明显的。

如果读者觉得公式计算起来太麻烦,也没有关系,有很多现成的网站已经支持计算空间占用的功能了,我们只要把参数输进去,就可以直接看到结果,比如 布隆计算器

应用5:Redis 中的布隆过滤器 - 图2

实际元素超出时,误判率会怎样变化

当实际元素超出预计元素时,错误率会有多大变化,它会急剧上升么,还是平缓地上升,这就需要另外一个公式,引入参数 t 表示实际元素和预计元素的倍数 t

  1. f=(1-0.5^t)^k # 极限近似,k 是 hash 函数的最佳数量

当 t 增大时,错误率,f 也会跟着增大,分别选择错误率为 10%,1%,0.1% 的 k 值,画出它的曲线进行直观观察

应用5:Redis 中的布隆过滤器 - 图3

从这个图中可以看出曲线还是比较陡峭的

  1. 错误率为 10% 时,倍数比为 2 时,错误率就会升至接近 40%,这个就比较危险了
  2. 错误率为 1% 时,倍数比为 2 时,错误率升至 15%,也挺可怕的
  3. 错误率为 0.1%,倍数比为 2 时,错误率升至 5%,也比较悬了

用不上 Redis4.0 怎么办?

Redis 4.0 之前也有第三方的布隆过滤器 lib 使用,只不过在实现上使用 redis 的位图来实现的,性能上也要差不少。比如一次 exists 查询会涉及到多次 getbit 操作,网络开销相比而言会高出不少。另外在实现上这些第三方 lib 也不尽完美,比如 pyrebloom 库就不支持重连和重试,在使用时需要对它做一层封装后才能在生产环境中使用。

  1. Python Redis Bloom Filter
  2. Java Redis Bloom Filter

布隆过滤器的其它应用

在爬虫系统中,我们需要对 URL 进行去重,已经爬过的网页就可以不用爬了。但是 URL 太多了,几千万几个亿,如果用一个集合装下这些 URL 地址那是非常浪费空间的。这时候就可以考虑使用布隆过滤器。它可以大幅降低去重存储消耗,只不过也会使得爬虫系统错过少量的页面。

布隆过滤器在 NoSQL 数据库领域使用非常广泛,我们平时用到的 HBase、Cassandra 还有 LevelDB、RocksDB 内部都有布隆过滤器结构,布隆过滤器可以显著降低数据库的 IO 请求数量。当用户来查询某个 row 时,可以先通过内存中的布隆过滤器过滤掉大量不存在的 row 请求,然后再去磁盘进行查询。

邮箱系统的垃圾邮件过滤功能也普遍用到了布隆过滤器,因为用了这个过滤器,所以平时也会遇到某些正常的邮件被放进了垃圾邮件目录中,这个就是误判所致,概率很低。

扩展阅读

布隆过滤器的原理涉及到较为复杂的数学知识,感兴趣可以阅读下面的链接文章继续深入了解内部原理:布隆过滤器

同样,如果你是个数学学渣,那老师我建议你谨慎观看,要注意保护好自己的 24K 钛合金狗眼,避免闪瞎。