Mongodb Document: есть ли способ подсчитать поддокументы с условием? - PullRequest
1 голос
/ 12 апреля 2019

У меня есть документы с вложенными документами, где поле " stars " в вложенных документах имеет либо " 1 ", либо " 0 ". Я хочу посчитать их на уровне документа на основе этих значений в поле " stars ".

Вот так выглядит текущий документ:

{
  "name": "Hotel A",
  "category": "hotel",
  "reviews": [
    {
      "title": "A",
      "stars": 1
    },
    {
      "title": "B",
      "stars": 1
    },
    {
      "title": "C",
      "stars": 0
    }
  ],
  "total_reviews": 3
},{
  "name": "Hotel B",
  "category": "hotel",
  "reviews": [
    {
      "title": "A",
      "stars": 1
    },
    {
      "title": "B",
      "stars": 1
    },
    {
      "title": "C",
      "stars": 0
    },
        {
      "title": "D",
      "stars": 0
    },
    {
      "title": "E",
      "stars": 1
    },
    {
      "title": "F",
      "stars": 0
    }
  ],
"total_reviews": 6
}

И это ожидаемый результат:

{
  "name": "Hotel A",
  "category": "hotel",
  "reviews": [
    {
      "title": "A",
      "stars": 1
    },
    {
      "title": "B",
      "stars": 1
    },
    {
      "title": "C",
      "stars": 0
    }
  ],
  "positive_reviews": 2,
  "negative_reviews": 1,
  "total_reviews": 3
},{
  "name": "Hotel B",
  "category": "hotel",
  "reviews": [
    {
      "title": "A",
      "stars": 1
    },
    {
      "title": "B",
      "stars": 1
    },
    {
      "title": "C",
      "stars": 0
    },
        {
      "title": "D",
      "stars": 0
    },
    {
      "title": "E",
      "stars": 1
    },
    {
      "title": "F",
      "stars": 0
    }
  ],
"positive_reviews": 3,
"negative_reviews": 3,
"total_reviews": 6
}

Добавив два новых поля: " positive_reviews " if {"reviews.stars": 1} и " positive_reviews ", если {" reviews.stars ": 0} со значениями счетчика

1 Ответ

2 голосов
/ 12 апреля 2019

Попробуйте, как показано ниже:

db.collection.aggregate([
    {
        $project: {
            "_id" : 1,
            "name" : 1,
            "category" : 1,
            "reviews" : 1,
            "positive_reviews":
            {
               $reduce: 
                 {
                   input: "$reviews",
                   initialValue: 0,
                   in: { 
                       $add: ["$$value", { $cond:[{ $eq: ["$$this.stars", 1]} , 1, 0 ] } ] 
                   }
                 }
            },
            "negative_reviews":
            {
               $reduce: 
                 {
                   input: "$reviews",
                   initialValue: 0,
                   in: { 
                       $add: ["$$value", { $cond:[{ $eq: ["$$this.stars", 0]} , 1, 0 ] } ] 
                   }
                 }
            },
            "total_reviews": { $size: "$reviews"}
        }
    }
])

Результат Ответ:

/* 1 createdAt:12/04/2019, 18:24:50*/
{
    "_id" : ObjectId("5cb08a9a952e3a179190d996"),
    "name" : "Hotel A",
    "category" : "hotel",
    "reviews" : [
        {
            "title" : "A",
            "stars" : 1
        },
        {
            "title" : "B",
            "stars" : 1
        },
        {
            "title" : "C",
            "stars" : 0
        }
    ],
    "positive_reviews" : 2,
    "negative_reviews" : 1,
    "total_reviews" : NumberInt(3)
},

/* 2 createdAt:12/04/2019, 18:24:50*/
{
    "_id" : ObjectId("5cb08a9a952e3a179190d997"),
    "name" : "Hotel B",
    "category" : "hotel",
    "reviews" : [
        {
            "title" : "A",
            "stars" : 1
        },
        {
            "title" : "B",
            "stars" : 1
        },
        {
            "title" : "C",
            "stars" : 0
        },
        {
            "title" : "D",
            "stars" : 0
        },
        {
            "title" : "E",
            "stars" : 1
        },
        {
            "title" : "F",
            "stars" : 0
        }
    ],
    "positive_reviews" : 3,
    "negative_reviews" : 3,
    "total_reviews" : NumberInt(6)
}
...