lib/arel/tree_manager.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

module Arel
  class TreeManager
    # FIXME: Remove this.
    include Arel::Relation

    VISITORS = {
      'postgresql' => Arel::Visitors::PostgreSQL,
      'mysql'      => Arel::Visitors::MySQL,
      'mysql2'     => Arel::Visitors::MySQL,
    }

    attr_accessor :visitor

    def initialize engine
      @engine  = engine
      @visitor = nil
    end

    def to_dot
      Visitors::Dot.new.accept @head
    end

    def visitor
      return @visitor if @visitor
      pool          = @engine.connection_pool
      adapter       = pool.spec.config[:adapter]
      @visitor = (VISITORS[adapter] || Visitors::ToSql).new(@engine)
    end

    def to_sql
      visitor.accept @head
    end

    def initialize_copy other
      super
      @head = @head.clone
    end
  end
end