feat: add cleanup for for clientAndLocatrs dict

vertexcover-io · Dec 26, 2024 · d69c6bd · d69c6bd
1 parent 3b415ed
commit d69c6bd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/server/main.go b/server/main.go
@@ -158,6 +158,9 @@ func acceptConnection(fd net.Conn) {
 			}
 			continue
 		}
+
+		defer delete(clientAndLocatrs, clientMessage.ClientId)
+
 		switch clientMessage.Type {
 		case "initial_handshake":
 			err := handleInitialHandshake(clientMessage)
@@ -211,6 +214,7 @@ func acceptConnection(fd net.Conn) {
 			}
 		}
 	}
+
 }
 
 func main() {