Порядок результатов в MongoDB с $в виде поля MySQL ('_ id',...)

Я создаю довольно причудливый "трендовый" алгоритм сообщений с использованием MongoDB. Поскольку алгоритм потребляет довольно много времени, я выполняю свой алгоритм в задаче cron, а затем кешу отсортированного массива сообщений ids. Как это (PHP vardump):

"trending" => array("548ac5ce05ea675e468b4966", "5469c6d5039069a5030041a7", ...)

Дело в том, что я не могу найти способ получить их в этом порядке с помощью MongoDB. Используя MySQL, это будет сделано, просто сделав:

SELECT * FROM posts
ORDER BY FIELD(_id, '548ac5ce05ea675e468b4966', '5469c6d5039069a5030041a7',...)

То, что я пробовал до сих пор, это то, что можно найти здесь, так что теперь я могу получить отсортированный список с весом, но не постов. Ответ такой:

{
    "result" : [ 
        {
            "_id" : ObjectId("548ac5ce05ea675e468b4966"),
            "weight" : 1
        }, 
        {
            "_id" : ObjectId("5469c6d5039069a5030041a7"),
            "weight" : 2
        }
    ], 
    "ok" : 1
}

Кто-нибудь достиг этого или даже знает, с чего начать?

Спасибо!

Ответ 1

Я прихожу из сообщения SO, которое вы связали. То, что я нашел, это использовать переменную $$ ROOT, предоставленную MongoDB.

Мой код выглядит следующим образом:

var ids = [456, 123, 789]
  , stack = []
  ;

// Note that `i` is decremented here, not incremented like
// in the linked SO post. I think they have a typo.
for (var i = ids.length-1; i > 0; i--) {
  var rec = {
    $cond: [
      {$eq: ['$_id', ids[i - 1]]},
      i
    ]
  };

  if (stack.length === 0) {
    rec.$cond.push(i + 1);
  }
  else {
    var lval = stack.pop();
    rec.$cond.push(lval);
  }

  stack.push(rec);
}

var pipeline = [
  {$match: {
    _id: {$in: ids}
  }},
  {$project: {
    weight: stack[0],
    data: '$$ROOT' // This will give you the whole document.
  }},
  {$sort: {weight: 1}}
];

Posts.aggregate(pipeline, function (err, posts) {
  if (err) return next();

  // Use Underscore to grab the docs stored on `data` from above.
  res.json(_.pluck(posts, 'data'));
});

Обратите внимание, что я лично не совсем уверен, что происходит в этом цикле for для создания запроса.: -P Поэтому большинство кредитов должно пойти на них. Также я не уверен, что если бы здесь были включены виртуальные поля (если вы используете Mongoose, как и я), но я подозреваю, что нет.