当前位置: 移动技术网 > IT编程>脚本编程>Go语言 > golang并发ping主机的方法

golang并发ping主机的方法

2019年09月09日  | 移动技术网IT编程  | 我要评论
利用了golang对高并发的良好支持,同目录下将ip每行一个写入pinglist.txt文件即可 其实这个功能用linux一条命令就能搞定: cat pinglist

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

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

cat pinglist.txt | xargs -p 10 -i {} ping -fc 100 {}

package main

import (
  "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主机的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持移动技术网。

如您对本文有疑问或者有任何想说的,请 点击进行留言回复,万千网友为您解惑!

相关文章:

验证码:
移动技术网