2014-04-08 20:01:53 -04:00
|
|
|
require 'arel/visitors/visitor'
|
|
|
|
|
|
|
|
module Arel
|
|
|
|
module Visitors
|
|
|
|
class Reduce < Arel::Visitors::Visitor
|
|
|
|
def accept object, collector
|
|
|
|
visit object, collector
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def visit object, collector
|
2014-09-22 14:01:03 -04:00
|
|
|
send dispatch[object.class.name], object, collector
|
2014-04-08 20:01:53 -04:00
|
|
|
rescue NoMethodError => e
|
2014-09-22 14:01:03 -04:00
|
|
|
raise e if respond_to?(dispatch[object.class.name], true)
|
2014-04-08 20:01:53 -04:00
|
|
|
superklass = object.class.ancestors.find { |klass|
|
2014-09-22 14:01:03 -04:00
|
|
|
respond_to?(dispatch[klass.name], true)
|
2014-04-08 20:01:53 -04:00
|
|
|
}
|
|
|
|
raise(TypeError, "Cannot visit #{object.class}") unless superklass
|
2014-09-22 14:01:03 -04:00
|
|
|
dispatch[object.class.name] = dispatch[superklass.name]
|
2014-04-08 20:01:53 -04:00
|
|
|
retry
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|