From 6cd43f5044e169c5044cd337f7f101c9ee18712d Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Tue, 19 Oct 2010 11:42:02 -0700
Subject: adding a base class visitor

---
 lib/arel/visitors/to_sql.rb  | 12 ++----------
 lib/arel/visitors/visitor.rb | 19 +++++++++++++++++++
 2 files changed, 21 insertions(+), 10 deletions(-)
 create mode 100644 lib/arel/visitors/visitor.rb

(limited to 'lib/arel/visitors')

diff --git a/lib/arel/visitors/to_sql.rb b/lib/arel/visitors/to_sql.rb
index 17779b4037..e68890004a 100644
--- a/lib/arel/visitors/to_sql.rb
+++ b/lib/arel/visitors/to_sql.rb
@@ -3,7 +3,7 @@ require 'date'
 
 module Arel
   module Visitors
-    class ToSql
+    class ToSql < Arel::Visitors::Visitor
       def initialize engine
         @engine         = engine
         @connection     = nil
@@ -16,7 +16,7 @@ module Arel
         @last_column = nil
         @engine.connection_pool.with_connection do |conn|
           @connection = conn
-          visit object
+          super
         end
       end
 
@@ -277,14 +277,6 @@ module Arel
       alias :visit_Time :visit_String
       alias :visit_TrueClass :visit_String
 
-      DISPATCH = Hash.new do |hash, klass|
-        hash[klass] = "visit_#{klass.name.gsub('::', '_')}"
-      end
-
-      def visit object
-        send DISPATCH[object.class], object
-      end
-
       def quote value, column = nil
         @connection.quote value, column
       end
diff --git a/lib/arel/visitors/visitor.rb b/lib/arel/visitors/visitor.rb
new file mode 100644
index 0000000000..510d50399b
--- /dev/null
+++ b/lib/arel/visitors/visitor.rb
@@ -0,0 +1,19 @@
+module Arel
+  module Visitors
+    class Visitor
+      def accept object
+        visit object
+      end
+
+      private
+
+      DISPATCH = Hash.new do |hash, klass|
+        hash[klass] = "visit_#{klass.name.gsub('::', '_')}"
+      end
+
+      def visit object
+        send DISPATCH[object.class], object
+      end
+    end
+  end
+end
-- 
cgit v1.2.3