@codingame/monaco-vscode-extensions-service-override 8.0.4 → 9.0.2

package/vscode/src/vs/workbench/services/extensions/common/remoteExtensionHost.js

@@ -42,6 +42,7 @@ let RemoteExtensionHost = class RemoteExtensionHost extends Disposable {
         this.extensions = null;
         this._onExit = this._register(( new Emitter()));
         this.onExit = this._onExit.event;
+        this._hasDisconnected = false;
         this.remoteAuthority = this._initDataProvider.remoteAuthority;
         this._protocol = null;
         this._hasLostConnection = false;
@@ -195,15 +196,20 @@ let RemoteExtensionHost = class RemoteExtensionHost extends Disposable {
     enableInspectPort() {
         return Promise.resolve(false);
     }
+    async disconnect() {
+        if (this._protocol && !this._hasDisconnected) {
+            this._protocol.send(createMessageOfType(MessageType.Terminate));
+            this._protocol.sendDisconnect();
+            this._hasDisconnected = true;
+            await this._protocol.drain();
+        }
+    }
     dispose() {
         super.dispose();
         this._terminating = true;
+        this.disconnect();
         if (this._protocol) {
-            const socket = this._protocol.getSocket();
-            this._protocol.send(createMessageOfType(MessageType.Terminate));
-            this._protocol.sendDisconnect();
-            this._protocol.dispose();
-            socket.end();
+            this._protocol.getSocket().end();
             this._protocol = null;
         }
     }