Module: Mongoid::Ids::Finders

Defined in:
lib/mongoid/ids/finders.rb

Class Method Summary collapse

Class Method Details

.define_custom_token_finder_for(klass, field_name = :token) ⇒ Object



4
5
6
7
8
9
10
11
12
# File 'lib/mongoid/ids/finders.rb', line 4

def self.define_custom_token_finder_for(klass, field_name = :token)
  klass.define_singleton_method(:"find_by_#{field_name.to_s}") do |token|
    self.find_by(field_name.to_sym => token)
  end

  klass.define_singleton_method :"find_with_#{field_name}" do |*args| # this is going to be painful if tokens happen to look like legal object ids
    args.all?{|arg| BSON::ObjectId.legal?(arg)} ? send(:"find_without_#{field_name}",*args) : klass.send(:"find_by_#{field_name.to_s}", args.first)
  end
end