Расчеты ранга Монго на основе подсчета - PullRequest
Новая
       2

Расчеты ранга Монго на основе подсчета

1 голос
/ 06 мая 2019

Я пытаюсь вычислить ранг монго на основе количества и упомянутой ниже схемы БД. Я не получаю ожидаемых результатов. Кто-нибудь поможет решить эту проблему?

Запрос Монго: 

    db.company.aggregate([
        {
            "$group": {
                "_id": { 
                    "name1": "$name1",
                    "name2": "$name2",
                }, 
                "expanded": { 
                    "$push": { 
                        "name1": "$name1", 
                        "name2": "$name2", 
                    } 
                },
                "count": { "$sum": 1 } 
            } 
        },
        { "$sort": { "count": -1 } },
        {
            $unwind: {
                path: '$expanded',
                includeArrayIndex: 'count'
            }
        }
    ]);

Ожидаемые результаты, такие как 

Name|Count|Rank
Google|3|1 
FB|2|2
Yahoo|1| 3

Схема БД: 

{
    "_id" : 1.0,
    "name1" : "Yahoo",
    "name2" : "Google",
    "salary" : 1000.0
}

/* 2 */
{
    "_id" : 2.0,
    "name1" : "FB",
    "name2" : "Google",
    "salary" : 2000.0
}

/* 3 */
{
    "_id" : 3.0,
    "name1" : "Google",
    "name2" : "FB",
    "salary" : 1500.0
}

Ответы [ 2 ]

1 голос
/ 06 мая 2019

попробуйте это 

db.company.aggregate([
{
  $group: {
    _id:null,
    names1: {$push: "$name1"},
    names2: {$push:"$name2"},
  }
},
{
  $project: {
    _id: 0,
    names:{$concatArrays: ["$names1", "$names2"]}
  }
},
{$unwind: "$names"},
{$sortByCount: "$names"},
{$addFields:{name: "$_id"}},
{
  $group : {
    _id: null,
    records : { $push : {count : "$count", name : "$name"}}
  }
},
{
  $project: {
   total_docs: {$size: "$records"},
   records: 1
  }
},
{$unwind: "$records"},
{
  $project: {
    _id: 0,
    name: "$records.name",
    count:"$records.count",
    rank: {
      $add:[
        {
          $subtract:["$total_docs", "$records.count"]
        }, 1]
    }
  }
}])
1 голос
/ 06 мая 2019

Похоже, вы должны считать name1 и name2 по отдельности, чтобы вы могли создать временный 2-элементный массив и затем запустить $ unwind для этого массива. Кроме того, чтобы получить rank, вам нужно $ group на null, чтобы получить один массив всех групп, попробуйте: 

db.collection.aggregate([
    {
        $project: {
            key: [ "$name1", "$name2" ]
        }
    },
    {
        $unwind: "$key"
    },
    {
        $group: {
            _id: "$key",
            count: { $sum: 1 }
        }
    },
    {
        $sort: {
            count: -1
        }
    },
    {
        $group: {
            _id: null,
            groups: { $push: "$$ROOT" }
        }
    },
    {
        $unwind: {
            path: '$groups',
            includeArrayIndex: 'rank'
        }
    },
    {
        $project: {
            _id: 0,
            name: "$groups._id",
            rank: { $add: [ "$rank", 1 ] },
            count: "$groups.count"
        }
    }
])

Детская площадка Монго

...