Close output file

anjmao · May 22, 2019 · 01a8355 · 01a8355
1 parent f008635
commit 01a8355
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -4,7 +4,6 @@ import (
 	"flag"
 	"go/token"
 	"go/types"
-	"golang.org/x/tools/go/packages"
 	"log"
 	"os"
 	"path/filepath"
@@ -13,6 +12,8 @@ import (
 	"text/template"
 	"unicode"
 	"unicode/utf8"
+
+	"golang.org/x/tools/go/packages"
 )
 
 type arrFlags []string
@@ -202,6 +203,7 @@ message {{.Name}} {
 	if err != nil {
 		return err
 	}
+	defer f.Close()
 
 	return tmpl.Execute(f, msgs)
 }