Laravel Query Builder 复杂查询案例：子查询实现分区查询 partition by

案例

案例：Laravel 在文章列表中附带上前10条评论?，在获取文章列表时同时把每一个文章的前10条评论一同查询出来。

这是典型分区查询案例，须要根据 comments 表中的 post_id 字段进行分区，同时根据条件进行排序，把符合条件的前 N 条是数据取出来。

在其余数据库(Oracle, SQL Server，Vertica) 包含了 row_number partition by 这样的函数，可以比较容易的实现。

好比在 SQL Server 中:

SELECT * FROM (
SELECT *, row_number() OVER (partition by post_id ORDER BY created_at desc) rank FROM comments where post_id in (1,2,3,4,5) 
) b where rand < 11;

在 mysql 中要复杂一些，咱们先来看看上面案例中实现需求的几种解决办法。

解决办法

方法1：

在 blade 中要显示评论数据的地方 post->comments()->limit(10)

问题：若是取了 20 条 Post 数据，就会有 20 条取 comments 的 sql 语句，会形成执行的 sql 语句过多。

不是很是可取，主要问题会形成 SQL 语句过多，对数据库服务器产生压力，不过这里可使用缓存来改进，可是不在本文章讨论范围里。

方法2：

直接经过 with 把 Post 的全部 comments 数据都取出来，在 blade 中 post->comments->take(10)

问题：Laravel 会预先把文章全部的评论数据查询出来，若是文章的评论数据很是多，可能会形成内存泄漏。

方法3：

$posts = Post::paginate(15);

$postIds = $posts->pluck('id')->all();

//找出符合条件的 comments ，同时定义 @post, @rank 变量，这里没有用 all,get 等函数，此时并不会执行 SQL 语句。
$sub = Comment::whereIn('post_id',$postIds)->select(DB::raw('*,@post := NULL ,@rank := 0'))->orderBy('post_id');

//把上面构造的 sql 查询做为子表进行查询，根据 post_id 进行分区的同时 @rank 变量不断+1
$sub2 = DB::table( DB::raw("({$sub->toSql()}) as b") )
            ->mergeBindings($sub->getQuery())
            ->select(DB::raw('b.*,IF (
            @post = b.post_id ,@rank :=@rank + 1 ,@rank := 1
        ) AS rank,
        @post := b.post_id'));

//取出符合条件的前10条comment
$commentIds = DB::table( DB::raw("({$sub2->toSql()}) as c") )
            ->mergeBindings($sub2)
        ->where('rank','<',11)->select('c.id')->pluck('id')->toArray();

$comments = Comment::whereIn('id',$commentIds)->get();

$posts = $posts->each(function ($item, $key) use ($comments) {
    $item->comments = $comments->where('post_id',$item->id);
});

会产生三条sql

select * from `posts` limit 15 offset 0;

select `c`.`id` from (select b.*,IF (
@post = b.post_id ,@rank :=@rank + 1 ,@rank := 1
) AS rank,
@post := b.post_id from (select *,@post := NULL ,@rank := 0 from `comments` where `post_id` in ('2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16') order by `post_id` asc) as b) as c where `rank` < '11';

select * from `comments` where `id` in ('180', '589', '590', '3736');

知识点

1. toSql() 方法的做用是为了获取不带有 binding 参数的 SQL, 也就是说带问号的 SQL
2. getQuery() 方法的做用是为了获取 binding 参数并代替 toSql() 得到SQL的问号，从而获得完整的SQL
3. raw() 的做用是直接把 SQL 套进 Laravel 的查询构造器中。
4. mysql 查询语句中定义变量 [@post](https://my.oschina.net/u/210428) := NULL ,[@rank](https://my.oschina.net/ranky) := 0 以及 IF 函数的使用
5. 如何构建子查询。

为何不直接用原生 SQL 语句来实现？

这里之因此坚持使用 Laravel Query Builder 来实现，能够有效防止 SQL 注入，而且和 ORM 的 Model 对象关联起来。

讨论交流