如何原子弹出随机元素？

Question

有没有办法用MongoDB像Redis的SPOP那样原子地弹出（删除和检索）随机元素？

我已经阅读了RandomAttribute教程，但是现在我需要确保在提取元素时也将其删除，并且这必须以原子方式进行。

我猜作为一种替代方案，我可以将数据推送到预先排序的数组字段中，但是我真的更希望让它获取随机记录。

查看$pop的文档，似乎它不能接受参数，因此它将删除数组的第一个或最后一个元素。

Answer 1

Mongo集合中的更新是原子性的，因此将findAndModify与RandomAttribute方法结合使用将达到目的：

var rand = Math.rand();
db.collection.findAndModify({query: {random: {$gte: rand}}, remove: true});

只要确保还查询random: {$lt: rand}当上面的返回null 。

提示：一些驱动程序具有findAndRemove ，与带有remove: true findAndModify相同。

Answer 2

基本上，目前尚没有一种理想的方式来做您想做的事情。 最大的问题是，没有一种原子方法可以通过其位置删除数组元素。

有关更多信息，请参见此问题 。

我可以想到的最好的方法（最接近原子的方法）如下。 它依赖于您能够在短时间内处理数组中的空元素（这不是原子位）。

第一步

在客户端生成一个随机数。 使用此属性以点表示法（即“ my_array.4”）指定要删除的数组元素的键，因为AFAIK不能在查询中动态完成。

第二步

使用随机数以原子方式取消设置数组的选定元素，同时使用new设置为false的findAndModify进行更新之前 ，将其findAndModify 。 假设您生成的随机数是4 ，您将执行以下操作：

db.mycollection.findAndModify({
  query: {_id: 1},
  update: {$unset: {"my_array.4" : true }},
  fields: {my_array : {$slice : [4, 1]}, _id : false},
  new: false
});

（注意： findAndModify当前默认情况下会返回更新前的文档，但是我读到了可能会更改的地方，因此始终最好使用new选项指定要new 。）

这将保留在数组中被选择为null值的数组元素。

第三步

使用$ pull使用另一个更新从数组中删除null元素：

db.mycollection.update({_id: 1}, {$pull: {my_array: null}})

唷！ 似乎需要做很多工作。 我不知道Redis，但SPOP听起来容易得多。 无论如何，我希望这能有所帮助。 这不是单个原子操作，但是（只要您可以处理这些null），我认为它最重要的部分是。 您永远不会陷入两个不同线程几乎同时弹出同一元素的情况，我想这就是您要避免的情况。