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 := NULL ,@rank := 0 以及 IF 函數的使用
5. 如何構建子查詢。

為什麼不直接用原生 SQL 語句來實現？

這裡之所以堅持使用 Laravel Query Builder 來實現，可以有效防止 SQL 註入，並且和 ORM 的 Model 對象關聯起來。