美文网首页
Golang 获取网卡时时带宽,可用于测速

Golang 获取网卡时时带宽,可用于测速

作者: 彩色代码 | 来源:发表于2020-07-28 20:02 被阅读0次

每秒计算一次网卡出口带宽,并保存到指定文件,用于其他程序,如:php、shell读取带宽

那么如果是shell ,为啥不写在shell里面呢?答案是可以的,只是场景不一样,这里用go实现读取一次,然后其他程序可以多任务执行,直接使用结果。

package main

import (
    "bytes"
    "fmt"
    "io/ioutil"
    "os"
    "os/exec"
    "strings"
    "time"
)

func main()  {

    for {
        var stdOut, stdErr bytes.Buffer
        cmd := exec.Command("sh", "-c","dstat -n 1 1 | awk 'END {print}' | awk '{print $2}'")
        cmd.Stdout = &stdOut
        cmd.Stderr = &stdErr
        err := cmd.Run();
        if err!=nil {
            fmt.Println(cmd.Stderr,err.Error())
            os.Exit(1)
        }
        fmt.Print(cmd.Stdout)
        path := "/workspace/tmp/network-speed.data"
        content := stdOut.String()
        content = strings.Replace(content, "\n", "", -1)
        if content=="0" {
            time.Sleep(1*time.Second)
            continue
        }
        if strings.Contains(content, "B") {
            content = "1"
        } else if strings.Contains(content, "k"){
            content = "1"
        } else if strings.Contains(content, "M"){
            content = strings.Replace(content,"M","",-1)
        }
        err = ioutil.WriteFile(path, []byte(content), 0666)

        if err != nil {
            panic(err)
            time.Sleep(1*time.Second)
        }

    }

}

相关文章

网友评论

      本文标题:Golang 获取网卡时时带宽,可用于测速

      本文链接:https://www.haomeiwen.com/subject/nweqrktx.html