首页 > 后端开发 > Golang > 如何为循环扫描的bufio扫描仪设置超时?

如何为循环扫描的bufio扫描仪设置超时?

WBOY
发布: 2024-02-06 08:05:07
转载
1119 人浏览过

如何为循环扫描的bufio扫描仪设置超时?

问题内容

我正在使用 golang 启动进程并监视输出。该进程将运行很长时间,我需要能够发送信号来结束它。

我有以下代码,在大多数情况下运行良好。但由于某些原因,进程可能没有输出,for循环会被scan()方法阻塞,无法接收来自processfinishchan的消息。

是否有一种简单的方法可以为 scan() 方法设置超时?我尝试了一个解决方案,在另一个 goroutine 中运行 scan() ,并使用另一个 select 从新的 goroutine 和超时通道接收,但考虑到外部 for 循环,是否会有越来越多的 goroutine 被 scan 阻塞? p>

// code that start the process...
scanner := bufio.NewScanner(stdout)

for {
    select {
    case <-processFinishChan: // send to this channel to terminate the process
        log.Println("Killing Process")
        err := cmdObject.Process.Kill()
        if err != nil {
            log.Printf("Error Killing: %v", err)
        } else {
            return
        }
    default:
        // default case, read the output of process and send to user.
        if !scanner.Scan() && scanner.Err() == nil {
            // reach EOF
            return
        }
        m := scanner.Bytes()

        WSOutChanHolder.mu.Lock()
        for _, ch := range WSOutChanHolder.data {
            ch <- m
        }
        WSOutChanHolder.mu.Unlock()
    }
}
登录后复制


正确答案


假设 stdoutcmdobject.stdoutpipe() 的结果,则读取器应在进程退出后等待时关闭读取器并中断任何正在进行的读取。

wait 将在看到命令退出后关闭管道,因此大多数调用者不需要自己关闭管道。

所以,我们需要在一个单独的goroutine中杀死该进程,然后在杀死该进程后等待观察它的发生并关闭阅读器。

// code that start the process...
scanner := bufio.NewScanner(stdout)

go func() {
    <-processFinishChan: // send to this channel to terminate the process
    log.Println("Killing Process")
    err := cmdObject.Process.Kill()
    if err != nil {
        log.Printf("Error Killing: %v", err)
    } 

    cmdObject.Wait()
} ()

for {
    // default case, read the output of process and send to user.
    if !scanner.Scan() && scanner.Err() == nil {
        // reach EOF
        return
    }
    m := scanner.Bytes()

    WSOutChanHolder.mu.Lock()
    for _, ch := range WSOutChanHolder.data {
        ch <- m
    }
    WSOutChanHolder.mu.Unlock()
}
登录后复制

以上是如何为循环扫描的bufio扫描仪设置超时?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:stackoverflow.com
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板