47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
|
package cmd
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/go-ping/ping"
|
||
|
"github.com/spf13/cobra"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
host string
|
||
|
count int
|
||
|
)
|
||
|
|
||
|
var pingCmd = &cobra.Command{
|
||
|
Use: "ping",
|
||
|
Short: "Ping test.",
|
||
|
Long: "Uses ping to test network latency and aliveness.",
|
||
|
Run: func(cmd *cobra.Command, args []string) {
|
||
|
pinger, err := ping.NewPinger(host)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
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",
|
||
|
stats.PacketsSent, stats.PacketsRecv, stats.PacketLoss)
|
||
|
fmt.Printf("round-trip min/avg/max/stddev = %v/%v/%v/%v\n",
|
||
|
stats.MinRtt, stats.AvgRtt, stats.MaxRtt, stats.StdDevRtt)
|
||
|
}
|
||
|
fmt.Printf("PING %s (%s):\n", pinger.Addr(), pinger.IPAddr())
|
||
|
err = pinger.Run() // Blocks until finished.
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
},
|
||
|
}
|
||
|
|
||
|
func init() {
|
||
|
pingCmd.Flags().StringVarP(&host, "host", "a", "8.8.8.8", "Host to attempt pinging.")
|
||
|
pingCmd.Flags().IntVarP(&count, "count", "c", 50, "Number of pings to send.")
|
||
|
|
||
|
// add ping command to the main program
|
||
|
rootCmd.AddCommand(pingCmd)
|
||
|
}
|