fix

Rustin170506 · Sep 7, 2022 · 85a5746 · 85a5746
1 parent 67400c9
commit 85a5746
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pyroscope_ffi/ruby/lib/pyroscope.rb b/pyroscope_ffi/ruby/lib/pyroscope.rb
@@ -22,11 +22,13 @@ module Utils
     attach_function :thread_id, [], :uint64
   end
 
-  class Engine < Rails::Engine
-    config.after_initialize do
-      next unless ::Pyroscope.current_config && ::Pyroscope.current_config.autoinstrument_rails
+  if defined?(::Rails::Engine)
+    class Engine < ::Rails::Engine
+      config.after_initialize do
+        next unless ::Pyroscope.current_config && ::Pyroscope.current_config.autoinstrument_rails
 
-      ::Pyroscope.initialize_rails_hooks
+        ::Pyroscope.initialize_rails_hooks
+      end
     end
   end