如何关闭等待 I/O 的 goroutine

Question

我创建了两个goroutinesender和receiver，sender会不断的从用户（键盘）获取数据并写入stream，receiver会独立获取值从流中将其打印到屏幕上。两者都使用 goroutine 并发

在某个时间点，receiver 失败并关闭连接并退出receiver goroutine，但sender goroutine 等待用户输入（i/o 操作）不会被关闭。在这种情况下如何退出所有的 goroutine？

以下是此场景的示例代码。

package main

import (
    "fmt"
    "time"
)

var stop bool = false

func sender() {
    str := ""
    for !stop {
        fmt.Scanf("%s", &str)
        fmt.Println("Entered :", str)
    }   
    fmt.Println("Closing sender goroutine")
}

func receiver() {
    i := 0
    for !stop {
        i++
        if i > 5 { 
            stop = true
        }
        time.Sleep(1 * time.Second)
    }   
    fmt.Println("Closing receiver goroutine")
}

func main() {
    go sender()
    go receiver()

    /* Wait for goroutines to finish */
    for !stop {
        time.Sleep(1 * time.Millisecond)
    }   
    time.Sleep(1 * time.Second)

    panic("Display stack")
}

上述代码发送方将在 5 次循环后等待用户输入接收方将退出接收方 go 例程。 I expect when receiver close, go routine which waiting on i/o has to be closed.

请帮我解答这个问题。

Answer 1

正如 Dave C 和 JimB 所说，使用通道来协调 goroutine。这是一个可能有帮助的例子。

收到用户的5条消息后退出：

package main

import "fmt"

var pipe = make(chan string) //shares text entered by user
var stop = make(chan bool)   //shares stop signal

func listen() {
    for {
        var input string
        fmt.Scan(&input)
        pipe <- input
    }
}

func write() {
    for i := 0; i < 5; i++ {
        var output string
        output = <-pipe
        fmt.Println("Received", output)
    }

    stop <- true
}

func main() {
    go listen()
    go write()

    <-stop
}

Answer 2

首先，您的代码围绕stop 变量展开竞争。当存在数据竞争时，无法保证您的程序将按照定义运行。使用通道来同步 goroutine。然而，这并不是您继续编程的原因。

您的代码在fmt.Scanf 上阻塞，无法检查stop 条件。由于无法中断对 Stdin 的读取（发生在 fmt.Scanf 内部），您需要在再次调用 Scanf 之前检查停止条件。如果没有更多输入，但您在 Stdin 上有一个未决的读取，处理它的最简单方法是让该 goroutine 运行。有一些相当复杂的方法可以使用称为“self-pipe”的技巧来解决这个问题，但通常不值得付出努力，因为 goroutine 很小并且不会占用很多资源。

for {
    fmt.Scanf("%s", &str)
    fmt.Println("Entered :", str)
    // use a channel or context to detect when to exit
    select {
    case <-ctx.Done():
        return
    default:
    }
}