To ensure the program is restarted if the agent is already running

ystv · Jan 15, 2024 · a8712f4 · a8712f4
1 parent 150f909
commit a8712f4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/forwarder/main.go b/forwarder/main.go
@@ -163,6 +163,9 @@ func (v *Views) run(config Config, interrupt chan os.Signal) {
 				close(errorChannel)
 			}
 		}()
+		defer func(c *websocket.Conn) {
+			_ = c.Close()
+		}(c)
 		response := specialTransporter.InitiationTransporter{
 			Server:  server.Forwarder,
 			Version: Version,

diff --git a/recorder/main.go b/recorder/main.go
@@ -163,6 +163,9 @@ func (v *Views) run(config Config, interrupt chan os.Signal) {
 				close(errorChannel)
 			}
 		}()
+		defer func(c *websocket.Conn) {
+			_ = c.Close()
+		}(c)
 		response := specialTransporter.InitiationTransporter{
 			Server:  server.Recorder,
 			Version: Version,