lib/roodi/core/runner.rb in roodi-1.4.0 vs lib/roodi/core/runner.rb in roodi-2.0.0

@@ -1,10 +1,9 @@
 require 'pp'
 require 'yaml'
 
 require 'roodi/core/checking_visitor'
-require 'roodi/core/iterator_visitor'
 require 'roodi/core/parser'
 require 'roodi/core/visitable_sexp'
 
 module Roodi
   module Core
@@ -14,16 +13,17 @@
       attr_writer :config
       
       def initialize(*checks)
         @config = DEFAULT_CONFIG
         @checks = checks unless checks.empty?
+        @checks ||= load_checks
+        @checker ||= CheckingVisitor.new(@checks)
         @parser = Parser.new
       end
       
       def check(filename, content)
-        @checks ||= load_checks
         node = parse(filename, content)
-        node.accept(IteratorVisitor.new(CheckingVisitor.new(@checks))) if node
+        node.accept(@checker) if node
       end
 
       def check_content(content)
         check("dummy-file.rb", content)
       end