From dc03dd9cad39488052cf0a27ffe4a4230e4db4ac Mon Sep 17 00:00:00 2001 From: Chris Roberts Date: Tue, 11 Jan 2022 11:37:52 -0800 Subject: [PATCH] Wrap remote manager setup/teardown within usage tracker --- plugins/commands/serve/util/service_info.rb | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/plugins/commands/serve/util/service_info.rb b/plugins/commands/serve/util/service_info.rb index 1823ec0c8..8d63a48d7 100644 --- a/plugins/commands/serve/util/service_info.rb +++ b/plugins/commands/serve/util/service_info.rb @@ -14,11 +14,13 @@ module VagrantPlugins broker: broker ) if context.metadata["plugin_manager"] && info.broker - info.plugin_manager = Client::PluginManager.load( - context.metadata["plugin_manager"], - broker: info.broker - ) - Vagrant.plugin("2").enable_remote_manager + Service::ServiceInfo.manager_tracker.activate do + info.plugin_manager = Client::PluginManager.load( + context.metadata["plugin_manager"], + broker: info.broker + ) + Vagrant.plugin("2").enable_remote_manager + end end Thread.current.thread_variable_set(:service_info, info) return if !block_given? @@ -27,7 +29,9 @@ module VagrantPlugins rescue => e raise "#{e.class}: #{e}\n#{e.backtrace.join("\n")}" ensure - Vagrant.plugin("2").disable_remote_manager + Service::ServiceInfo.manager_tracker.deactivate do + Vagrant.plugin("2").disable_remote_manager + end Thread.current.thread_variable_set(:service_info, nil) end end