Mongoid 随机文档

Question

假设我有一个用户集合。有没有一种方法可以使用 mongoid 在集合中找到 n 个随机用户，但它不会两次返回同一个用户？现在假设用户集合如下所示：

class User
  include Mongoid::Document
  field :name
end

简单吧？

谢谢

Answer 1

如果您只想要一个文档，并且不想定义新的标准方法，您可以这样做：

random_model = Model.skip(rand(Model.count)).first

如果您想根据某些标准找到随机模型：

criteria = Model.scoped_whatever.where(conditions) # query example
random_model = criteria.skip(rand(criteria.count)).first

Answer 2

最佳解决方案将取决于集合的预期大小。

对于小型集合，只需获取所有集合和 .shuffle.slice！

对于小尺寸的 n，您可以使用以下方法：

result = (0..User.count-1).sort_by{rand}.slice(0, n).collect! do |i| User.skip(i).first end

对于较大的 n，我建议创建一个“随机”列进行排序。详情见这里：http://cookbook.mongodb.org/patterns/random-attribute/https://github.com/mongodb/cookbook/blob/master/content/patterns/random-attribute.txt

Answer 3

MongoDB 3.2 借助 $sample (link to doc) 来救援

EDIT : 最新的 Mongoid 已经实现了 $sample，所以你可以调用YourCollection.all.sample(5)

以前版本的 mongoid

在 Mongoid 6 之前，Mongoid 不支持 sample，因此您必须使用 Mongo 驱动程序运行此聚合查询：

samples = User.collection.aggregate([ { '$sample': { size: 3 } } ])
# call samples.to_a if you want to get the objects in memory

你可以用它做什么

我相信这个功能应该很快就会进入 Mongoid，但与此同时

module Utility
  module_function
  def sample(model, count)
    ids = model.collection.aggregate([ 
      { '$sample': { size: count } }, # Sample from the collection
      { '$project': { _id: 1} }       # Keep only ID fields
    ]).to_a.map(&:values).flatten     # Some Ruby magic

    model.find(ids)
  end
end

Utility.sample(User, 50)

Answer 4

如果你真的想要简单，你可以改用这个：

class Mongoid::Criteria

  def random(n = 1)
    indexes = (0..self.count-1).sort_by{rand}.slice(0,n).collect!

    if n == 1
      return self.skip(indexes.first).first
    else
      return indexes.map{ |index| self.skip(index).first }
    end
  end

end

module Mongoid
  module Finders

    def random(n = 1)
      criteria.random(n)
    end

  end
end

您只需拨打User.random(5) 即可获得5 个随机用户。 它也适用于过滤，因此如果您只想要注册用户，您可以使用User.where(:registered => true).random(5)。

对于大型集合，这将需要一段时间，因此我建议您使用另一种方法，您可以对计数进行随机除法（例如：25 000 到 30 000）并随机化该范围。

Answer 5

你可以这样做

1. 生成随机偏移，这将进一步满足选择下一个 n 元素（不超过限制）
2. 假设count为10，n为5
3. 为此检查给定的 n 是否小于总计数
4. 如果没有设置偏移量为0，并转到步骤8
5. 如果是，从总数中减去n，你会得到一个数字5
6. 用这个来找一个随机数，这个数肯定是从0到5（假设2）
7. 使用随机数 2 作为偏移量
8. 现在您可以通过简单地传递此偏移量和 n (5) 作为限制来随机获取 5 个用户。
9. 现在您的用户数从 3 到 7

代码

>> cnt = User.count
=> 10
>> n = 5
=> 5
>> offset = 0
=> 0
>> if n<cnt
>>    offset = rand(cnt-n)
>>  end
>> 2
>> User.skip(offset).limit(n)

你可以把它放在一个方法中

def get_random_users(n)
  offset = 0
  cnt = User.count
  if n < cnt
    offset = rand(cnt-n)
  end
  User.skip(offset).limit(n)
end

然后这样称呼它

rand_users = get_random_users(5)

希望对你有帮助

Answer 6

因为我想保留一个标准，所以我这样做：

scope :random, ->{
  random_field_for_ordering = fields.keys.sample
  random_direction_to_order = %w(asc desc).sample
  order_by([[random_field_for_ordering, random_direction_to_order]])
}

Answer 7

刚遇到这样的问题。试过了

Model.all.sample

它对我有用

Answer 8

@moox 的方法非常有趣，但我怀疑猴子修补整个 Mongoid 是否是个好主意。所以我的方法只是写一个关注Randomizable，它可以包含在您使用此功能的每个模型中。这转到app/models/concerns/randomizeable.rb：

module Randomizable
  extend ActiveSupport::Concern

  module ClassMethods
    def random(n = 1)
      indexes = (0..count - 1).sort_by { rand }.slice(0, n).collect!

      return skip(indexes.first).first if n == 1
      indexes.map { |index| skip(index).first }
    end
  end
end

那么您的 User 模型将如下所示：

class User
  include Mongoid::Document
  include Randomizable

  field :name
end

还有测试......

require 'spec_helper'

class RandomizableCollection
  include Mongoid::Document
  include Randomizable

  field :name
end

describe RandomizableCollection do
  before do
    RandomizableCollection.create name: 'Hans Bratwurst'
    RandomizableCollection.create name: 'Werner Salami'
    RandomizableCollection.create name: 'Susi Wienerli'
  end

  it 'returns a random document' do
    srand(2)

    expect(RandomizableCollection.random(1).name).to eq 'Werner Salami'
  end

  it 'returns an array of random documents' do
    srand(1)

    expect(RandomizableCollection.random(2).map &:name).to eq ['Susi Wienerli', 'Hans Bratwurst']
  end
end

Answer 9

我认为最好专注于随机化返回的结果集，所以我尝试了：

Model.all.to_a.shuffle

希望这会有所帮助。