Fix shutdown condition for counter server

kimutansk · Apr 4, 2018 · 59f8441 · 59f8441
1 parent 44f3413
commit 59f8441
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/fluent/supervisor.rb b/lib/fluent/supervisor.rb
@@ -42,6 +42,8 @@ module Fluent
   module ServerModule
     def before_run
       @start_time = Time.now
+      @rpc_server = nil
+      @counter = nil
 
       if config[:rpc_endpoint]
         @rpc_endpoint = config[:rpc_endpoint]
@@ -66,7 +68,7 @@ def before_run
 
     def after_run
       stop_rpc_server if @rpc_endpoint
-      stop_counter_server if @counter_endpoint
+      stop_counter_server if @counter
       Fluent::Supervisor.cleanup_resources
     end