27 lines
696 B
Ruby
27 lines
696 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
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
|
||
|
|
send dispatch[object.class], object, collector
|
||
|
|
rescue NoMethodError => e
|
||
|
|
raise e if respond_to?(dispatch[object.class], true)
|
||
|
|
superklass = object.class.ancestors.find { |klass|
|
||
|
|
respond_to?(dispatch[klass], true)
|
||
|
|
}
|
||
|
|
raise(TypeError, "Cannot visit #{object.class}") unless superklass
|
||
|
|
dispatch[object.class] = dispatch[superklass]
|
||
|
|
retry
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|