Module: Octopus::Association

Defined in:
lib/octopus/association.rb

Defined Under Namespace

Modules: InstanceMethods, QueryOnCurrentShard

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.extended(base) ⇒ Object 



2
3
4
 
# File 'lib/octopus/association.rb', line 2

def self.extended(base)
  base.send(:include, InstanceMethods)
end

Instance Method Details

#default_octopus_opts(options) ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/octopus/association.rb', line 84

def default_octopus_opts(options)
  if options[:before_add].is_a?(Array)
    options[:before_add] << :set_connection_on_association
  elsif options[:before_add].is_a?(Symbol)
    options[:before_add] = [:set_connection_on_association, options[:before_add]]
  else
    options[:before_add] = :set_connection_on_association
  end

  if options[:before_remove].is_a?(Array)
    options[:before_remove] << :set_connection_on_association
  elsif options[:before_remove].is_a?(Symbol)
    options[:before_remove] = [:set_connection_on_association, options[:before_remove]]
  else
    options[:before_remove] = :set_connection_on_association
  end

  options[:extend] = [ Octopus::Association::QueryOnCurrentShard, options[:extend] ].flatten.compact
end

#has_and_belongs_to_many(association_id, options = {}, &extension) ⇒ Object 



69
70
71
72
73
74
75
76
 
# File 'lib/octopus/association.rb', line 69

def has_and_belongs_to_many(association_id, scope=nil, options={}, &extension)
  if options == {} && scope.is_a?(Hash)
    default_octopus_opts(scope)
  else
    default_octopus_opts(options)
  end
  super
end

#has_many(association_id, options = {}, &extension) ⇒ Object 



52
53
54
55
56
57
58
59
 
# File 'lib/octopus/association.rb', line 52

def has_many(association_id, scope=nil, options={}, &extension)
  if options == {} && scope.is_a?(Hash)
    default_octopus_opts(scope)
  else
    default_octopus_opts(options)
  end
  super
end