Misc

microsoft · May 18, 2023 · 57fc09a · 57fc09a
1 parent 8efe654
commit 57fc09a
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/kernels/jupyter/connection/jupyterConnection.ts b/src/kernels/jupyter/connection/jupyterConnection.ts
@@ -72,11 +72,8 @@ export class JupyterConnection implements IExtensionSyncActivationService {
         return savedList.find((item) => item.serverId === serverId)?.uri;
     }
     private async createConnectionInfoFromUri(uri: string) {
-        // Prepare our map of server URIs
-        await this.updateServerUri(uri);
-
+        const server = await this.updateServerUri(uri);
         const idAndHandle = extractJupyterServerHandleAndId(uri);
-        const server = this.uriToJupyterServerUri.get(uri);
         return createRemoteConnectionInfo(uri, server, idAndHandle?.id);
     }
 
@@ -96,7 +93,7 @@ export class JupyterConnection implements IExtensionSyncActivationService {
         }
     }
 
-    private async updateServerUri(uri: string): Promise<void> {
+    private async updateServerUri(uri: string) {
         const idAndHandle = extractJupyterServerHandleAndId(uri);
         if (idAndHandle) {
             try {
@@ -113,6 +110,7 @@ export class JupyterConnection implements IExtensionSyncActivationService {
                         this.pendingTimeouts.push(setTimeout(() => this.updateServerUri(uri).catch(noop), timeout));
                     }
                 }
+                return serverUri;
             } catch (ex) {
                 if (ex instanceof BaseError) {
                     throw ex;