文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

深入浅出 MySQL 优先队列

2024-12-03 01:01

关注

本文转载自微信公众号「Java课代表」,作者Java课代表 。转载本文请联系Java课代表公众号。

本文适用于 MySQL 5.6 及以上版本

0.先抛问题

假设字段category无索引且有重复值,order by category 和limit组合使用的结果会和预期不符。

问题复现:

表结构(就是两个字段)

  1. CREATE TABLE `ratings` ( 
  2.   `id` int(11) NOT NULL AUTO_INCREMENT, 
  3.   `category` int(11) DEFAULT NULL
  4.   PRIMARY KEY (`id`) 
  5. ) ENGINE=InnoDB AUTO_INCREMENT=11 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci; 

对所有数据按category字段排序:select * from ratings order by category;

id category
1 1
5 1
10 1
3 2
4 2
6 2
9 2
2 3
7 3
8 3

当我们想分页展示前5条时使用select * from ratings order by category limit 5;

期望得到的ID顺序是1 5 10 3 4。

但实际结果如下:

id category
1 1
10 1
5 1
3 2
4 2

怎么肥似?MySQL 出 Bug 了?

可能有同学遇到过这个问题,百度或谷歌一下解决了,你有没有想过,你查到的办法是最优解吗?别人是怎么得出这个办法的?MySQL 为什么会这样做,跟版本有关吗?

先抛结论:

  1. 最优解是后面再加个列值唯一的排序字段,如:order by category,id
  2. MySQL 为什么这样做?答案是为了快!(MySQL 5.6及其之后才有此优化)
  3. 次优解是对order by后面的category 加索引(为什么是次优解?看完本文你将会有答案)
  4. 下面课代表将还原一下这 3 条结论的产出过程。

1. 最优解

MySQL 文档 8.2.1.19 LIMIT Query Optimization 中对此场景有如下描述:

总结来说就是:

当 ORDER BY 列的字段值存在重复,那么这条 ORDER BY 语句返回的数据顺序会因为LIMIT的存在而变得不一样

这是 MySQL 默认对该场景做的优化,如果你需要保证加不加 LIMIT 顺序都要一致,官方也给出了办法:

就是在ORDER BY 后面再多加一个排序字段(比如 ID 字段)。

以上描述最早出现在MySQL 5.6文档中,从这个版本开始,引入了这个针对ORDER BY LIMIT的优化。

好了, 针对文中的场景,我们只需要select * from ratings order by category,id;即可解决。

那么问题来了,MySQL 为什么要做这么一个看似是 Bug 的优化?

2.MySQL 的 ORDER BY 逻辑

顾名思义,ORDER BY 就是排序。

执行一下explain select * from ratings order by category limit 5;

  1. *************************** 1. row *************************** 
  2.            id: 1 
  3.   select_type: SIMPLE 
  4.         table: ratings 
  5.    partitions: NULL 
  6.          type: ALL 
  7. possible_keys: NULL 
  8.           keyNULL 
  9.       key_len: NULL 
  10.           ref: NULL 
  11.          rows: 10 
  12.      filtered: 100.00 
  13.         Extra: Using filesort 
  14. 1 row in set, 1 warning (0.00 sec) 

可以看到 Extra: Using filesort 表示需要排序。

正常情况下, MySQL 会有内存排序和外部排序两种:

很明显,这两种排序都是对所有结果全部排序,讲道理,不管有没有LIMIT,都是从排完序的结果中按顺序取需要的条数,有没有LIMIT是不会影响返回的结果顺序的。

但是,MySQL 5.6 版本针对 ORDER BY LIMIT做了个小优化(排序字段无索引,且列值不唯一时):优化器在遇到 ORDER BY LIMIT语句的时候,使用了priority queue。

filesort.cc 中有如下伪代码描述该优化:

  1. while (get_next_sortkey()) 
  2.    { 
  3.      if (using priority queue) 
  4.        push sort key into queue 
  5.      else 
  6.      { 
  7.        try to put sort key into buffer; 
  8.        if (no free space in sort buffer) 
  9.        { 
  10.          do { 
  11.            allocate new, larger buffer; 
  12.            retry putting sort key into buffer; 
  13.          } until (record fits or no space for new buffer) 
  14.          if (no space for new buffer) 
  15.          { 
  16.            sort record pointers (all buffers); 
  17.            dump sorted sequence to 'tempfile'
  18.            dump Merge_chunk describing sequence location into 'chunk_file'
  19.          } 
  20.        } 
  21.        if (key was packed) 
  22.          tell sort buffer the actual number of bytes used; 
  23.      } 
  24.    } 
  25.    if (buffer has some elements && dumped at least once) 
  26.      sort-dump-dump as above; 
  27.    else 
  28.      don't sort, leave sort buffer to be sorted by caller. 

并在 WL#1393: Optimizing filesort with small limit 中阐述了该优化逻辑:

  1. Many web customers have to do 
  2. "SELECT ... ORDER BY non_index_column LIMIT X"
  3.  
  4. When X *  is smaller than sort_buff_size we can use 
  5. the following algoritm to speed up the sort: 
  6.  
  7. Create a queue to hold 'limit' keys. 
  8. - Scan through the table and store the first (last if DESC) keys in the queue 
  9. Return values from queue 
  10.  
  11. This is much faster than the current algoritm that works as

该 WorkLog 中记录了优化后的效果:10 to 20 times faster than a quicksort(感兴趣的同学可以去阅读原文)。

所以,就是为了快!

MySQL 认为这种场景就是求 TOP N 的问题,使用 priority queue 就能解决。

3.priority queue(优先级队列)

priority queue 其实就是堆,Java 中有java.util.PriorityQueue类,其本质就是 [堆] 这种数据结构。

简单解释一下什么是堆:

堆是一个完全二叉树;

堆中每一个节点的值都必须大于等于(大顶堆)或小于等于(小顶堆)其子树中每个节点的值。

如果 MySQL 使用归并或快排,需要把所有数据都排好序,再取LIMIT 的前几条,剩余已排序的数据就白白浪费了。

而采用 priority queue 可以根据 LIMIT的条数维护一个堆,只需要把所有数据在这个堆里过一遍就能得到结果。

使用如下语句可以验证 MySQL 使用了 priority queue:

  1. SET optimizer_trace='enabled=on'
  2. select * from ratings order by category limit 5; 
  3. SELECT * FROM `information_schema`.`OPTIMIZER_TRACE`\G; 
  1. "filesort_priority_queue_optimization": { 
  2.              "limit": 5, 
  3.              "chosen"true 
  4.            }, 

可以看到 filesort_priority_queue_optimization.chosen = true

下面用流程图还原一下 priority queue 的执行逻辑(以LIMIT 5为例):

友情提示:图中的小顶堆以 category 值的大小排序

取前五条数据构成一个小顶堆:

取下一行数据(6,2),发现 2 小于当前堆中最大的category 3,于是把(2,3)从堆中删掉,把(6,2) 入堆:

重复步骤 2,直至符合查询条件的数据都经历过比较入堆,最终堆中数据如图:

以上就是通过 priority queue 找到 最小的 5 行 category 数据的执行过程。

最后我们将其出堆即可得到结果,每次出堆最小元素后将最后一个元素放入堆顶,按照小顶堆重新堆化,过程如图:

可以看到,这个结果和select * from ratings order by category limit 5;的输出一致

4.加索引为什么是次优解

显然,按照ORDER BY 的逻辑,直接对排序字段加索引也可以省去内存排序步骤,从而解决这个问题。

但索引也不是银弹,多出来的category索引会增加表的维护成本,如果没有明显的业务需要,单纯为了绕过这个priority queue的优化而加索引,课代表认为有点得不偿失。

尤其是当表数据量非常大的时候,索引的体量会很可观。而且,针对文中场景,category作为分类字段,重复率会比较高,即使有按分类查询的业务 SQL ,MySQL 也不一定会选取这条索引。

综上,针对本场景,个人认为order by category,id才是该问题的最优解。

PS:会不会有人问:关我鸟事,我从没写过带 LIMIT 的 SQL 啊!

难道你写的 CRUD 功能都不带分页的吗?PageHelper 源码去了解一下?

5. 总结

本文案例是课代表上线过程中遭遇到的实际问题,咨询了下周围同学,有好几个都遇到过此问题,网上文章大多浅入浅出,读完有隔靴搔痒之感,无法解答心中疑惑。遂整理此文。

其中涉及 数据结构,PageHelper,MySQL 文档,相关参考资料罗列在文末,如果有时间能顺着文章思路亲自读一遍参考文档,相信会有更深的收获。

6.参考资料:

《数据结构与算法之美》---王争 第 28,29 讲

《MySQL实战45讲》---林晓斌 第 04、05、10、16、17 讲

2.1.16 LIMIT Query Optimization---https://dev.mysql.com/doc/refman/5.6/en/limit-optimization.html

MySQL · 答疑解惑 · MySQL Sort 分页---http://mysql.taobao.org/monthly/2015/06/04/

filesort.cc---https://dev.mysql.com/doc/dev/mysql-server/8.0.18/filesort_8cc.html

WL#1393: Optimizing filesort with small limit---https://dev.mysql.com/worklog/task/?id=1393

来源:Java课代表内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯