lib/heimdall_apm/agent.rb in heimdall_apm-0.1.0 vs lib/heimdall_apm/agent.rb in heimdall_apm-0.1.1

@@ -43,25 +43,27 @@
     end
 
     def start(options = {})
       return unless context.config.value('enabled')
 
-      @background_thread = Thread.new { background_run }
-
       # TODO: use instruments manager
       require 'heimdall_apm/instruments/active_record'      if defined?(ActiveRecord)
       require 'heimdall_apm/instruments/action_controller'  if defined?(ActionController)
       require 'heimdall_apm/instruments/elasticsearch'      if defined?(Elasticsearch)
 
       if (options[:app])
         require 'heimdall_apm/instruments/middleware'
         # TODO: make the position configurable
         options[:app].config.middleware.insert_after Rack::Cors, HeimdallApm::Instruments::Middleware
       end
+
+      context.started!
+      @background_thread = Thread.new { background_run }
     end
 
     def stop
       @stopped = true
+      context.stopped!
       if @background_thread.alive?
         @background_thread.wakeup
         @background_thread.join
       end
     end