From 4d99fce3bb9eb8a9e0026e3739b34a8cb878d7b6 Mon Sep 17 00:00:00 2001
From: speatzle <sam@soontm.de>
Date: Sat, 31 May 2025 22:25:58 +0000
Subject: [PATCH] Cleanup conn if exists

---
 worker/worker.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/worker/worker.go b/worker/worker.go
index ab4621f..98d6de9 100644
--- a/worker/worker.go
+++ b/worker/worker.go
@@ -89,7 +89,13 @@ func connectToServer(ctx context.Context, uuid uuid.UUID) {
 	c, res, err := websocket.Dial(ctx, conf.Worker.Address+"/worker", &websocket.DialOptions{
 		HTTPHeader: headers,
 	})
-	defer c.CloseNow()
+
+	// Cleanup connection if exists to prevent FD leak
+	defer func(){
+		if c != nil {
+			c.CloseNow()
+		}
+	}
 	
 	if err != nil {
 		if res != nil {