module Sequel::Model::Associations::ClassMethods

Each kind of association adds a number of instance methods to the model class which are specialized according to the association type and optional parameters given in the definition. Example:

class Project < Sequel::Model
  many_to_one :portfolio
  # or: one_to_one :portfolio
  one_to_many :milestones
  # or: many_to_many :milestones
end

The project class now has the following instance methods:

portfolio: Returns the associated portfolio.
portfolio=(obj): Sets the associated portfolio to the object, but the change is not persisted until you save the record (for many_to_one associations).
portfolio_dataset: Returns a dataset that would return the associated portfolio, only useful in fairly specific circumstances.
milestones: Returns an array of associated milestones
add_milestone(obj): Associates the passed milestone with this object.
remove_milestone(obj): Removes the association with the passed milestone.
remove_all_milestones: Removes associations with all associated milestones.
milestones_dataset: Returns a dataset that would return the associated milestones, allowing for further filtering/limiting/etc.

If you want to override the behavior of the add_/remove_/remove_all_/ methods or the association setter method, use the :adder, :remover, :clearer, and/or :setter options. These options override the default behavior.

By default the classes for the associations are inferred from the association name, so for example the Project#portfolio will return an instance of Portfolio, and Project#milestones will return an array of Milestone instances. You can use the :class option to change which class is used.

Association definitions are also reflected by the class, e.g.:

Project.associations
=> [:portfolio, :milestones]
Project.association_reflection(:portfolio)
=> #<Sequel::Model::Associations::ManyToOneAssociationReflection Project.many_to_one :portfolio>

Associations should not have the same names as any of the columns in the model's current table they reference. If you are dealing with an existing schema that has a column named status, you can't name the association status, you'd have to name it foo_status or something else. If you give an association the same name as a column, you will probably end up with an association that doesn't work, or a SystemStackError.

For a more in depth general overview, as well as a reference guide, see the Association Basics guide. For examples of advanced usage, see the Advanced Associations guide.

Attributes

association_reflections[R]

All association reflections defined for this model (default: {}).

autoreloading_associations[R]

Hash with column symbol keys and arrays of many_to_one association symbols that should be cleared when the column value changes.

cache_associations[RW]

Whether association metadata should be cached in the association reflection. If not cached, it will be computed on demand. In general you only want to set this to false when using code reloading. When using code reloading, setting this will make sure that if an associated class is removed or modified, this class will not have a reference to the previous class.

default_association_options[RW]

The default options to use for all associations. This hash is merged into the association reflection hash for all association reflections.

default_association_type_options[RW]

The default options to use for all associations of a given type. This is a hash keyed by association type symbol. If there is a value for the association type symbol key, the resulting hash will be merged into the association reflection hash for all association reflections of that type.

default_eager_limit_strategy[RW]

The default :eager_limit_strategy option to use for limited or offset associations (default: true, causing Sequel to use what it considers the most appropriate strategy).

Public Instance Methods

all_association_reflections() click to toggle source

Array of all association reflections for this model class

     # File lib/sequel/model/associations.rb
1793 def all_association_reflections
1794   association_reflections.values
1795 end

associate(type, name, opts = OPTS, &block) click to toggle source

Associates a related model with the current model. The following types are supported:

:many_to_one: Foreign key in current model's table points to associated model's primary key. Each associated model object can be associated with more than one current model objects. Each current model object can be associated with only one associated model object.
:one_to_many: Foreign key in associated model's table points to this model's primary key. Each current model object can be associated with more than one associated model objects. Each associated model object can be associated with only one current model object.
:one_through_one: Similar to many_to_many in terms of foreign keys, but only one object is associated to the current object through the association. Provides only getter methods, no setter or modification methods.
:one_to_one: Similar to one_to_many in terms of foreign keys, but only one object is associated to the current object through the association. The methods created are similar to many_to_one, except that the one_to_one setter method saves the passed object.
:many_to_many: A join table is used that has a foreign key that points to this model's primary key and a foreign key that points to the associated model's primary key. Each current model object can be associated with many associated model objects, and each associated model object can be associated with many current model objects.

The following options can be supplied:

Multiple Types¶ ↑

:adder: Proc used to define the private add* method for doing the database work to associate the given object to the current object (*_to_many assocations). Set to nil to not define a add_* method for the association.
:after_add: Symbol, Proc, or array of both/either specifying a callback to call after a new item is added to the association.
:after_load: Symbol, Proc, or array of both/either specifying a callback to call after the associated record(s) have been retrieved from the database.
:after_remove: Symbol, Proc, or array of both/either specifying a callback to call after an item is removed from the association.
:after_set: Symbol, Proc, or array of both/either specifying a callback to call after an item is set using the association setter method.
:allow_eager: If set to false, you cannot load the association eagerly via eager or eager_graph
:allow_eager_graph: If set to false, you cannot load the association eagerly via eager_graph.
:allow_filtering_by: If set to false, you cannot use the association when filtering
:before_add: Symbol, Proc, or array of both/either specifying a callback to call before a new item is added to the association.
:before_remove: Symbol, Proc, or array of both/either specifying a callback to call before an item is removed from the association.
:before_set: Symbol, Proc, or array of both/either specifying a callback to call before an item is set using the association setter method.
:cartesian_product_number: the number of joins completed by this association that could cause more than one row for each row in the current table (default: 0 for many_to_one, one_to_one, and one_through_one associations, 1 for one_to_many and many_to_many associations).
:class: The associated class or its name as a string or symbol. If not given, uses the association's name, which is camelized (and singularized unless the type is :many_to_one, :one_to_one, or one_through_one). If this is specified as a string or symbol, you must specify the full class name (e.g. “::SomeModule::MyModel”).
:class_namespace: If :class is given as a string or symbol, sets the default namespace in which to look for the class. class: 'Foo', class_namespace: 'Bar' looks for ::Bar::Foo.)
:clearer: Proc used to define the private remove_all* method for doing the database work to remove all objects associated to the current object (*_to_many assocations). Set to nil to not define a remove_all_* method for the association.
:clone: Merge the current options and block into the options and block used in defining the given association. Can be used to DRY up a bunch of similar associations that all share the same options such as :class and :key, while changing the order and block used.
:conditions: The conditions to use to filter the association, can be any argument passed to where. This option is not respected when using eager_graph or association_join, unless it is hash or array of two element arrays. Consider also specifying the :graph_block option if the value for this option is not a hash or array of two element arrays and you plan to use this association in eager_graph or association_join.
:dataset: A proc that is used to define the method to get the base dataset to use (before the other options are applied). If the proc accepts an argument, it is passed the related association reflection. It is a best practice to always have the dataset accept an argument and use the argument to return the appropriate dataset.
:distinct: Use the DISTINCT clause when selecting associating object, both when lazy loading and eager loading via .eager (but not when using .eager_graph).
:eager: The associations to eagerly load via eager when loading the associated object(s).
:eager_block: If given, use the block instead of the default block when eagerly loading. To not use a block when eager loading (when one is used normally), set to nil.
:eager_graph: The associations to eagerly load via eager_graph when loading the associated object(s). many_to_many associations with this option cannot be eagerly loaded via eager.
:eager_grapher: A proc to use to implement eager loading via eager_graph, overriding the default. Takes an options hash with at least the entries :self (the receiver of the eager_graph call), :table_alias (the alias to use for table to graph into the association), and :implicit_qualifier (the alias that was used for the current table). Should return a copy of the dataset with the association graphed into it.
:eager_limit_strategy: Determines the strategy used for enforcing limits and offsets when eager loading associations via the eager method.
:eager_loader: A proc to use to implement eager loading, overriding the default. Takes a single hash argument, with at least the keys: :rows, which is an array of current model instances, :associations, which is a hash of dependent associations, :self, which is the dataset doing the eager loading, :eager_block, which is a dynamic callback that should be called with the dataset, and :id_map, which is a mapping of key values to arrays of current model instances. In the proc, the associated records should be queried from the database and the associations cache for each record should be populated.
:eager_loader_key: A symbol for the key column to use to populate the key_hash for the eager loader. Can be set to nil to not populate the key_hash.
:eager_loading_predicate_transform: A callable object with which to transform the predicate key values used when eager loading. Called with two arguments, the array of predicate key values, and a the reflection for the association being eager loaded.
:extend: A module or array of modules to extend the dataset with.
:filter_limit_strategy: Determines the strategy used for enforcing limits and offsets when filtering by limited associations. Possible options are :window_function, :distinct_on, or :correlated_subquery depending on association type and database type.
:graph_alias_base: The base name to use for the table alias when eager graphing. Defaults to the name of the association. If the alias name has already been used in the query, Sequel will create a unique alias by appending a numeric suffix (e.g. alias_0, alias_1, …) until the alias is unique.
:graph_block: The block to pass to join_table when eagerly loading the association via eager_graph.
:graph_conditions: The additional conditions to use on the SQL join when eagerly loading the association via eager_graph. Should be a hash or an array of two element arrays. If not specified, the :conditions option is used if it is a hash or array of two element arrays.
:graph_join_type: The type of SQL join to use when eagerly loading the association via eager_graph. Defaults to :left_outer.
:graph_only_conditions: The conditions to use on the SQL join when eagerly loading the association via eager_graph, instead of the default conditions specified by the foreign/primary keys. This option causes the :graph_conditions option to be ignored.
:graph_order: the order to use when using eager_graph, instead of the default order. This should be used in the case where :order contains an identifier qualified by the table's name, which may not match the alias used when eager graphing. By setting this to the unqualified identifier, it will be automatically qualified when using eager_graph.
:graph_select: A column or array of columns to select from the associated table when eagerly loading the association via eager_graph. Defaults to all columns in the associated table.
:graph_use_association_block: Makes eager_graph consider the association block. Without this, eager_graph ignores the bock and only use the :graph_* options.
:instance_specific: Marks the association as instance specific. Should be used if the association block uses instance specific state, or transient state (accessing current date/time, etc.).
:limit: Limit the number of records to the provided value. Use an array with two elements for the value to specify a limit (first element) and an offset (second element).
:methods_module: The module that methods the association creates will be placed into. Defaults to the module containing the model's columns.
:no_association_method: Do not add a method for the association. This can save memory if the association method is never used.
:no_dataset_method: Do not add a method for the association dataset. This can save memory if the dataset method is never used.
:order: the column(s) by which to order the association dataset. Can be a singular column symbol or an array of column symbols.
:order_eager_graph: Whether to add the association's order to the graphed dataset's order when graphing via eager_graph. Defaults to true, so set to false to disable.
:read_only: Do not add a setter method (for many_to_one or one_to_one associations), or add_/remove_/remove_all_ methods (for one_to_many and many_to_many associations).
:reciprocal: the symbol name of the reciprocal association, if it exists. By default, Sequel will try to determine it by looking at the associated model's assocations for a association that matches the current association's key(s). Set to nil to not use a reciprocal.
:remover: Proc used to define the private remove* method for doing the database work to remove the association between the given object and the current object (*_to_many assocations). Set to nil to not define a remove_* method for the association.
:select: the columns to select. Defaults to the associated class's table_name.* in an association that uses joins, which means it doesn't include the attributes from the join table. If you want to include the join table attributes, you can use this option, but beware that the join table attributes can clash with attributes from the model table, so you should alias any attributes that have the same name in both the join table and the associated table.
:setter: Proc used to define the private _*= method for doing the work to setup the assocation between the given object and the current object (*_to_one associations). Set to nil to not define a setter method for the association.
:subqueries_per_union: The number of subqueries to use in each UNION query, for eager loading limited associations using the default :union strategy.
:use_placeholder_loader: Whether to use a placeholder loader when eager loading the association. Can be set to false to disable the use of a placeholder loader if one would be used by default.
:validate: Set to false to not validate when implicitly saving any associated object.

:many_to_one¶ ↑

:key: foreign key in current model's table that references associated model's primary key, as a symbol. Defaults to :“#{name}_id”. Can use an array of symbols for a composite key association.
:key_column: Similar to, and usually identical to, :key, but :key refers to the model method to call, where :key_column refers to the underlying column. Should only be used if the model method differs from the foreign key column, in conjunction with defining a model alias method for the key column.
:primary_key: column in the associated table that :key option references, as a symbol. Defaults to the primary key of the associated table. Can use an array of symbols for a composite key association.
:primary_key_method: the method symbol or array of method symbols to call on the associated object to get the foreign key values. Defaults to :primary_key option.
:qualify: Whether to use qualified primary keys when loading the association. The default is true, so you must set to false to not qualify. Qualification rarely causes problems, but it's necessary to disable in some cases, such as when you are doing a JOIN USING operation on the column on Oracle.

:one_to_many and :one_to_one¶ ↑

:key: foreign key in associated model's table that references current model's primary key, as a symbol. Defaults to :“#{self.name.underscore}_id”. Can use an array of symbols for a composite key association.
:key_method: the method symbol or array of method symbols to call on the associated object to get the foreign key values. Defaults to :key option.
:primary_key: column in the current table that :key option references, as a symbol. Defaults to primary key of the current table. Can use an array of symbols for a composite key association.
:primary_key_column: Similar to, and usually identical to, :primary_key, but :primary_key refers to the model method call, where :primary_key_column refers to the underlying column. Should only be used if the model method differs from the primary key column, in conjunction with defining a model alias method for the primary key column.
:raise_on_save_failure: Do not raise exceptions for hook or validation failures when saving associated objects in the add/remove methods (return nil instead) [one_to_many only].

:many_to_many and :one_through_one¶ ↑

:graph_join_table_block: The block to pass to join_table for the join table when eagerly loading the association via eager_graph.
:graph_join_table_conditions: The additional conditions to use on the SQL join for the join table when eagerly loading the association via eager_graph. Should be a hash or an array of two element arrays.
:graph_join_table_join_type: The type of SQL join to use for the join table when eagerly loading the association via eager_graph. Defaults to the :graph_join_type option or :left_outer.
:graph_join_table_only_conditions: The conditions to use on the SQL join for the join table when eagerly loading the association via eager_graph, instead of the default conditions specified by the foreign/primary keys. This option causes the :graph_join_table_conditions option to be ignored.
:join_table: name of table that includes the foreign keys to both the current model and the associated model, as a symbol. Defaults to the name of current model and name of associated model, pluralized, underscored, sorted, and joined with '_'.
:join_table_block: proc that can be used to modify the dataset used in the add/remove/remove_all methods. Should accept a dataset argument and return a modified dataset if present.
:join_table_db: When retrieving records when using lazy loading or eager loading via eager, instead of a join between to the join table and the associated table, use a separate query for the join table using the given Database object.
:left_key: foreign key in join table that points to current model's primary key, as a symbol. Defaults to :“#{self.name.underscore}_id”. Can use an array of symbols for a composite key association.
:left_primary_key: column in current table that :left_key points to, as a symbol. Defaults to primary key of current table. Can use an array of symbols for a composite key association.
:left_primary_key_column: Similar to, and usually identical to, :left_primary_key, but :left_primary_key refers to the model method to call, where :left_primary_key_column refers to the underlying column. Should only be used if the model method differs from the left primary key column, in conjunction with defining a model alias method for the left primary key column.
:right_key: foreign key in join table that points to associated model's primary key, as a symbol. Defaults to :“#{name.to_s.singularize}_id”. Can use an array of symbols for a composite key association.
:right_primary_key: column in associated table that :right_key points to, as a symbol. Defaults to primary key of the associated table. Can use an array of symbols for a composite key association.
:right_primary_key_method: the method symbol or array of method symbols to call on the associated object to get the foreign key values for the join table. Defaults to :right_primary_key option.
:uniq: Adds a after_load callback that makes the array of objects unique.

     # File lib/sequel/model/associations.rb
2038 def associate(type, name, opts = OPTS, &block)
2039   raise(Error, 'invalid association type') unless assoc_class = Sequel.synchronize{ASSOCIATION_TYPES[type]}
2040   raise(Error, 'Model.associate name argument must be a symbol') unless name.is_a?(Symbol)
2041 
2042   # dup early so we don't modify opts
2043   orig_opts = opts.dup
2044 
2045   if opts[:clone]
2046     cloned_assoc = association_reflection(opts[:clone])
2047     remove_class_name = orig_opts[:class] && !orig_opts[:class_name]
2048     orig_opts = cloned_assoc[:orig_opts].merge(orig_opts)
2049     orig_opts.delete(:class_name) if remove_class_name
2050   end
2051 
2052   opts = Hash[default_association_options]
2053   if type_options = default_association_type_options[type]
2054     opts.merge!(type_options)
2055   end
2056   opts.merge!(orig_opts)
2057   opts.merge!(:type => type, :name => name, :cache=>({} if cache_associations), :model => self)
2058 
2059   opts[:block] = block if block
2060   opts[:instance_specific] = true if orig_opts[:dataset]
2061   if !opts.has_key?(:instance_specific) && (block || orig_opts[:block])
2062     # It's possible the association is instance specific, in that it depends on
2063     # values other than the foreign key value.  This needs to be checked for
2064     # in certain places to disable optimizations.
2065     opts[:instance_specific] = _association_instance_specific_default(name)
2066   end
2067   if (orig_opts[:instance_specific] || orig_opts[:dataset]) && !opts.has_key?(:allow_eager) && !opts[:eager_loader]
2068     # For associations explicitly marked as instance specific, or that use the
2069     # :dataset option, where :allow_eager is not set, and no :eager_loader is
2070     # provided, disallow eager loading.  In these cases, eager loading is
2071     # unlikely to work.  This is not done for implicit setting of :instance_specific,
2072     # because implicit use is done by default for all associations with blocks,
2073     # and the vast majority of associations with blocks use the block for filtering
2074     # in a manner compatible with eager loading.
2075     opts[:allow_eager] = false
2076   end
2077   opts = assoc_class.new.merge!(opts)
2078 
2079   if opts[:clone] && !opts.cloneable?(cloned_assoc)
2080     raise(Error, "cannot clone an association to an association of different type (association #{name} with type #{type} cloning #{opts[:clone]} with type #{cloned_assoc[:type]})")
2081   end
2082 
2083   opts[:use_placeholder_loader] = !opts[:instance_specific] && !opts[:eager_graph] unless opts.include?(:use_placeholder_loader)
2084   opts[:eager_block] = opts[:block] unless opts.include?(:eager_block)
2085   opts[:graph_join_type] ||= :left_outer
2086   opts[:order_eager_graph] = true unless opts.include?(:order_eager_graph)
2087   conds = opts[:conditions]
2088   opts[:graph_alias_base] ||= name
2089   opts[:graph_conditions] = conds if !opts.include?(:graph_conditions) and Sequel.condition_specifier?(conds)
2090   opts[:graph_conditions] = opts.fetch(:graph_conditions, []).to_a
2091   opts[:graph_select] = Array(opts[:graph_select]) if opts[:graph_select]
2092   [:before_add, :before_remove, :after_add, :after_remove, :after_load, :before_set, :after_set].each do |cb_type|
2093     opts[cb_type] = Array(opts[cb_type]) if opts[cb_type]
2094   end
2095 
2096   if opts[:extend]
2097     opts[:extend] = Array(opts[:extend])
2098     opts[:reverse_extend] = opts[:extend].reverse
2099   end
2100 
2101   late_binding_class_option(opts, opts.returns_array? ? singularize(name) : name)
2102   
2103   # Remove :class entry if it exists and is nil, to work with cached_fetch
2104   opts.delete(:class) unless opts[:class]
2105 
2106   opts[:_hash] = [self, name].hash
2107 
2108   def_association(opts)
2109       
2110   orig_opts.delete(:clone)
2111   opts[:orig_class] = orig_opts[:class] || orig_opts[:class_name]
2112   orig_opts.merge!(:class_name=>opts[:class_name], :class=>opts[:class], :block=>opts[:block])
2113   opts[:orig_opts] = orig_opts
2114   # don't add to association_reflections until we are sure there are no errors
2115   association_reflections[name] = opts
2116 end

association_reflection(name) click to toggle source

The association reflection hash for the association of the given name.

     # File lib/sequel/model/associations.rb
2119 def association_reflection(name)
2120   association_reflections[name]
2121 end

associations() click to toggle source

Array of association name symbols

     # File lib/sequel/model/associations.rb
2124 def associations
2125   association_reflections.keys
2126 end

eager_load_results(opts, eo, &block) click to toggle source

Eager load the association with the given eager loader options.

     # File lib/sequel/model/associations.rb
2129 def eager_load_results(opts, eo, &block)
2130   opts.eager_load_results(eo, &block)
2131 end

finalize_associations() click to toggle source

Finalize all associations such that values that are looked up dynamically in associated classes are set statically. As this modifies the associations, it must be done before calling freeze.

     # File lib/sequel/model/associations.rb
2148 def finalize_associations
2149   @association_reflections.each_value(&:finalize)
2150 end

freeze() click to toggle source

Freeze association related metadata when freezing model class.

Calls superclass method

     # File lib/sequel/model/associations.rb
2134 def freeze
2135   @association_reflections.freeze.each_value(&:freeze)
2136   @autoreloading_associations.freeze.each_value(&:freeze)
2137   @default_association_options.freeze
2138   @default_association_type_options.freeze
2139   @default_association_type_options.each_value(&:freeze)
2140 
2141   super
2142 end

many_to_many(name, opts=OPTS, &block) click to toggle source

Shortcut for adding a many_to_many association, see associate

     # File lib/sequel/model/associations.rb
2153 def many_to_many(name, opts=OPTS, &block)
2154   associate(:many_to_many, name, opts, &block)
2155 end

many_to_one(name, opts=OPTS, &block) click to toggle source

Shortcut for adding a many_to_one association, see associate

     # File lib/sequel/model/associations.rb
2158 def many_to_one(name, opts=OPTS, &block)
2159   associate(:many_to_one, name, opts, &block)
2160 end

one_through_one(name, opts=OPTS, &block) click to toggle source

Shortcut for adding a one_through_one association, see associate

     # File lib/sequel/model/associations.rb
2163 def one_through_one(name, opts=OPTS, &block)
2164   associate(:one_through_one, name, opts, &block)
2165 end

one_to_many(name, opts=OPTS, &block) click to toggle source

Shortcut for adding a one_to_many association, see associate

     # File lib/sequel/model/associations.rb
2168 def one_to_many(name, opts=OPTS, &block)
2169   associate(:one_to_many, name, opts, &block)
2170 end

one_to_one(name, opts=OPTS, &block) click to toggle source

Shortcut for adding a one_to_one association, see associate

     # File lib/sequel/model/associations.rb
2173 def one_to_one(name, opts=OPTS, &block)
2174   associate(:one_to_one, name, opts, &block)
2175 end

Private Instance Methods

_association_instance_specific_default(_) click to toggle source

The default value for the instance_specific option, if the association could be instance specific and the :instance_specific option is not specified.

     # File lib/sequel/model/associations.rb
2185 def _association_instance_specific_default(_)
2186   true
2187 end

association_module(opts=OPTS) click to toggle source

The module to use for the association's methods. Defaults to the overridable_methods_module.

     # File lib/sequel/model/associations.rb
2191 def association_module(opts=OPTS)
2192   opts.fetch(:methods_module, overridable_methods_module)
2193 end

association_module_def(name, opts=OPTS, &block) click to toggle source

Add a method to the module included in the class, so the method can be easily overridden in the class itself while allowing for super to be called.

     # File lib/sequel/model/associations.rb
2198 def association_module_def(name, opts=OPTS, &block)
2199   mod = association_module(opts)
2200   mod.send(:define_method, name, &block)
2201   mod.send(:alias_method, name, name)
2202 end

association_module_delegate_def(name, opts, &block) click to toggle source

Add a method to the module included in the class, so the method can be easily overridden in the class itself while allowing for super to be called. This method allows passing keywords through the defined methods.

     # File lib/sequel/model/associations.rb
2208 def association_module_delegate_def(name, opts, &block)
2209   mod = association_module(opts)
2210   mod.send(:define_method, name, &block)
2211   # :nocov:
2212   mod.send(:ruby2_keywords, name) if mod.respond_to?(:ruby2_keywords, true)
2213   # :nocov:
2214   mod.send(:alias_method, name, name)
2215 end

association_module_private_def(name, opts=OPTS, &block) click to toggle source

Add a private method to the module included in the class.

     # File lib/sequel/model/associations.rb
2218 def association_module_private_def(name, opts=OPTS, &block)
2219   association_module_def(name, opts, &block)
2220   association_module(opts).send(:private, name)
2221 end

def_association(opts) click to toggle source

Delegate to the type-specific association method to setup the association, and define the association instance methods.

     # File lib/sequel/model/associations.rb
2225 def def_association(opts)
2226   send(:"def_#{opts[:type]}", opts)
2227   def_association_instance_methods(opts)
2228 end

def_association_instance_methods(opts) click to toggle source

Define all of the association instance methods for this association.

     # File lib/sequel/model/associations.rb
2238 def def_association_instance_methods(opts)
2239   # Always set the method names in the association reflection, even if they
2240   # are not used, for backwards compatibility.
2241   opts[:dataset_method] = :"#{opts[:name]}_dataset"
2242   if opts.returns_array?
2243     sname = singularize(opts[:name])
2244     opts[:_add_method] = :"_add_#{sname}"
2245     opts[:add_method] = :"add_#{sname}"
2246     opts[:_remove_method] = :"_remove_#{sname}"
2247     opts[:remove_method] = :"remove_#{sname}"
2248     opts[:_remove_all_method] = :"_remove_all_#{opts[:name]}"
2249     opts[:remove_all_method] = :"remove_all_#{opts[:name]}"
2250   else
2251     opts[:_setter_method] = :"_#{opts[:name]}="
2252     opts[:setter_method] = :"#{opts[:name]}="
2253   end
2254 
2255   association_module_def(opts.dataset_method, opts){_dataset(opts)} unless opts[:no_dataset_method]
2256   if opts[:block]
2257     opts[:block_method] = Plugins.def_sequel_method(association_module(opts), "#{opts[:name]}_block", 1, &opts[:block])
2258   end
2259   opts[:dataset_opt_arity] = opts[:dataset].arity == 0 ? 0 : 1
2260   opts[:dataset_opt_method] = Plugins.def_sequel_method(association_module(opts), "#{opts[:name]}_dataset_opt", opts[:dataset_opt_arity], &opts[:dataset])
2261   def_association_method(opts) unless opts[:no_association_method]
2262 
2263   return if opts[:read_only]
2264 
2265   if opts[:setter] && opts[:_setter]
2266     # This is backwards due to backwards compatibility
2267     association_module_private_def(opts[:_setter_method], opts, &opts[:setter])
2268     association_module_def(opts[:setter_method], opts, &opts[:_setter])
2269   end
2270 
2271   if adder = opts[:adder]
2272     association_module_private_def(opts[:_add_method], opts, &adder)
2273     association_module_delegate_def(opts[:add_method], opts){|o,*args| add_associated_object(opts, o, *args)}
2274   end
2275 
2276   if remover = opts[:remover]
2277     association_module_private_def(opts[:_remove_method], opts, &remover)
2278     association_module_delegate_def(opts[:remove_method], opts){|o,*args| remove_associated_object(opts, o, *args)}
2279   end
2280 
2281   if clearer = opts[:clearer]
2282     association_module_private_def(opts[:_remove_all_method], opts, &clearer)
2283     association_module_delegate_def(opts[:remove_all_method], opts){|*args| remove_all_associated_objects(opts, *args)}
2284   end
2285 end

def_association_method(opts) click to toggle source

Adds the association method to the association methods module.

     # File lib/sequel/model/associations.rb
2231 def def_association_method(opts)
2232   association_module_def(opts.association_method, opts) do |dynamic_opts=OPTS, &block|
2233     load_associated_objects(opts, dynamic_opts, &block)
2234   end
2235 end

def_many_to_many(opts) click to toggle source

Configures many_to_many and one_through_one association reflection and adds the related association methods

     # File lib/sequel/model/associations.rb
2288 def def_many_to_many(opts)
2289   one_through_one = opts[:type] == :one_through_one
2290   left = (opts[:left_key] ||= opts.default_left_key)
2291   lcks = opts[:left_keys] = Array(left)
2292   right = (opts[:right_key] ||= opts.default_right_key)
2293   rcks = opts[:right_keys] = Array(right)
2294   left_pk = (opts[:left_primary_key] ||= self.primary_key)
2295   opts[:eager_loader_key] = left_pk unless opts.has_key?(:eager_loader_key)
2296   lcpks = opts[:left_primary_keys] = Array(left_pk)
2297   lpkc = opts[:left_primary_key_column] ||= left_pk
2298   lpkcs = opts[:left_primary_key_columns] ||= Array(lpkc)
2299   raise(Error, "mismatched number of left keys: #{lcks.inspect} vs #{lcpks.inspect}") unless lcks.length == lcpks.length
2300   if opts[:right_primary_key]
2301     rcpks = Array(opts[:right_primary_key])
2302     raise(Error, "mismatched number of right keys: #{rcks.inspect} vs #{rcpks.inspect}") unless rcks.length == rcpks.length
2303   end
2304   opts[:uses_left_composite_keys] = lcks.length > 1
2305   uses_rcks = opts[:uses_right_composite_keys] = rcks.length > 1
2306   opts[:cartesian_product_number] ||= one_through_one ? 0 : 1
2307   join_table = (opts[:join_table] ||= opts.default_join_table)
2308   opts[:left_key_alias] ||= opts.default_associated_key_alias
2309   opts[:graph_join_table_join_type] ||= opts[:graph_join_type]
2310   if opts[:uniq]
2311     opts[:after_load] ||= []
2312     opts[:after_load].unshift(:array_uniq!)
2313   end
2314   if join_table_db = opts[:join_table_db]
2315     opts[:use_placeholder_loader] = false
2316     opts[:allow_eager_graph] = false
2317     opts[:allow_filtering_by] = false
2318     opts[:eager_limit_strategy] = nil
2319     join_table_ds = join_table_db.from(join_table)
2320     opts[:dataset] ||= proc do |r|
2321       vals = join_table_ds.where(lcks.zip(lcpks.map{|k| get_column_value(k)})).select_map(right)
2322       ds = r.associated_dataset.where(opts.right_primary_key => vals)
2323       if uses_rcks
2324         vals.delete_if{|v| v.any?(&:nil?)}
2325       else
2326         vals.delete(nil)
2327       end
2328       ds = ds.clone(:no_results=>true) if vals.empty?
2329       ds
2330     end
2331     opts[:eager_loader] ||= proc do |eo|
2332       h = eo[:id_map]
2333       assign_singular = opts.assign_singular?
2334       rpk = opts.right_primary_key
2335       name = opts[:name]
2336 
2337       join_map = join_table_ds.where(left=>h.keys).select_hash_groups(right, left)
2338 
2339       if uses_rcks
2340         join_map.delete_if{|v,| v.any?(&:nil?)}
2341       else
2342         join_map.delete(nil)
2343       end
2344 
2345       eo = Hash[eo]
2346 
2347       if join_map.empty?
2348         eo[:no_results] = true
2349       else
2350         join_map.each_value do |vs|
2351           vs.replace(vs.flat_map{|v| h[v]})
2352           vs.uniq!
2353         end
2354 
2355         eo[:loader] = false
2356         eo[:right_keys] = join_map.keys
2357       end
2358 
2359       opts[:model].eager_load_results(opts, eo) do |assoc_record|
2360         rpkv = if uses_rcks
2361           assoc_record.values.values_at(*rpk)
2362         else
2363           assoc_record.values[rpk]
2364         end
2365 
2366         objects = join_map[rpkv]
2367 
2368         if assign_singular
2369           objects.each do |object|
2370             object.associations[name] ||= assoc_record
2371           end
2372         else
2373           objects.each do |object|
2374             object.associations[name].push(assoc_record)
2375           end
2376         end
2377       end
2378     end
2379   else
2380     opts[:dataset] ||= opts.association_dataset_proc
2381     opts[:eager_loader] ||= opts.method(:default_eager_loader)
2382   end
2383   
2384   join_type = opts[:graph_join_type]
2385   select = opts[:graph_select]
2386   use_only_conditions = opts.include?(:graph_only_conditions)
2387   only_conditions = opts[:graph_only_conditions]
2388   conditions = opts[:graph_conditions]
2389   graph_block = opts[:graph_block]
2390   graph_jt_conds = opts[:graph_join_table_conditions] = opts.fetch(:graph_join_table_conditions, []).to_a
2391   use_jt_only_conditions = opts.include?(:graph_join_table_only_conditions)
2392   jt_only_conditions = opts[:graph_join_table_only_conditions]
2393   jt_join_type = opts[:graph_join_table_join_type]
2394   jt_graph_block = opts[:graph_join_table_block]
2395   opts[:eager_grapher] ||= proc do |eo|
2396     ds = eo[:self]
2397     egls = eo[:limit_strategy]
2398     if egls && egls != :ruby
2399       associated_key_array = opts.associated_key_array
2400       orig_egds = egds = eager_graph_dataset(opts, eo)
2401       egds = egds.
2402         inner_join(join_table, rcks.zip(opts.right_primary_keys) + graph_jt_conds, :qualify=>:deep).
2403         select_all(egds.first_source).
2404         select_append(*associated_key_array)
2405       egds = opts.apply_eager_graph_limit_strategy(egls, egds)
2406       ds.graph(egds, associated_key_array.map(&:alias).zip(lpkcs) + conditions, :qualify=>:deep, :table_alias=>eo[:table_alias], :implicit_qualifier=>eo[:implicit_qualifier], :join_type=>eo[:join_type]||join_type, :from_self_alias=>eo[:from_self_alias], :join_only=>eo[:join_only], :select=>select||orig_egds.columns, &graph_block)
2407     else
2408       ds = ds.graph(join_table, use_jt_only_conditions ? jt_only_conditions : lcks.zip(lpkcs) + graph_jt_conds, :select=>false, :table_alias=>ds.unused_table_alias(join_table, [eo[:table_alias]]), :join_type=>eo[:join_type]||jt_join_type, :join_only=>eo[:join_only], :implicit_qualifier=>eo[:implicit_qualifier], :qualify=>:deep, :from_self_alias=>eo[:from_self_alias], &jt_graph_block)
2409       ds.graph(eager_graph_dataset(opts, eo), use_only_conditions ? only_conditions : opts.right_primary_keys.zip(rcks) + conditions, :select=>select, :table_alias=>eo[:table_alias], :qualify=>:deep, :join_type=>eo[:join_type]||join_type, :join_only=>eo[:join_only], &graph_block)
2410     end
2411   end
2412       
2413   return if opts[:read_only]
2414       
2415   if one_through_one
2416     unless opts.has_key?(:setter)
2417       opts[:setter] = proc do |o|
2418         h = {}
2419         lh = lcks.zip(lcpks.map{|k| get_column_value(k)})
2420         jtds = _join_table_dataset(opts).where(lh)
2421 
2422         checked_transaction do
2423           current = jtds.first
2424 
2425           if o
2426             new_values = []
2427             rcks.zip(opts.right_primary_key_methods).each{|k, pk| new_values << (h[k] = o.get_column_value(pk))}
2428           end
2429 
2430           if current
2431             current_values = rcks.map{|k| current[k]}
2432             jtds = jtds.where(rcks.zip(current_values))
2433             if o
2434               if current_values != new_values
2435                 jtds.update(h)
2436               end
2437             else
2438               jtds.delete
2439             end
2440           elsif o
2441             lh.each{|k,v| h[k] = v}
2442             jtds.insert(h)
2443           end
2444         end
2445       end
2446     end
2447     if opts.fetch(:setter, true)
2448       opts[:_setter] = proc{|o| set_one_through_one_associated_object(opts, o)}
2449     end
2450   else 
2451     unless opts.has_key?(:adder)
2452       opts[:adder] = proc do |o|
2453         h = {}
2454         lcks.zip(lcpks).each{|k, pk| h[k] = get_column_value(pk)}
2455         rcks.zip(opts.right_primary_key_methods).each{|k, pk| h[k] = o.get_column_value(pk)}
2456         _join_table_dataset(opts).insert(h)
2457       end
2458     end
2459 
2460     unless opts.has_key?(:remover)
2461       opts[:remover] = proc do |o|
2462         _join_table_dataset(opts).where(lcks.zip(lcpks.map{|k| get_column_value(k)}) + rcks.zip(opts.right_primary_key_methods.map{|k| o.get_column_value(k)})).delete
2463       end
2464     end
2465 
2466     unless opts.has_key?(:clearer)
2467       opts[:clearer] = proc do
2468         _join_table_dataset(opts).where(lcks.zip(lcpks.map{|k| get_column_value(k)})).delete
2469       end
2470     end
2471   end
2472 end

def_many_to_one(opts) click to toggle source

Configures many_to_one association reflection and adds the related association methods

     # File lib/sequel/model/associations.rb
2475 def def_many_to_one(opts)
2476   name = opts[:name]
2477   opts[:key] = opts.default_key unless opts.has_key?(:key)
2478   key = opts[:key]
2479   opts[:eager_loader_key] = key unless opts.has_key?(:eager_loader_key)
2480   cks = opts[:graph_keys] = opts[:keys] = Array(key)
2481   opts[:key_column] ||= key
2482   opts[:graph_keys] = opts[:key_columns] = Array(opts[:key_column])
2483   opts[:qualified_key] = opts.qualify_cur(key)
2484   if opts[:primary_key]
2485     cpks = Array(opts[:primary_key])
2486     raise(Error, "mismatched number of keys: #{cks.inspect} vs #{cpks.inspect}") unless cks.length == cpks.length
2487   end
2488   uses_cks = opts[:uses_composite_keys] = cks.length > 1
2489   opts[:cartesian_product_number] ||= 0
2490 
2491   if !opts.has_key?(:many_to_one_pk_lookup) &&
2492      (opts[:dataset] || opts[:conditions] || opts[:block] || opts[:select] ||
2493       (opts.has_key?(:key) && opts[:key] == nil))
2494     opts[:many_to_one_pk_lookup] = false
2495   end
2496   auto_assocs = @autoreloading_associations
2497   cks.each do |k|
2498     (auto_assocs[k] ||= []) << name
2499   end
2500 
2501   opts[:dataset] ||= opts.association_dataset_proc
2502   opts[:eager_loader] ||= proc do |eo|
2503     h = eo[:id_map]
2504     pk_meths = opts.primary_key_methods
2505 
2506     eager_load_results(opts, eo) do |assoc_record|
2507       hash_key = uses_cks ? pk_meths.map{|k| assoc_record.get_column_value(k)} : assoc_record.get_column_value(opts.primary_key_method)
2508       h[hash_key].each{|object| object.associations[name] = assoc_record}
2509     end
2510   end
2511       
2512   join_type = opts[:graph_join_type]
2513   select = opts[:graph_select]
2514   use_only_conditions = opts.include?(:graph_only_conditions)
2515   only_conditions = opts[:graph_only_conditions]
2516   conditions = opts[:graph_conditions]
2517   graph_block = opts[:graph_block]
2518   graph_cks = opts[:graph_keys]
2519   opts[:eager_grapher] ||= proc do |eo|
2520     ds = eo[:self]
2521     ds.graph(eager_graph_dataset(opts, eo), use_only_conditions ? only_conditions : opts.primary_keys.zip(graph_cks) + conditions, eo.merge(:select=>select, :join_type=>eo[:join_type]||join_type, :qualify=>:deep), &graph_block)
2522   end
2523       
2524   return if opts[:read_only]
2525       
2526   unless opts.has_key?(:setter)
2527     opts[:setter] = proc{|o| cks.zip(opts.primary_key_methods).each{|k, pk| set_column_value(:"#{k}=", (o.get_column_value(pk) if o))}}
2528   end
2529   if opts.fetch(:setter, true)
2530     opts[:_setter] = proc{|o| set_associated_object(opts, o)}
2531   end
2532 end

def_one_through_one(opts) click to toggle source

Alias of def_many_to_many, since they share pretty much the same code.

     # File lib/sequel/model/associations.rb
2661 def def_one_through_one(opts)
2662   def_many_to_many(opts)
2663 end

def_one_to_many(opts) click to toggle source

Configures one_to_many and one_to_one association reflections and adds the related association methods

     # File lib/sequel/model/associations.rb
2535 def def_one_to_many(opts)
2536   one_to_one = opts[:type] == :one_to_one
2537   name = opts[:name]
2538   key = (opts[:key] ||= opts.default_key)
2539   km = opts[:key_method] ||= opts[:key]
2540   cks = opts[:keys] = Array(key)
2541   opts[:key_methods] = Array(opts[:key_method])
2542   primary_key = (opts[:primary_key] ||= self.primary_key)
2543   opts[:eager_loader_key] = primary_key unless opts.has_key?(:eager_loader_key)
2544   cpks = opts[:primary_keys] = Array(primary_key)
2545   pkc = opts[:primary_key_column] ||= primary_key
2546   pkcs = opts[:primary_key_columns] ||= Array(pkc)
2547   raise(Error, "mismatched number of keys: #{cks.inspect} vs #{cpks.inspect}") unless cks.length == cpks.length
2548   uses_cks = opts[:uses_composite_keys] = cks.length > 1
2549   opts[:dataset] ||= opts.association_dataset_proc
2550   opts[:eager_loader] ||= proc do |eo|
2551     h = eo[:id_map]
2552     reciprocal = opts.reciprocal
2553     assign_singular = opts.assign_singular?
2554     delete_rn = opts.delete_row_number_column
2555 
2556     eager_load_results(opts, eo) do |assoc_record|
2557       assoc_record.remove_key!(delete_rn) if delete_rn
2558       hash_key = uses_cks ? km.map{|k| assoc_record.get_column_value(k)} : assoc_record.get_column_value(km)
2559       objects = h[hash_key]
2560       if assign_singular
2561         objects.each do |object| 
2562           unless object.associations[name]
2563             object.associations[name] = assoc_record
2564             assoc_record.associations[reciprocal] = object if reciprocal
2565           end
2566         end
2567       else
2568         objects.each do |object| 
2569           object.associations[name].push(assoc_record)
2570           assoc_record.associations[reciprocal] = object if reciprocal
2571         end
2572       end
2573     end
2574   end
2575   
2576   join_type = opts[:graph_join_type]
2577   select = opts[:graph_select]
2578   use_only_conditions = opts.include?(:graph_only_conditions)
2579   only_conditions = opts[:graph_only_conditions]
2580   conditions = opts[:graph_conditions]
2581   opts[:cartesian_product_number] ||= one_to_one ? 0 : 1
2582   graph_block = opts[:graph_block]
2583   graph_conditions = opts[:_graph_conditions] = use_only_conditions ? only_conditions : cks.zip(pkcs) + conditions
2584   opts[:eager_grapher] ||= proc do |eo|
2585     ds = eo[:self]
2586     graph_limit_strategy = eo[:limit_strategy]
2587     egds = opts.apply_eager_graph_limit_strategy(graph_limit_strategy, eager_graph_dataset(opts, eo))
2588     graph_conditions_true = true if graph_limit_strategy == :lateral_subquery
2589 
2590     ds = ds.graph(egds, graph_conditions_true || graph_conditions, eo.merge(:select=>select, :join_type=>eo[:join_type]||join_type, :qualify=>:deep), &graph_block)
2591     # We only load reciprocals for one_to_many associations, as other reciprocals don't make sense
2592     ds.opts[:eager_graph][:reciprocals][eo[:table_alias]] = opts.reciprocal
2593     ds
2594   end
2595       
2596   return if opts[:read_only]
2597 
2598   save_opts = {:validate=>opts[:validate]}
2599   ck_nil_hash ={}
2600   cks.each{|k| ck_nil_hash[k] = nil}
2601 
2602   if one_to_one
2603     unless opts.has_key?(:setter)
2604       opts[:setter] = proc do |o|
2605         up_ds = _apply_association_options(opts, opts.associated_dataset.where(cks.zip(cpks.map{|k| get_column_value(k)})))
2606 
2607         if (froms = up_ds.opts[:from]) && (from = froms[0]) && (from.is_a?(Sequel::Dataset) || (from.is_a?(Sequel::SQL::AliasedExpression) && from.expression.is_a?(Sequel::Dataset)))
2608           if old = up_ds.first
2609             cks.each{|k| old.set_column_value(:"#{k}=", nil)}
2610           end
2611           save_old = true
2612         end
2613 
2614         if o
2615           if !o.new? && !save_old
2616             up_ds = up_ds.exclude(o.pk_hash)
2617           end
2618           cks.zip(cpks).each{|k, pk| o.set_column_value(:"#{k}=", get_column_value(pk))}
2619         end
2620 
2621         checked_transaction do
2622           if save_old
2623             old.save(save_opts) || raise(Sequel::Error, "invalid previously associated object, cannot save") if old
2624           else
2625             up_ds.skip_limit_check.update(ck_nil_hash)
2626           end
2627 
2628           o.save(save_opts) || raise(Sequel::Error, "invalid associated object, cannot save") if o
2629         end
2630       end
2631     end
2632     if opts.fetch(:setter, true)
2633       opts[:_setter] = proc{|o| set_one_to_one_associated_object(opts, o)}
2634     end
2635   else 
2636     save_opts[:raise_on_failure] = opts[:raise_on_save_failure] != false
2637 
2638     unless opts.has_key?(:adder)
2639       opts[:adder] = proc do |o|
2640         cks.zip(cpks).each{|k, pk| o.set_column_value(:"#{k}=", get_column_value(pk))}
2641         o.save(save_opts)
2642       end
2643     end
2644     
2645     unless opts.has_key?(:remover)
2646       opts[:remover] = proc do |o|
2647         cks.each{|k| o.set_column_value(:"#{k}=", nil)}
2648         o.save(save_opts)
2649       end
2650     end
2651 
2652     unless opts.has_key?(:clearer)
2653       opts[:clearer] = proc do
2654         _apply_association_options(opts, opts.associated_dataset.where(cks.zip(cpks.map{|k| get_column_value(k)}))).update(ck_nil_hash)
2655       end
2656     end
2657   end
2658 end

def_one_to_one(opts) click to toggle source

Alias of def_one_to_many, since they share pretty much the same code.

     # File lib/sequel/model/associations.rb
2666 def def_one_to_one(opts)
2667   def_one_to_many(opts)
2668 end

eager_graph_dataset(opts, eager_options) click to toggle source

Return dataset to graph into given the association reflection, applying the :callback option if set.

     # File lib/sequel/model/associations.rb
2671 def eager_graph_dataset(opts, eager_options)
2672   ds = opts.associated_class.dataset
2673   if eager_options[:limit_strategy] == :lateral_subquery
2674     ds = ds.clone(:eager_options=>eager_options)
2675   end
2676   if opts[:graph_use_association_block] && (b = opts[:block])
2677     ds = b.call(ds)
2678   end
2679   if cb = eager_options[:callback]
2680     ds = cb.call(ds)
2681   end
2682   ds
2683 end

reload_db_schema?() click to toggle source

If not caching associations, reload the database schema by default, ignoring any cached values.

     # File lib/sequel/model/associations.rb
2687 def reload_db_schema?
2688   !@cache_associations
2689 end