Add timeout for sanity check

beeper · Dec 20, 2023 · 78d70c9 · 78d70c9
1 parent ac13182
commit 78d70c9
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -73,10 +73,19 @@ func main() {
 		panic(err)
 	}
 	log.Println("Running sanity check...")
+	safetyExitCancel := make(chan struct{})
+	go func() {
+		select {
+		case <-time.After(5 * time.Second):
+			log.Fatalln("Sanity check timed out")
+		case <-safetyExitCancel:
+		}
+	}()
 	err = InitSanityCheck()
 	if err != nil {
 		panic(err)
 	}
+	close(safetyExitCancel)
 	if *checkCompatibility {
 		log.Println("Compatibility check successful")
 		if *jsonOutput {