lib/flamegraph.rb in flamegraph-0.0.7 vs lib/flamegraph.rb in flamegraph-0.0.8

@@ -1,11 +1,10 @@
 require "json"
 
 if RUBY_VERSION >= "2.1.0".freeze
   begin
     require "stackprof"
-    require "flamegraph/stackprof_sampler"
   rescue
     STDERR.puts "Please require the stackprof gem falling back to fast_stack"
     require "fast_stack"
   end
 else
@@ -23,9 +22,10 @@
   def self.generate(filename=nil, opts = {})
     fidelity = opts[:fidelity]  || 0.5
 
     backtraces =
       if defined? StackProf
+        require "flamegraph/stackprof_sampler" unless defined? StackProfSampler
         StackProfSampler.collect(fidelity) do
           yield
         end
       else
         FastStack.profile(fidelity) do # , opts[:mode] || :ruby) do