handle file closure

abdulsattar · Jan 25, 2024 · f927423 · f927423
1 parent 779d4bd
commit f927423
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/daemon.go b/daemon.go
@@ -13,7 +13,12 @@ func main() {
 	if err != nil {
 		log.Fatalln("Error initializing KV store", err)
 	}
-	defer kv.Close()
+	defer func(kv *KV) {
+		err := kv.Close()
+		if err != nil {
+			log.Fatalln("Failed to close")
+		}
+	}(kv)
 	scanner := bufio.NewScanner(os.Stdin)
 	for scanner.Scan() {
 		line := scanner.Text()