From f68b7c4e7e25415c41308a5c37f1096a524b6d09 Mon Sep 17 00:00:00 2001 From: Aaron Patterson Date: Mon, 29 Nov 2010 15:31:28 -0800 Subject: base class works with visitor --- lib/arel/nodes/node.rb | 2 ++ lib/arel/visitors/depth_first.rb | 1 + 2 files changed, 3 insertions(+) (limited to 'lib/arel') diff --git a/lib/arel/nodes/node.rb b/lib/arel/nodes/node.rb index 90c63f0be9..634e580a8f 100644 --- a/lib/arel/nodes/node.rb +++ b/lib/arel/nodes/node.rb @@ -37,6 +37,8 @@ module Arel # Iterate through AST, nodes will be yielded depth-first def each &block + return enum_for(:each) unless block_given? + Visitors::DepthFirst.new(block).accept self end end diff --git a/lib/arel/visitors/depth_first.rb b/lib/arel/visitors/depth_first.rb index 46171dc5bd..00f18727f0 100644 --- a/lib/arel/visitors/depth_first.rb +++ b/lib/arel/visitors/depth_first.rb @@ -93,6 +93,7 @@ module Arel alias :visit_ActiveSupport_Multibyte_Chars :terminal alias :visit_ActiveSupport_StringInquirer :terminal alias :visit_Arel_Nodes_Lock :terminal + alias :visit_Arel_Nodes_Node :terminal alias :visit_Arel_Nodes_SqlLiteral :terminal alias :visit_Arel_SqlLiteral :terminal alias :visit_BigDecimal :terminal -- cgit v1.2.3