From 433b81a17fc006e2e6862da5da7af520e605b2f1 Mon Sep 17 00:00:00 2001 From: Bobby Brennan Date: Thu, 18 Apr 2019 16:32:25 +0000 Subject: [PATCH] better error handling, i/o --- main.go | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/main.go b/main.go index 03bfcd24..fe72bc76 100644 --- a/main.go +++ b/main.go @@ -173,11 +173,11 @@ func runAudit(c conf.Configuration, outputFile string, outputURL string) { } if outputURL == "" && outputFile == "" { - y, err := yaml.Marshal(auditData) + yamlBytes, err := yaml.Marshal(auditData) if err != nil { panic(err) } - fmt.Println(string(y)) + os.Stdout.Write(yamlBytes) } else { jsonData, err := json.Marshal(auditData) if err != nil { @@ -202,8 +202,12 @@ func runAudit(c conf.Configuration, outputFile string, outputURL string) { } defer resp.Body.Close() - body, _ := ioutil.ReadAll(resp.Body) - glog.Println(string(body)) + body, err := ioutil.ReadAll(resp.Body) + if err != nil { + fmt.Println("Error reading audit output URL response") + } else { + glog.Println(string(body)) + } } } }