Golang - 使用两个 Reader 返回的行作为 select 语句

Question

对于一个简单的 golang chat/telnet 客户端，我想将从两个 bufio Readers 接收到的字符串传递给一个 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

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

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无限期阻塞。它正在尝试o通过 channel发送outgoing，但outgoing没有缓冲，也没有任何东西在监听它。此外，由于ReadString()在读取一行之前两个调用都不会返回，因此您的for循环只会在依次从两个读取器读取一行时继续（然后将阻塞通道发送）。

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