Module: Volt::Associations::ClassMethods
- Defined in:
- lib/volt/models/associations.rb
Instance Method Summary collapse
- #belongs_to(method_name, options = {}) ⇒ Object
- #has_many(method_name, options = {}) ⇒ Object
-
#has_one(method_name) ⇒ Object
has_one creates a method on the Volt::Model that returns a promise to get the associated model.
Instance Method Details
#belongs_to(method_name, options = {}) ⇒ Object
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# File 'lib/volt/models/associations.rb', line 4 def belongs_to(method_name, = {}) collection ||= .fetch(:collection, method_name).pluralize foreign_key ||= .fetch(:foreign_key, :id) local_key ||= .fetch(:local_key, "#{method_name}_id") # Add a field for the association_id field(local_key) # getter define_method(method_name) do association_with_root_model('belongs_to') do |root| # Lookup the associated model id lookup_key = get(local_key) # Return a promise for the belongs_to root.get(collection).where(foreign_key => lookup_key).first end end define_method(:"#{method_name}=") do |obj| id = obj.is_a?(Fixnum) ? obj : obj.id # Assign the current model's something_id to the object's id set(local_key, id) end end |
#has_many(method_name, options = {}) ⇒ Object
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# File 'lib/volt/models/associations.rb', line 31 def has_many(method_name, = {}) collection ||= .fetch(:collection, method_name).pluralize # Use the underscored current class name as the something_id. foreign_key ||= .fetch(:foreign_key, "#{to_s.underscore.singularize}_id") local_key ||= .fetch(:local_key, :id) define_method(method_name) do lookup_key = get(local_key) array_model = root.get(collection).where(foreign_key => lookup_key) # Since we are coming off of the root collection, we need to setup # the right parent and path. new_path = array_model.[:path] # Assign path and parent array_model.path = self.path + new_path array_model.parent = self array_model end end |
#has_one(method_name) ⇒ Object
has_one creates a method on the Volt::Model that returns a promise to get the associated model.
55 56 57 58 59 60 61 62 63 64 65 66 |
# File 'lib/volt/models/associations.rb', line 55 def has_one(method_name) if method_name.plural? raise NameError, "has_one takes a singluar association name" end define_method(method_name) do association_with_root_model('has_one') do |root| key = self.class.to_s.underscore + '_id' root.send(method_name.pluralize).where(key => id).first end end end |