MongoDB Aggregation에서 하위 집합 내부에 조건 적용하기

MongoDB Aggregation 파이프라인에서 배열(하위 집합) 내부에 조건을 적용하려면 $setIsSubset 연산자를 사용하면 됩니다. 이 연산자는 첫 번째 배열이 두 번째 배열의 부분 집합인지 여부를 판단하여 true 또는 false를 반환합니다.

샘플 컬렉션 생성

먼저 문서가 포함된 컬렉션을 생성해 보겠습니다.

> 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")
}

위 예제에서는 학생 이름(StudentName)과 좋아하는 과목 목록(StudentFavouriteSubject)을 저장하는 두 개의 문서를 생성했습니다. 두 번째 문서에는 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에서 하위 집합 내부에 조건을 적용하는 쿼리입니다.

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

위 쿼리의 동작 방식을 살펴보면 다음과 같습니다.

• $project 단계에서 _id 필드는 제외하고(_id: 0), StudentName 필드만 포함시킵니다.
• isMySQL이라는 새로운 필드를 만들어, ["MySQL"] 배열이 StudentFavouriteSubject 배열의 부분 집합인지 검사합니다.
• 학생의 좋아하는 과목 목록에 MySQL이 포함되어 있으면 true, 없으면 false가 반환됩니다.

이 쿼리는 다음과 같은 결과를 출력합니다.

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

결과를 보면 첫 번째 문서는 과목 목록에 MySQL이 없어 false가 반환되었고, 두 번째 문서는 MySQL이 포함되어 있어 true가 반환된 것을 확인할 수 있습니다. 이처럼 $setIsSubset을 활용하면 Aggregation 파이프라인에서 배열 필드에 대한 조건 판별을 손쉽게 처리할 수 있습니다.