MongoDBのAggregationでサブセットに条件を適用する方法（$setIsSubsetの使い方）

MongoDBの集計パイプライン（Aggregation）では、$setIsSubset演算子を使うことで、配列が指定したサブセットを含んでいるかどうかを判定し、その結果に基づいて条件分岐を行うことができます。この記事では、実際のサンプルコードとともに、$setIsSubsetの基本的な使い方を解説します。

サンプルコレクションの作成

まずは、ドキュメントを含むコレクションを作成しましょう。ここでは、学生の名前と好きな科目のリストを持つ「subsetDemo」コレクションを使用します。

> db.subsetDemo.insertOne({"StudentName":"Chris","StudentFavouriteSubject":["Java","Python"]});
{
   "acknowledged" : true,
   "insertedId" : ObjectId("5e063e49150ee0e76c06a052")
}
> db.subsetDemo.insertOne({"StudentName":"Chris","StudentFavouriteSubject":["Java","Python","MySQL"]});
{
   "acknowledged" : true,
   "insertedId" : ObjectId("5e063e4f150ee0e76c06a053")
}

上記のコードでは、1人目の学生は「Java」と「Python」を、2人目の学生はそれらに加えて「MySQL」も好きな科目として登録しています。

登録したドキュメントの確認

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

> db.subsetDemo.find().pretty();

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

{
   "_id" : ObjectId("5e063e49150ee0e76c06a052"),
   "StudentName" : "Chris",
   "StudentFavouriteSubject" : [
      "Java",
      "Python"
   ]
}
{
   "_id" : ObjectId("5e063e4f150ee0e76c06a053"),
   "StudentName" : "Chris",
   "StudentFavouriteSubject" : [
      "Java",
      "Python",
      "MySQL"
   ]
}

$setIsSubsetで条件を適用する

それでは、MongoDBのAggregationパイプライン内でサブセットに条件を適用するクエリを見ていきましょう。ここでは、「好きな科目の中にMySQLが含まれているかどうか」を判定するフィールド「isMySQL」を追加しています。

> db.subsetDemo.aggregate([
...    {
...       $project: {
...          _id: 0,
...          StudentName: 1,
...          isMySQL: {
...             $setIsSubset: [["MySQL"], "$StudentFavouriteSubject"]
...          }
...       }
...    }
... ]);

このクエリのポイントは以下の通りです。

• $projectステージで、出力フィールドを制御しています（_id: 0でIDを非表示にし、学生名を表示）。
• $setIsSubsetは、第一引数の配列が第二引数の配列の部分集合である場合にtrue、そうでなければfalseを返します。
• ここでは["MySQL"]が$StudentFavouriteSubjectに含まれるかをチェックしています。

実行結果

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

{ "StudentName" : "Chris", "isMySQL" : false }
{ "StudentName" : "Chris", "isMySQL" : true }

1つ目のドキュメントは「MySQL」を含んでいないためfalseとなり、2つ目のドキュメントは「MySQL」を含んでいるためtrueとなっています。このように、$setIsSubsetを使えば、配列フィールドに対して柔軟に条件判定を行うことができます。