changed the defer os remove

OliverLok · Aug 9, 2024 · 8886705 · 8886705
1 parent b4bd181
commit 8886705
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/internal/api/server.go b/internal/api/server.go
@@ -22,14 +22,26 @@ var AUTHENTICATION_NONCE = "nonce"
 var socketPath = config.UNIX_SOCKET_PATH.Get()
 var pidPath = config.PID_PATH.Get()
 
+func cleanup() {
+	err := os.Remove(pidPath)
+	if err != nil {
+		log.Error("Failed to remove pid path.", err)
+	}
+
+	err = os.Remove(socketPath)
+	if err != nil {
+		log.Error("Failed to remove unix socket path.", err)
+	}
+}
+
 func StartServer() error {
+	defer cleanup()
 	var serverShutdown sync.WaitGroup
 
 	serverShutdown.Add(1)
 
 	go func() {
 		defer serverShutdown.Done()
-		defer os.Remove(pidPath)
 
 		err := startAPIServer()
 		if err != nil {
@@ -39,7 +51,6 @@ func StartServer() error {
 
 	go func() {
 		defer serverShutdown.Done()
-		defer os.Remove(socketPath)
 
 		err := startUnixServer()
 		if err != nil {