cmd/ping.go

@@ -5,6 +5,8 @@ import (
 
   "github.com/go-ping/ping"
   "github.com/spf13/cobra"
+
+  "git.metaunix.net/bitgoblin/hardware-tests/lib"
 )
 
 var (
@@ -23,11 +25,17 @@ var pingCmd = &cobra.Command{
     }
     pinger.Count = count
     pinger.OnFinish = func(stats *ping.Statistics) {
-    	fmt.Printf("--- %s ping statistics ---\n", stats.Addr)
-    	fmt.Printf("%d packets transmitted, %d packets received, %v%% packet loss\n",
+      result := fmt.Sprintf("--- %s ping statistics ---\n", stats.Addr)
+    	result += fmt.Sprintf("%d packets transmitted, %d packets received, %v%% packet loss\n",
     		stats.PacketsSent, stats.PacketsRecv, stats.PacketLoss)
-    	fmt.Printf("round-trip min/avg/max/stddev = %v/%v/%v/%v\n",
+    	result += fmt.Sprintf("round-trip min/avg/max/stddev = %v/%v/%v/%v\n",
     		stats.MinRtt, stats.AvgRtt, stats.MaxRtt, stats.StdDevRtt)
+
+      if Log != nil {
+        lib.WriteToFile(Log, result)
+      } else {
+        fmt.Printf("%s", result)
+      }
     }
     fmt.Printf("PING %s (%s):\n", pinger.Addr(), pinger.IPAddr())
     err = pinger.Run() // Blocks until finished.

cmd/root.go

@@ -7,6 +7,10 @@ import (
   "github.com/spf13/cobra"
 )
 
+var (
+  Log string
+)
+
 var rootCmd = &cobra.Command{
   Use:   "bgbench",
   Short: "Bit Goblin hardware benching tool.",
@@ -20,3 +24,7 @@ func Execute() {
     os.Exit(1)
   }
 }
+
+func init() {
+  rootCmd.PersistentFlags().StringVarP(&Log, "log", "l", "false", "File location to write results to.")
+}

lib/io.go

@@ -0,0 +1,16 @@
+package lib
+
+import (
+  "fmt"
+  "os"
+)
+
+func WriteToFile(filepath string, content string) {
+  file, err := os.Create(filepath)
+	if err != nil {
+		fmt.Println(err)
+	} else {
+		file.WriteString(content)
+	}
+	file.Close()
+}