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

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

    VISITORS = {
      'postgresql' => Arel::Visitors::PostgreSQL
    }

    def initialize engine
      @engine        = engine
      @pool          = engine.connection_pool
      @adapter       = @pool.spec.config[:adapter]
      @visitor_klass = VISITORS[@adapter] || Visitors::ToSql
    end

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

    def to_sql
      viz = @visitor_klass.new @engine
      viz.accept @head
    end

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