panic if failing to close generated file

josephcopenhaver · Nov 29, 2023 · 57559ea · 57559ea
1 parent e321ff8
commit 57559ea
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/loadtester/internal/cmd/generate/main.go b/loadtester/internal/cmd/generate/main.go
@@ -74,7 +74,15 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
-	defer f.Close()
+	defer func() {
+		if err := f.Close(); err != nil {
+			v := any(err)
+			if r := recover(); r != nil {
+				v = r
+			}
+			panic(v)
+		}
+	}()
 
 	var buf bytes.Buffer