aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/relation.rb
blob: b69a12bc7b43b24bf4177aeea8b6d381beb0c8fb (plain) (blame)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
module ActiveRecord
  class Relation
    delegate :to_sql, :to => :relation
    attr_reader :relation, :klass

    def initialize(klass, relation)
      @klass, @relation = klass, relation
    end

    def to_a
      @klass.find_by_sql(@relation.to_sql)
    end

    def each(&block)
      to_a.each(&block)
    end

    def first
      @relation = @relation.take(1)
      to_a.first
    end

    def select(selects)
      Relation.new(@klass, @relation.project(selects))
    end

    def group(groups)
      Relation.new(@klass, @relation.group(groups))
    end

    def order(orders)
      Relation.new(@klass, @relation.order(orders))
    end

    def limit(limits)
      Relation.new(@klass, @relation.take(limits))
    end

    def offset(offsets)
      Relation.new(@klass, @relation.skip(offsets))
    end

    def on(join)
      Relation.new(@klass, @relation.on(join))
    end

    def joins(join, join_type = nil)
      if join.blank?
        self
      else
        join = case join
          when String
            @relation.join(join)
          when Hash, Array, Symbol
            if @klass.send(:array_of_strings?, join)
              @relation.join(join.join(' '))
            else
              @relation.join(@klass.send(:build_association_joins, join))
            end
          else
            @relation.join(join, join_type)
        end
        Relation.new(@klass, join)
      end
    end

    def conditions(conditions)
      if conditions.blank?
        self
      else
        conditions = @klass.send(:merge_conditions, conditions) if [String, Hash, Array].include?(conditions.class)
        Relation.new(@klass, @relation.where(conditions))
      end
    end

    def respond_to?(method)
      if @relation.respond_to?(method) || Array.instance_methods.include?(method.to_s)
        true
      else
        super
      end
    end

    private
      def method_missing(method, *args, &block)
        if @relation.respond_to?(method)
          @relation.send(method, *args, &block)
        elsif Array.instance_methods.include?(method.to_s)
          to_a.send(method, *args, &block)
        end
      end
  end
end