gojay

commit 7dc0f34a209c70ac17c2bb27e5a94ba2f5f97b75
parent f324ba095c277e728ad680061a19cf9ff080bced
Author: francoispqt <francois@parquet.ninja>
Date:   Mon, 23 Jul 2018 21:46:41 +0800

change code gen to create writer after gen to avoid errors when writing file in inspected package

Diffstat:
M
gojay/main.go
 | 
15
++++++++++-----

1 file changed, 10 insertions(+), 5 deletions(-)
diff --git a/gojay/main.go b/gojay/main.go
@@ -82,19 +82,18 @@ func getOutput() (stringWriter, error) {
 	return os.Stdout, nil
 }
 
-func parseArgs() (p string, t []string, o stringWriter, err error) {
+func parseArgs() (p string, t []string, err error) {
 	flag.Parse()
 	p, err = getPath()
 	if err != nil {
-		return p, t, o, err
+		return p, t, err
 	}
 	t = getTypes()
-	o, err = getOutput()
-	return p, t, o, err
+	return p, t, err
 }
 
 func main() {
-	p, t, o, err := parseArgs()
+	p, t, err := parseArgs()
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -111,6 +110,12 @@ func main() {
 		log.Fatal(err)
 		return
 	}
+	var o stringWriter
+	o, err = getOutput()
+	if err != nil {
+		log.Fatal(err)
+		return
+	}
 	// write content to output
 	o.WriteString(g.b.String())
 }