Send a disconnection message via the management connection before killing the local extension host (microsoft#157557)

Send a disconnection message via the management connection before killing the local extension host (microsoft#157269)

Fixes microsoft#156690: Send a disconnection message via the management connection before killing the local extension host

Author: alexdima

src/vs/workbench/services/extensions/common/abstractExtensionService.ts

@@ -258,6 +258,12 @@ export abstract class AbstractExtensionService extends Disposable implements IEx
 		}));
 
 		this._register(this._lifecycleService.onDidShutdown(() => {
+			// We need to disconnect the management connection before killing the local extension host.
+			// Otherwise, the local extension host might terminate the underlying tunnel before the
+			// management connection has a chance to send its disconnection message.
+			const connection = this._remoteAgentService.getConnection();
+			connection?.dispose();
+
 			this.stopExtensionHosts();
 		}));
 	}