Golang - 使用两条 Reader 返回的行进行选择语句

Question

对于一个简单的 golang 聊天/远程登录客户端，我想将从两个 bufio 阅读器接收到的字符串传递给一个 select 语句，这样我就可以将用户输入发送到服务器，或者打印服务器发送的数据。

conn, _ := net.Dial("tcp", "localhost:8998")
for {
    select{
    case line, _ := bufio.NewReader(os.Stdin).ReadString('\n'):
        fmt.Print("> ")
        fmt.Fprintf(conn, line + "\n")
    case data, _ := bufio.NewReader(conn).ReadString('\n'):
        fmt.Print(data)
    }
}

编译器返回这个错误

select case must be receive, send or assign recv

我怀疑我应该使用频道。但是

conn, _ := net.Dial("tcp", "localhost:8998")
outgoing := make(chan string)
incoming := make(chan string)
for {
    inputReader := bufio.NewReader(os.Stdin)
    connReader := bufio.NewReader(conn)

    o, _ := inputReader.ReadString('\n')
    i, _ := connReader.ReadString('\n')
    outgoing <- o
    incoming <- i

    select{
    case out := <-outgoing:
        fmt.Print("> ")
        fmt.Fprintf(conn, out + "\n")
    case in := <-incoming:
        fmt.Print(in)
    }
}

但代码不接受或接收数据。最后，我怀疑我应该使用两个 go 例程来检查 Reader 返回值？

Answer 1

您需要在 goroutine 中运行它们，否则它们不能同时发生：

conn, _ := net.Dial("tcp", "localhost:8998")

// Make outgoing reader routine
outgoing := make(chan string)
go func() {
    inputReader := bufio.NewReader(os.Stdin)
    for {
        o, err := inputReader.ReadString('\n')
        if err != nil {
            fmt.Printf("outgoing error: %v", err)
            return
        }
        outgoing <- o
    }
}()

// Make incoming reader routine
incoming := make(chan string)
go func() {
    connReader := bufio.NewReader(conn)
    for {
        i, err := connReader.ReadString('\n')
        if err != nil {
            fmt.Printf("incoming error: %v", err)
            return
        }
        incoming <- i
    }
}()

for {
    select {
    case out := <-outgoing:
        fmt.Print("> ")
        fmt.Fprintf(conn, out+"\n")
    case in := <-incoming:
        fmt.Print(in)
    }
}

编辑

进一步说明：您的第一个示例不起作用，因为select 要求每种情况都是通道读取或通道发送操作，而函数调用bufio.Reader.ReadString() 两者都不是。

您的第二个示例不起作用，因为outgoing <- o 行将无限期阻塞。它试图通过通道outgoing 发送o，但outgoing 没有被缓冲，并且没有任何东西在监听它。此外，由于在读取一行之前，ReadString() 调用都不会返回，因此您的 for 循环只会在依次从 BOTH 阅读器读取一行后继续进行（然后将阻塞通道发送)。

这就是为什么您需要每个 Reader 都有自己的 goroutine，因此每个 Reader 都可以在输入允许的情况下独立读取其 Reader，并在读取行时将其存储到相关通道，然后您的 select 语句可以对每个响应行，因为它来自各个 goroutine。