MongoDBでMySQLのUNIONに似たクエリを実装する方法

MySQLのUNIONに相当するクエリをMongoDBで実現したい場合は、aggregate()と$lookup（JOIN）を組み合わせることで実装できます。この記事では、2つのコレクションを結合して集計する具体的な手順を、サンプルコードとともに解説します。

最初のコレクション（demo486）を作成する

まず、ドキュメントを含むコレクションを作成します。

> db.demo486.insertOne({_id:1,"Amount":30,"No":4});
{ "acknowledged" : true, "insertedId" : 1 }
> db.demo486.insertOne({_id:2,"Amount":40,"No":2});
{ "acknowledged" : true, "insertedId" : 2 }
> db.demo486.insertOne({_id:3,"Amount":60,"No":6});
{ "acknowledged" : true, "insertedId" : 3 }

find()メソッドを使って、コレクション内のすべてのドキュメントを表示してみましょう。

> db.demo486.find();

実行すると、次のような出力が得られます。

{ "_id" : 1, "Amount" : 30, "No" : 4 }
{ "_id" : 2, "Amount" : 40, "No" : 2 }
{ "_id" : 3, "Amount" : 60, "No" : 6 }

2つ目のコレクション（demo487）を作成する

続いて、結合先となる2つ目のコレクションをドキュメント付きで作成します。

> db.demo487.insertOne({_id:1,"Price":10,"No":4});
{ "acknowledged" : true, "insertedId" : 1 }
> db.demo487.insertOne({_id:2,"Price":80,"No":9});
{ "acknowledged" : true, "insertedId" : 2 }
> db.demo487.insertOne({_id:3,"Price":20,"No":6});
{ "acknowledged" : true, "insertedId" : 3 }

こちらもfind()メソッドで、コレクション内のすべてのドキュメントを確認します。

> db.demo487.find();

実行結果は以下のとおりです。

{ "_id" : 1, "Price" : 10, "No" : 4 }
{ "_id" : 2, "Price" : 80, "No" : 9 }
{ "_id" : 3, "Price" : 20, "No" : 6 }

UNIONに相当するクエリを実行する

それでは、2つのコレクションを結合して集計するクエリを見ていきましょう。$lookupでコレクションを結合し、$unwindで配列を展開、$groupで「No」ごとに「Amount」と「Price」の合計を計算しています。最後に$sortと$projectで結果を整形します。

> db.getCollection('demo486').aggregate([
... {$lookup : { from : "demo487",localField : "No", foreignField :"No", as:"demo487"}},
... {$unwind : "$demo487"},
... {
...     $group : {
...        _id : {
...           No :"$No",
...        },
...        TotalValue : { $sum : { $add: [ "$Amount", "$demo487.Price" ] }}
...     }
... },
... {$sort : {"_id.No":1}},
... {
...     $project : {
...        No :"$_id.No",
...        TotalValue : 1,
...        _id : 0
...     }
... }
... ])

このクエリを実行すると、次のような結果が出力されます。

{ "TotalValue" : 40, "No" : 4 }
{ "TotalValue" : 80, "No" : 6 }

このように、MongoDBではaggregate()パイプラインの各ステージを組み合わせることで、MySQLのUNIONやJOINに近い処理を柔軟に実現できます。$lookupによる結合、$groupによる集計、$projectによる出力フィールドの制御など、目的に応じてステージを追加・調整してみてください。