/
association_relation.rb
50 lines (41 loc) · 1.2 KB
/
association_relation.rb
1
2
3
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# frozen_string_literal: true
module ActiveRecord
class AssociationRelation < Relation # :nodoc:
def initialize(klass, association, **)
super(klass)
@association = association
end
def proxy_association
@association
end
def ==(other)
other == records
end
%w(insert insert_all insert! insert_all! upsert upsert_all).each do |method|
class_eval <<~RUBY
def #{method}(attributes, **kwargs)
if @association.reflection.through_reflection?
raise ArgumentError, "Bulk insert or upsert is currently not supported for has_many through association"
end
scoping { klass.#{method}(attributes, **kwargs) }
end
RUBY
end
private
def _new(attributes, &block)
@association.build(attributes, &block)
end
def _create(attributes, &block)
@association.create(attributes, &block)
end
def _create!(attributes, &block)
@association.create!(attributes, &block)
end
def exec_queries
super do |record|
@association.set_inverse_instance_from_queries(record)
yield record if block_given?
end
end
end
end