使用$ in和MySQL字段('_ id',...)的MongoDB中的结果顺序

时间:2014-12-17 12:05:25

标签: mysql mongodb sorting

我正在使用MongoDB构建一个非常精彩的“趋势”帖子算法。由于算法消耗了相当多的时间,我在cron任务中执行我的算法然后我缓存一个排序的帖子ID数组。像这样(PHP vardump):

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

关键是我无法找到任何方法使用MongoDB按顺序检索它们。 使用MySQL它只需要执行:

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

我到目前为止所尝试的是founde here,所以现在我能够检索带有权重的排序列表,而不是帖子。答案是这样的:

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

有没有人达到这个目标,甚至知道从哪里开始?

谢谢!

1 个答案:

答案 0 :(得分:1)

我来自您链接的SO帖子。我发现使用MongoDB提供的$$ROOT变量。

我的代码如下所示:

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,就像我一样)会包含在这里,但我怀疑不是。