Restart relay when closes unexpectedly

GuacLive · Jun 8, 2019 · a33ef89 · a33ef89
1 parent 2768a22
commit a33ef89
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/node_relay_session.js b/node_relay_session.js
@@ -48,12 +48,29 @@ class NodeRelaySession extends EventEmitter {
 
     this.ffmpeg_exec.on('close', (code) => {
       Logger.log('[Relay end] id=', this.id);
-      this.emit('end', this.id);
+      this.ffmpeg_exec = null;
+      if (!this._ended) {
+        this._runtimeout = setTimeout(() => {
+          this._runtimeout = null;
+          this.run();
+        }, 1000);
+      } else {
+        this.emit('end', this.id);
+      }
     });
   }
 
   end() {
-    this.ffmpeg_exec.kill();
+    this._ended = true;
+    if (this._runtimeout != null) {
+      clearTimeout(this._runtimeout);
+      this._runtimeout = null;
+    }
+    if (this.ffmpeg_exec) {
+      this.ffmpeg_exec.kill();
+    } else {
+      this.emit('end', this.id);
+    }
   }
 }