掉线自动重连

src/bootstrap.ts

@@ -1,5 +1,4 @@
 import * as colors from 'colors/safe'
-import * as got from 'got'
 import {Cluster} from './cluster'
 import ms = require('ms')
 
@@ -21,9 +20,6 @@ export async function bootstrap(version: string): Promise<void> {
   try {
     await cluster.enable()
   } catch (e) {
-    if (e instanceof got.HTTPError) {
-      console.error(e.response.body)
-    }
     console.error(e)
     process.exit(1)
   }

src/cluster.ts

@@ -24,6 +24,7 @@ interface ICounters {
 
 export class Cluster {
   public readonly counters: ICounters = {hits: 0, bytes: 0}
+  public isEnabled = false
 
   private readonly prefixUrl = process.env.CLUSTER_BMCLAPI || 'https://openbmclapi.bangbang93.com'
   private readonly cacheDir = join(__dirname, '..', 'cache')
@@ -62,10 +63,6 @@ export class Cluster {
         clusterId: this.clusterId, clusterSecret: this.clusterSecret,
       },
     })
-    this.io.on('connect', () => console.log('connected'))
-    this.io.on('message', (msg) => console.log(msg))
-    this.io.on('disconnect', () => console.log('disconnect'))
-    this.io.on('error', (err) => console.error(err))
   }
 
   public async getFileList(): Promise<IFileList> {
@@ -139,16 +136,21 @@ export class Cluster {
   }
 
   public async enable(): Promise<void> {
-    return new Promise((resolve, reject) => {
-      this.io.emit('enable', {
-        host: this.host,
-        port: this.publicPort,
-        version: this.version,
-      }, (ack) => {
-        if (ack !== true) return reject(ack)
-        resolve()
-      })
+    if (this.isEnabled) return
+    if (this.io.connected) {
+      await this._enable()
+    }
+    this.io.on('connect', async () => {
+      console.log('connected')
+      await this._enable()
+      this.isEnabled = true
+    })
+    this.io.on('message', (msg) => console.log(msg))
+    this.io.on('disconnect', () => {
+      console.log('disconnect')
+      this.isEnabled = false
     })
+    this.io.on('error', (err) => console.error(err))
   }
 
   public async disable(): Promise<void> {
@@ -184,4 +186,17 @@ export class Cluster {
       })
     })
   }
+
+  private async _enable() {
+    return new Promise((resolve, reject) => {
+      this.io.emit('enable', {
+        host: this.host,
+        port: this.publicPort,
+        version: this.version,
+      }, (ack) => {
+        if (ack !== true) return reject(ack)
+        resolve()
+      })
+    })
+  }
 }