improved report objects

Author: czechbol

report/csv.go

@@ -7,7 +7,7 @@ import (
 )
 
 // JSONReport represents the output data fields in a JSON file
-type JSONReport struct {
+type Report struct {
 	Timestamp     time.Time `json:"timestamp"`
 	Server        Server    `json:"server"`
 	Client        Client    `json:"client"`
@@ -20,6 +20,18 @@ type JSONReport struct {
 	Share         string    `json:"share"`
 }
 
+type FlatReport struct {
+	Timestamp time.Time `csv:"Timestamp"`
+	Name      string    `csv:"Server Name"`
+	Address   string    `csv:"Address"`
+	Ping      float64   `csv:"Ping"`
+	Jitter    float64   `csv:"Jitter"`
+	Download  float64   `csv:"Download"`
+	Upload    float64   `csv:"Upload"`
+	Share     string    `csv:"Share"`
+	IP        string    `csv:"IP"`
+}
+
 // Server represents the speed test server's information
 type Server struct {
 	Name string `json:"name"`
@@ -30,3 +42,19 @@ type Server struct {
 type Client struct {
 	defs.IPInfoResponse
 }
+
+func (r Report) GetFlatReport() FlatReport {
+	var rep FlatReport
+	rep.Timestamp = time.Now()
+
+	rep.Name = r.Server.Name
+	rep.Address = r.Server.URL
+	rep.Ping = r.Ping
+	rep.Jitter = r.Jitter
+	rep.Download = r.Download
+	rep.Upload = r.Upload
+	rep.Share = r.Share
+	rep.IP = r.Client.IP
+
+	return rep
+}

report/json.go report/testreport.goreport/json.go renamed to report/testreport.go

@@ -33,8 +33,7 @@ func doSpeedTest(c *cli.Context, servers []defs.Server, telemetryServer defs.Tel
 		log.Infof("Testing against %d servers", serverCount)
 	}
 
-	var reps_json []report.JSONReport
-	var reps_csv []report.CSVReport
+	var reps []report.Report
 
 	// fetch current user's IP info
 	for _, currentServer := range servers {
@@ -141,43 +140,24 @@ func doSpeedTest(c *cli.Context, servers []defs.Server, telemetryServer defs.Tel
 				}
 			}
 
-			// check for --csv or --json. the program prioritize the --csv before the --json. this is the same behavior as speedtest-cli
-			if c.Bool(defs.OptionCSV) {
-				// print csv if --csv is given
-				var rep report.CSVReport
-				rep.Timestamp = time.Now()
-
-				rep.Name = currentServer.Name
-				rep.Address = u.String()
-				rep.Ping = math.Round(p*100) / 100
-				rep.Jitter = math.Round(jitter*100) / 100
-				rep.Download = math.Round(downloadValue*100) / 100
-				rep.Upload = math.Round(uploadValue*100) / 100
-				rep.Share = shareLink
-				rep.IP = ispInfo.RawISPInfo.IP
-
-				reps_csv = append(reps_csv, rep)
-			} else if c.Bool(defs.OptionJSON) {
-				// print json if --json is given
-				var rep report.JSONReport
-				rep.Timestamp = time.Now()
-
-				rep.Ping = math.Round(p*100) / 100
-				rep.Jitter = math.Round(jitter*100) / 100
-				rep.Download = math.Round(downloadValue*100) / 100
-				rep.Upload = math.Round(uploadValue*100) / 100
-				rep.BytesReceived = bytesRead
-				rep.BytesSent = bytesWritten
-				rep.Share = shareLink
-
-				rep.Server.Name = currentServer.Name
-				rep.Server.URL = u.String()
-
-				rep.Client = report.Client{ispInfo.RawISPInfo}
-				rep.Client.Readme = ""
-
-				reps_json = append(reps_json, rep)
-			}
+			var rep report.Report
+			rep.Timestamp = time.Now()
+
+			rep.Ping = math.Round(p*100) / 100
+			rep.Jitter = math.Round(jitter*100) / 100
+			rep.Download = math.Round(downloadValue*100) / 100
+			rep.Upload = math.Round(uploadValue*100) / 100
+			rep.BytesReceived = bytesRead
+			rep.BytesSent = bytesWritten
+			rep.Share = shareLink
+
+			rep.Server.Name = currentServer.Name
+			rep.Server.URL = u.String()
+
+			rep.Client = report.Client{IPInfoResponse: ispInfo.RawISPInfo}
+			rep.Client.Readme = ""
+
+			reps = append(reps, rep)
 		} else {
 			log.Infof("Selected server %s (%s) is not responding at the moment, try again later", currentServer.Name, u.Hostname())
 		}
@@ -191,13 +171,17 @@ func doSpeedTest(c *cli.Context, servers []defs.Server, telemetryServer defs.Tel
 	// check for --csv or --json. the program prioritize the --csv before the --json. this is the same behavior as speedtest-cli
 	if c.Bool(defs.OptionCSV) {
 		var buf bytes.Buffer
+		var reps_csv []report.FlatReport
+		for _, rep := range reps {
+			reps_csv = append(reps_csv, rep.GetFlatReport())
+		}
 		if err := gocsv.MarshalWithoutHeaders(&reps_csv, &buf); err != nil {
 			log.Errorf("Error generating CSV report: %s", err)
 		} else {
 			os.Stdout.WriteString(buf.String())
 		}
 	} else if c.Bool(defs.OptionJSON) {
-		if b, err := json.Marshal(&reps_json); err != nil {
+		if b, err := json.Marshal(&reps); err != nil {
 			log.Errorf("Error generating JSON report: %s", err)
 		} else {
 			os.Stdout.Write(b[:])

speedtest/helper.go

@@ -79,7 +79,7 @@ func SpeedTest(c *cli.Context) error {
 
 	// if --csv-header is given, print the header and exit (same behavior speedtest-cli)
 	if c.Bool(defs.OptionCSVHeader) {
-		var rep []report.CSVReport
+		var rep []report.FlatReport
 		b, _ := gocsv.MarshalBytes(&rep)
 		os.Stdout.WriteString(string(b))
 		return nil