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 對象關聯起來。

若是你們還有更多相似這種複雜的需求，歡迎你們投稿。

討論交流