Add support to Ruby 2.1

efficiency20 · Sep 23, 2014 · 9babdf3 · 9babdf3
1 parent 78dac01
commit 9babdf3
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/lib/delayed/psych_ext.rb b/lib/delayed/psych_ext.rb
@@ -45,9 +45,16 @@ def self.load yaml, filename = nil, visitor = nil
 
   module Nodes
     class Node
-      def to_ruby(visitor)
-        visitor ||= Visitors::ToRuby
-        visitor.new.accept(self)
+      if Gem::Version.new(VERSION) >= Gem::Version.new('2.0.2')
+        def to_ruby(visitor)
+          visitor ||= Visitors::ToRuby
+          visitor.create.accept(self)
+        end
+      else
+        def to_ruby(visitor)
+          visitor ||= Visitors::ToRuby
+          visitor.new.accept(self)
+        end
       end
     end
   end