From fcd11dcb99d69d40ff60b56bcd7330264128ecfd Mon Sep 17 00:00:00 2001 From: Jiri Pospisil Date: Sat, 14 Jun 2014 23:33:05 +0200 Subject: Modify Visitors::Dot's API to use collectors --- lib/arel/tree_manager.rb | 4 +++- lib/arel/visitors/dot.rb | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) (limited to 'lib/arel') diff --git a/lib/arel/tree_manager.rb b/lib/arel/tree_manager.rb index 87887800d0..8bff97af78 100644 --- a/lib/arel/tree_manager.rb +++ b/lib/arel/tree_manager.rb @@ -15,7 +15,9 @@ module Arel end def to_dot - Visitors::Dot.new.accept @ast + collector = Arel::Collectors::PlainString.new + collector = Visitors::Dot.new.accept @ast, collector + collector.value end def visitor diff --git a/lib/arel/visitors/dot.rb b/lib/arel/visitors/dot.rb index ba35223ac9..d96cf7a6a0 100644 --- a/lib/arel/visitors/dot.rb +++ b/lib/arel/visitors/dot.rb @@ -22,9 +22,9 @@ module Arel @seen = {} end - def accept object + def accept object, collector visit object - to_dot + collector << to_dot end private -- cgit v1.2.3