欢迎来到代码驿站!

Golang

当前位置:首页 > 脚本语言 > Golang

golang并发ping主机的方法

时间:2020-10-21 20:54:22|栏目:Golang|点击:

利用了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 bufio包中Write方法的深入讲解

栏    目:Golang

下一篇:golang time包下定时器的实现方法

本文标题:golang并发ping主机的方法

本文地址:http://www.codeinn.net/misctech/14483.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有