运行超时命令并在 go 中一次读取一行 stdout

从现在开始，我们要努力学习啦！今天我给大家带来《运行超时命令并在 go 中一次读取一行 stdout》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

我想运行一个命令并打印其 stdout 中的每一行（当它可用时）。此外，如果命令在 n 秒内没有完成，我也想终止它。

有一些在 golang 中实现超时的示例（特别是在 golang 中终止使用 os/exec 启动的进程）。我在 select 中有一个 time.after() 子句，我希望在 2 秒后命中，此时 runtraceroute 应该返回 - 但这不会发生。

我的代码如下（以及正在进行的演示：http://play.golang.org/p/d4acoawemt）

package main

import (
    "bufio"
    "fmt"
    "log"
    "os"
    "os/exec"
    "time"
)

func runtraceroute(host string) {
    errch := make(chan error, 1)
    cmd := exec.command("/usr/bin/traceroute", host)
    stdout, err := cmd.stdoutpipe()
    if err != nil {
        log.fatal(err)
    }
    if err := cmd.start(); err != nil {
        log.fatal(err)
    }
    go func() {
        errch <- cmd.wait()
    }()
    select {
    case <-time.after(time.second * 2):
        log.println("timeout hit..")
        return
    case err := <-errch:
        if err != nil {
            log.println("traceroute failed:", err)
        }
    default:
        for _, char := range "|/-\\" {
            fmt.printf("\r%s...%c", "running traceroute", char)
            time.sleep(100 * time.millisecond)
        }
        scanner := bufio.newscanner(stdout)
        fmt.println("")
        for scanner.scan() {
            line := scanner.text()
            log.println(line)
        }
    }
}

func main() {
    runtraceroute(os.args[1])
}

输出（剪掉本地 ip 地址和网关的前几行）：

$ go run pipe.go xinhua.com
Running traceroute...\
2016/04/01 11:16:43 traceroute to xinhua.com (58.64.200.76), 30 hops max, 60 byte packets
.......
.......
.....More deleted lines.....

2016/04/01 11:16:49 12  * * *
2016/04/01 11:16:49 13  4.68.63.214 (4.68.63.214)  3.208 ms  3.176 ms  3.241 ms
2016/04/01 11:16:49 14  * * *
2016/04/01 11:16:49 15  if-ae-9-2.tcore1.TV2-Tokyo.as6453.net (180.87.180.18)  160.314 ms  158.837 ms  161.438 ms
2016/04/01 11:16:49 16  if-ae-5-7.tcore1.HK2-Hong-Kong.as6453.net (180.87.112.189)  157.497 ms if-ae-3-2.tcore1.HK2-Hong-Kong.as6453.net (180.87.112.5)  161.397 ms if-ae-5-7.tcore1.HK2-Hong-Kong.as6453.net (180.87.112.189)  159.351 ms
2016/04/01 11:16:49 17  if-ge-10-0-0-1128.core1.undefined.as6453.net (180.87.160.73)  156.412 ms  156.522 ms if-ge-14-0-0-1126.core1.undefined.as6453.net (180.87.112.30)  156.605 ms
2016/04/01 11:16:49 18  * * *
2016/04/01 11:16:49 19  * * *
2016/04/01 11:16:49 20  * * *
2016/04/01 11:16:49 21  113.10.229.113 (113.10.229.113)  165.578 ms  165.818 ms  163.451 ms
2016/04/01 11:16:49 22  113.10.229.74 (113.10.229.74)  163.564 ms ae5.10g-idc.wpc.nwtgigalink.com (113.10.229.66)  162.384 ms 113.10.229.74 (113.10.229.74)  167.026 ms
2016/04/01 11:16:49 23  113.10.230.162 (113.10.230.162)  162.988 ms  162.777 ms  163.807 ms
2016/04/01 11:16:49 24  58.64.160.164 (58.64.160.164)  161.902 ms  162.396 ms  164.986 ms
2016/04/01 11:16:54 25  * * *
2016/04/01 11:16:54 26  58.64.200.76 (58.64.200.76)  162.178 ms !X  162.509 ms !X  162.356 ms !X

解决方案

我认为你想将 default: 案例的主体放入 goroutine 中；我怀疑它会阻止您的 case <-time.after(time.second * 2): 标签被击中。

另外，请记住 time.after 并不保证它会在该持续时间之后准确命中，它只是说在该持续时间之后的任何时间它将在通道上发送信号，这可能是在指定的持续时间。请参阅底层文档 time.NewTimer

我修改了你的示例：http://play.golang.org/p/TggNQ1d57Y

package main

import (
    "bufio"
    "fmt"
    "log"
    "os/exec"
    "time"
)

func RunTraceroute(host string) {
    errch := make(chan error, 1)
    cmd := exec.Command("/usr/bin/traceroute", host)

    stdout, err := cmd.StdoutPipe()
    if err != nil {
        log.Fatal(err)
    }

    if err := cmd.Start(); err != nil {
        log.Fatal(err)
    }

    go func() {
        errch <- cmd.Wait()
    }()

    go func() {
        for _, char := range "|/-\\" {
            fmt.Printf("\r%s...%c", "Running traceroute", char)
            time.Sleep(100 * time.Millisecond)
        }
        scanner := bufio.NewScanner(stdout)
        fmt.Println("")
        for scanner.Scan() {
            line := scanner.Text()
            log.Println(line)
        }
    }()

    select {
    case <-time.After(time.Second * 1):
        log.Println("Timeout hit..")
        return
    case err := <-errch:
        if err != nil {
            log.Println("traceroute failed:", err)
        }
    }
}

func main() {
    RunTraceroute("8.8.8.8")
}

这对我有用

今天关于《运行超时命令并在 go 中一次读取一行 stdout》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！