Merge pull request #42 from CodinGame/add-max-start-attempt-count-option

[FEAT] Add maxStartAttemptCount option

Author: samuel-olivier

src/index.ts

@@ -5,7 +5,7 @@ import { CodinGameInfrastructure, Infrastructure } from './infrastructure'
 import { WillShutdownParams } from './customRequests'
 import { loadExtensionConfigurations } from './extensionConfiguration'
 import './hacks'
-import { createLanguageClientManager, LanguageClientManager, StatusChangeEvent } from './languageClient'
+import { createLanguageClientManager, LanguageClientManager, LanguageClientManagerOptions, StatusChangeEvent } from './languageClient'
 import { LanguageClientId, LanguageClientOptions, registerLanguageClient } from './languageClientOptions'
 import { StaticLanguageClientId } from './staticOptions'
 
@@ -27,5 +27,6 @@ export type {
   TextDocument,
   TextDocumentSaveReason,
   LanguageClientOptions,
+  LanguageClientManagerOptions,
   MessageTransports
 }

src/languageClient.ts

@@ -24,6 +24,13 @@ export interface StatusChangeEvent {
 
 const RETRY_CONNECTION_DELAY = 3000
 
+export interface LanguageClientManagerOptions {
+  /**
+   * The language client will stop trying to start after `maxStartAttemptCount` failed attempts
+   */
+   maxStartAttemptCount?: number
+}
+
 export class LanguageClientManager implements LanguageClient {
   languageClient?: MonacoLanguageClient
   private disposed: boolean = false
@@ -37,7 +44,8 @@ export class LanguageClientManager implements LanguageClient {
   constructor (
     private id: LanguageClientId,
     private clientOptions: LanguageClientOptions,
-    private infrastructure: Infrastructure
+    private infrastructure: Infrastructure,
+    private managerOptions: LanguageClientManagerOptions
   ) {
     this.useMutualizedProxy = this.infrastructure.useMutualizedProxy(this.id, this.clientOptions)
   }
@@ -124,7 +132,14 @@ export class LanguageClientManager implements LanguageClient {
       }))
     }
     let started = false
-    while (!this.isDisposed() && !started) {
+    let attempt = 0
+    while (
+      !this.isDisposed() &&
+      !started && (
+        this.managerOptions.maxStartAttemptCount == null ||
+        attempt < this.managerOptions.maxStartAttemptCount
+      )
+    ) {
       try {
         await this._start()
         started = true
@@ -134,6 +149,7 @@ export class LanguageClientManager implements LanguageClient {
         }))
         await delay(RETRY_CONNECTION_DELAY)
       }
+      ++attempt
     }
   }
 
@@ -302,7 +318,8 @@ export class LanguageClientManager implements LanguageClient {
 function createLanguageClientManager (
   id: LanguageClientId,
   infrastructure: Infrastructure,
-  clientOptions: LanguageClientOptions = getLanguageClientOptions(id)
+  clientOptions: LanguageClientOptions = getLanguageClientOptions(id),
+  managerOptions: LanguageClientManagerOptions = {}
 ): LanguageClientManager {
   // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
   if (clientOptions == null) {
@@ -320,7 +337,7 @@ function createLanguageClientManager (
 
   updateServices(infrastructure)
 
-  return new LanguageClientManager(id, clientOptions, infrastructure)
+  return new LanguageClientManager(id, clientOptions, infrastructure, managerOptions)
 }
 
 export {