golang并发ping主机的方法

Go  /  管理员 发布于 7年前   363

利用了golang对高并发的良好支持，同目录下将ip每行一个写入pinglist.txt文件即可

其实这个功能用linux一条命令就能搞定：

cat pinglist.txt | xargs -P 10 -I {} ping -fc 100 {}

package mainimport (  "bufio"  "bytes"  "fmt"  "io"  "io/ioutil"  "log"  "os"  "os/exec"  "strings"  "sync"  "runtime")func RunCMD(command string) string {  in := bytes.NewBuffer(nil)  cmd := exec.Command("sh")  cmd.Stdin = in  in.WriteString(command + "\n")  stdout, err := cmd.StdoutPipe()  if err != nil {    log.Fatal(err)  }  defer stdout.Close()  if err := cmd.Start(); err != nil {    log.Fatal(err)  }  opBytes, err := ioutil.ReadAll(stdout)  if err != nil {    log.Fatal(err)  }  return string(opBytes)}func ping(ip string) string {  cmd := "ping -fc 100 " + ip  return RunCMD(cmd)}var ips = []string{}func main() {  runtime.GOMAXPROCS(runtime.NumCPU())  fileName := "pinglist.txt"  f, err := os.Open(fileName)  if err != nil {    fmt.Println(err)  }  buf := bufio.NewReader(f)  for {    line, err := buf.ReadString('\n')    line = strings.TrimSpace(line)    if err != nil {      if err == io.EOF {        //fmt.Println("File read over!")        break      }      fmt.Println(err)    }    ips = append(ips, line)  }  wg := &sync.WaitGroup{}  ch := make(chan string, len(ips))  //fmt.Println(ips)  for i:= 0; i < len(ips); i++ {    wg.Add(1)    go func(i int) {      ch <- ping(ips[i])      wg.Done()    }(i)  }  wg.Wait()  //fmt.Println(len(ch))  for i := 0; i < len(ips); i++ {    fmt.Println(<-ch)  }}

以上这篇golang并发ping主机的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。