我正在使用 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()
}
}
正确答案
假设 stdout
是 cmdobject.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()
}