我喜欢 Go 在内部处理 I/O 多路复用的方式epoll
还有另一种机制并自行调度绿色线程(此处为 go-routine),从而可以自由地编写同步代码。
我知道 TCP 套接字是non-blocking
and read
会给EAGAIN
当没有可用数据时。鉴于,conn.Read(buffer)
将检测到这一点并且阻止 go 例程在套接字缓冲区中没有可用数据的情况下执行连接读取。有没有办法在不关闭底层连接的情况下停止这样的例程。我正在使用连接池,因此关闭 TCP 连接对我来说没有意义,并且希望将该连接返回到池中。
下面是模拟这种场景的代码:
func main() {
conn, _ := net.Dial("tcp", "127.0.0.1:9090")
// Spawning a go routine
go func(conn net.Conn) {
var message bytes.Buffer
for {
k := make([]byte, 255) // buffer
m, err := conn.Read(k) // blocks here
if err != nil {
if err != io.EOF {
fmt.Println("Read error : ", err)
} else {
fmt.Println("End of the file")
}
break // terminate loop if error
}
// converting bytes to string for printing
if m > 0 {
for _, b := range k {
message.WriteByte(b)
}
fmt.Println(message.String())
}
}
}(conn)
// prevent main from exiting
select {}
}
如果不可能,我可以采取哪些其他方法:
1) Call syscall.Read
并手动处理这个问题。在这种情况下,我需要一种方法来检查套接字是否可用readable打电话之前syscall.Read
否则我最终会浪费不必要的 CPU 周期。对于我的场景,我想我可以跳过基于事件的轮询并继续调用syscall.Read
因为我的用例中总是有数据。
2)任何建议:)
func receive(conn net.TCPConn, kill <-chan struct{}) error {
// Spawn a goroutine to read from the connection.
data := make(chan []byte)
readErr := make(chan error)
go func() {
for {
b := make([]byte, 255)
_, err := conn.Read(b)
if err != nil {
readErr <- err
break
}
data <- b
}
}()
for {
select {
case b := <-data:
// Do something with `b`.
case err := <-readErr:
// Handle the error.
return err
case <-kill:
// Received kill signal, returning without closing the connection.
return nil
}
}
}
发送一个空结构到kill
从另一个 goroutine 停止从连接接收。这是一个在一秒钟后停止接收的程序:
kill := make(chan struct{})
go func() {
if err := receive(conn, kill); err != nil {
log.Fatal(err)
}
}()
time.Sleep(time.Second)
kill <- struct{}{}
这可能不完全是您正在寻找的,因为读取 goroutine 仍然会被阻塞Read
即使您发送至kill
。然而,处理传入读取的 goroutine 将终止。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)