使用停止通道停止 bufio.Scanner

Question

我正在写一些从os.Stdin使用bufio.Scanner类似的读取行的东西：for s.scanner.Scan() {  line := s.scanner.Text()  // process line}这是在 goroutine 中运行的，我希望能够在 achan struct{}关闭时停止它。然而，Scan直到有另一条线为止，我不知道如何阻止它，如果没有更多的输入，它将无限期地阻塞。谁能在这里指出我正确的方向？

Answer 1

通过再创建一个间接并忽略底层，我们可以停止。

// actual reading, converts input stream to a channel

func readUnderlying(lines chan interface{}) {

    s := bufio.NewScanner(os.Stdin)

    for s.Scan() {

        lines <- s.Text()

    }

    lines <- s.Err()

}

func read(stop chan struct{}) {

    input := make(chan interface{}) // input stream

    go readUnderlying(input) // go and read

    for {

        select { // read or close

        case lineOrErr := <-input:

            fmt.Println(lineOrErr)

        case <-stop:

            return

        }

    }

}

func main() {

    stop := make(chan struct{})

    go read(stop)

    // wait some to simulate blocking

    time.Sleep(time.Second * 20) // it will print what is given

    close(stop)

    time.Sleep(time.Second * 20) // stopped so no more processing

}