Refactor - use out var to set where we want output

main.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"flag"
 	"fmt"
+	"io"
 	"net/http"
 	"net/url"
 	"os"
@@ -266,6 +267,7 @@ func run() error {
 		}
 	}
 
+	var out io.Writer
 	// If outputFile option not specified print on stdout
 	if outputFile != "" {
 		streamFile, err := os.Create(outputFile)
@@ -274,18 +276,12 @@ func run() error {
 		}
 
 		defer streamFile.Close()
-		// Save final json output into file
-		encoder := json.NewEncoder(streamFile)
-		if prettyPrint {
-			encoder.SetIndent("", "    ")
-		}
-		if err = encoder.Encode(&streamMetadata); err != nil {
-			return fmt.Errorf("Error while encoding: %v", err)
-		}
-		return nil
+		out = streamFile
+	} else {
+		out = os.Stdout
 	}
 
-	encoder := json.NewEncoder(os.Stdout)
+	encoder := json.NewEncoder(out)
 	if prettyPrint {
 		encoder.SetIndent("", "    ")
 	}