diff options
Diffstat (limited to 'activerecord/lib/active_record')
4 files changed, 78 insertions, 24 deletions
diff --git a/activerecord/lib/active_record/associations/association_collection.rb b/activerecord/lib/active_record/associations/association_collection.rb index cbfbdf1541..7ee567e0b4 100644 --- a/activerecord/lib/active_record/associations/association_collection.rb +++ b/activerecord/lib/active_record/associations/association_collection.rb @@ -124,14 +124,6 @@ module ActiveRecord end private - def method_missing(method, *args, &block) - if @target.respond_to?(method) or (not @association_class.respond_to?(method) and Class.respond_to?(method)) - super - else - @association_class.constrain(:conditions => @finder_sql, :joins => @join_sql, :readonly => false) { @association_class.send(method, *args, &block) } - end - end - def raise_on_type_mismatch(record) raise ActiveRecord::AssociationTypeMismatch, "#{@association_class} expected, got #{record.class}" unless record.is_a?(@association_class) end diff --git a/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb b/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb index 1b7adc3f39..2972aa0248 100644 --- a/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb +++ b/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb @@ -76,6 +76,16 @@ module ActiveRecord end protected + def method_missing(method, *args, &block) + if @target.respond_to?(method) || (!@association_class.respond_to?(method) && Class.respond_to?(method)) + super + else + @association_class.constrain(:conditions => @finder_sql, :joins => @join_sql, :readonly => false) do + @association_class.send(method, *args, &block) + end + end + end + def find_target if @options[:finder_sql] records = @association_class.find_by_sql(@finder_sql) diff --git a/activerecord/lib/active_record/associations/has_many_association.rb b/activerecord/lib/active_record/associations/has_many_association.rb index f5f27dc410..b04cb81a5c 100644 --- a/activerecord/lib/active_record/associations/has_many_association.rb +++ b/activerecord/lib/active_record/associations/has_many_association.rb @@ -85,6 +85,20 @@ module ActiveRecord end protected + def method_missing(method, *args, &block) + if @target.respond_to?(method) || (!@association_class.respond_to?(method) && Class.respond_to?(method)) + super + else + @association_class.constrain( + :conditions => @finder_sql, + :joins => @join_sql, + :readonly => false, + :creation => { @association_class_primary_key_name => @owner.id }) do + @association_class.send(method, *args, &block) + end + end + end + def find_target find_all end diff --git a/activerecord/lib/active_record/base.rb b/activerecord/lib/active_record/base.rb index 4aaa05a6c9..3fc55e0783 100755 --- a/activerecord/lib/active_record/base.rb +++ b/activerecord/lib/active_record/base.rb @@ -140,7 +140,7 @@ module ActiveRecord #:nodoc: # # == Dynamic attribute-based finders # - # Dynamic attribute-based finders are a cleaner way of getting objects by simple queries without turning to SQL. They work by + # Dynamic attribute-based finders are a cleaner way of getting (and/or creating) objects by simple queries without turning to SQL. They work by # appending the name of an attribute to <tt>find_by_</tt> or <tt>find_all_by_</tt>, so you get finders like Person.find_by_user_name, # Person.find_all_by_last_name, Payment.find_by_transaction_id. So instead of writing # <tt>Person.find(:first, ["user_name = ?", user_name])</tt>, you just do <tt>Person.find_by_user_name(user_name)</tt>. @@ -155,6 +155,15 @@ module ActiveRecord #:nodoc: # is actually Payment.find_all_by_amount(amount, options). And the full interface to Person.find_by_user_name is # actually Person.find_by_user_name(user_name, options). So you could call <tt>Payment.find_all_by_amount(50, :order => "created_on")</tt>. # + # The same dynamic finder style can be used to create the object if it doesn't already exist. This dynamic finder is called with + # <tt>find_or_create_by_</tt> and will return the object if it already exists and otherwise creates it, then returns it. Example: + # + # # No 'Summer' tag exists + # Tag.find_or_create_by_name("Summer") # equal to Tag.create(:name => "Summer") + # + # # Now the 'Summer' tag does exist + # Tag.find_or_create_by_name("Summer") # equal to Tag.find_by_name("Summer") + # # == Saving arrays, hashes, and other non-mappable objects in text columns # # Active Record can serialize any object in text columns using YAML. To do so, you must specify this with a call to the class method +serialize+. @@ -451,6 +460,8 @@ module ActiveRecord #:nodoc: if attributes.is_a?(Array) attributes.collect { |attr| create(attr) } else + attributes.reverse_merge!(scope_constraints[:creation]) if scope_constraints[:creation] + object = new(attributes) object.save object @@ -838,10 +849,10 @@ module ActiveRecord #:nodoc: # end def constrain(options = {}) options = options.dup - if !options[:joins].blank? and !options.has_key?(:readonly) - options[:readonly] = true - end + options[:readonly] = true if !options[:joins].blank? && !options.has_key?(:readonly) + self.scope_constraints = options + yield if block_given? ensure self.scope_constraints = nil @@ -948,27 +959,54 @@ module ActiveRecord #:nodoc: # It's even possible to use all the additional parameters to find. For example, the full interface for find_all_by_amount # is actually find_all_by_amount(amount, options). def method_missing(method_id, *arguments) - method_name = method_id.id2name + if match = /find_(all_by|by)_([_a-zA-Z]\w*)/.match(method_id.to_s) + finder = determine_finder(match) - if md = /find_(all_by|by)_([_a-zA-Z]\w*)/.match(method_id.to_s) - finder = md.captures.first == 'all_by' ? :all : :first - attributes = md.captures.last.split('_and_') - attributes.each { |attr_name| super unless column_methods_hash.include?(attr_name.to_sym) } + attribute_names = extract_attribute_names_from_match(match) + super unless all_attributes_exists?(attribute_names) - attr_index = -1 - conditions = attributes.collect { |attr_name| attr_index += 1; "#{table_name}.#{attr_name} #{attribute_condition(arguments[attr_index])} " }.join(" AND ") + conditions = construct_conditions_from_arguments(attribute_names, arguments) - if arguments[attributes.length].is_a?(Hash) - find(finder, { :conditions => [conditions, *arguments[0...attributes.length]] }.update(arguments[attributes.length])) + if arguments[attribute_names.length].is_a?(Hash) + find(finder, { :conditions => conditions }.update(arguments[attribute_names.length])) else - # deprecated API - send("find_#{finder}", [conditions, *arguments[0...attributes.length]], *arguments[attributes.length..-1]) + send("find_#{finder}", conditions, *arguments[attribute_names.length..-1]) # deprecated API end + elsif match = /find_or_create_by_([_a-zA-Z]\w*)/.match(method_id.to_s) + attribute_names = extract_attribute_names_from_match(match) + super unless all_attributes_exists?(attribute_names) + + find(:first, :conditions => construct_conditions_from_arguments(attribute_names, arguments)) || + create(construct_attributes_from_arguments(attribute_names, arguments)) else super end end + def determine_finder(match) + match.captures.first == 'all_by' ? :all : :first + end + + def extract_attribute_names_from_match(match) + match.captures.last.split('_and_') + end + + def construct_conditions_from_arguments(attribute_names, arguments) + conditions = [] + attribute_names.each_with_index { |name, idx| conditions << "#{table_name}.#{name} #{attribute_condition(arguments[idx])} " } + [ conditions.join(" AND "), *arguments[0...attribute_names.length] ] + end + + def construct_attributes_from_arguments(attribute_names, arguments) + attributes = {} + attribute_names.each_with_index { |name, idx| attributes[name] = arguments[idx] } + attributes + end + + def all_attributes_exists?(attribute_names) + attribute_names.all? { |name| column_methods_hash.include?(name.to_sym) } + end + def attribute_condition(argument) case argument when nil then "IS ?" @@ -1691,4 +1729,4 @@ module ActiveRecord #:nodoc: value end end -end +end
\ No newline at end of file |