lib/newrelic-rake/instrument.rb in newrelic-rake-1.1.0 vs lib/newrelic-rake/instrument.rb in newrelic-rake-1.2.0

@@ -15,13 +15,18 @@
     ::Rake::Task.class_eval do
       include ::NewRelic::Agent::Instrumentation::ControllerInstrumentation
 
       alias_method :origin_execute, :execute
       def execute(args=nil)
-        NewRelic::Agent.manual_start
+        NewRelic::Agent.manual_start(:dispatcher => :rake)
         perform_action_with_newrelic_trace(:name => self.name, :category => "OtherTransaction/Rake") do
           origin_execute(args)
         end
       end
+
+      # Make sure NewRelic agent flush data to the server according to
+      # https://newrelic.com/docs/ruby/monitoring-ruby-background-processes-and-daemons
+      # even though Agent configuration is :send_data_on_exit => true
+      at_exit { NewRelic::Agent.shutdown }
     end
   end
 end