如何为 GoLang bufio.Writer 设置写入截止日期?

How to set a write deadline for GoLang bufio.Writer?

我在 GoLang 中使用 buffio.Writer 如下。

conn, err := net.Dial("tcp", address) // address is of form ip:port
w := bufio.NewWriter(conn)
w.WriteByte(code) // code is a byte
w.Write(data) // data is a byte buffer
w.Flush()

我试图修改上面的代码,使 write() 事件有一个截止日期:当截止日期过去时,write() 事件将 return 不管成功与否。

在 GoLang 中,如果直接使用 conn(连接对象)使用 conn.SetWriteDeadline(time.Now().Add(n * time.Second)) 进行写入,则可能有截止日期。但是,当我使用 bufifo.writer 对象时,它本质上是 conn 的包装器用于缓冲 IO,没有 API 来设置截止日期。

虽然可以使用 conn.SetWriteDeadline(time.Now().Add(n * time.Second)) 并使用 conn.Write(b),但效率非常低,因为它不缓冲 write 事件(因此有很多上下文切换)

在 GoLang 中有没有一种方法可以将缓冲 IO 与 writeDeadline() 一起使用?

谢谢

这里有两种情况需要注意

如果您想要每个 write() 截止日期,则无法使用缓冲。当您使用缓冲时,实际的 write() 会在缓冲区已满时触发。因此从技术上讲,不可能知道您的 write() 何时完成。在这种情况下,您实际上是在使用 conn.write(),您可以使用 conn.SetWriteDeadline(time.Now().Add(n * time.Second)).

在第二种情况下,正如@icza 在评论中提到的,您可以在底层 conn 对象中设置截止日期,并且 buffio.writer() 包装器将遵守此规则。虽然这在语义上是正确的,但它没有提供您想要的网络抽象。