卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34097本站已运行391

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

如何为循环扫描的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()
}

卓越飞翔博客
上一篇: 有条件地将多个属性分配给结构的正确方法是什么
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏