aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/relation.rb
blob: f89baa1a7416d8aa8619b00852cf4cd8d63bf7c7 (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
module ActiveRecord
  class Relation
    delegate :delete, :to_sql, :to => :relation
    CLAUSES_METHODS = ["project", "group", "order", "take", "skip"].freeze
    attr_reader :relation, :klass

    def initialize(klass, table = nil)
      @klass = klass
      @relation = Arel::Table.new(table || @klass.table_name)
    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

    for clause in CLAUSES_METHODS
      class_eval %{
        def #{clause}(_#{clause})
          @relation = @relation.#{clause}(_#{clause}) if _#{clause}
          self
        end
      }
    end

    def join(joins)
      @relation = @relation.join(@klass.send(:construct_join, joins, nil)) if !joins.blank?
      self
    end

    def where(conditions)
      if !conditions.blank?
        conditions = @klass.send(:merge_conditions, conditions) if [String, Hash, Array].include?(conditions.class)
        @relation = @relation.where(conditions)
      end
      self
    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