我一个TCP服务,希望从外部关闭Listener,但这样就在listener.Close()与listener.Accept()之间产生了race。请教有什么好的解决办法?
简单的示例代码如下, 加"-race"参数运行,再ctrl+c退出运行,就会报race问题:
```
package main
import (
"log"
"net"
"os"
"os/signal"
"syscall"
"time"
)
var listener *net.TCPListener
// tcp服务
func tcpServer() {
addr, err := net.ResolveTCPAddr("tcp4", ":6666")
if err != nil {
log.Fatalln(err)
}
listener, err = net.ListenTCP("tcp4", addr)
if err != nil {
log.Fatalln(err)
}
for {
c, err := listener.AcceptTCP()
if err != nil {
if err.(*net.OpError).Err == net.ErrClosed {
log.Println("listener closed")
return
}
log.Println(err)
continue
}
connHandler(c)
}
}
// 新连接处理
func connHandler(c *net.TCPConn) {
defer func() {
c.Close()
log.Printf("%s closed", c.RemoteAddr().String())
}()
log.Printf("New connection: %s", c.RemoteAddr().String())
buf := make([]byte, 128)
for {
n, err := c.Read(buf)
if err != nil {
return
}
log.Println(buf[:n])
}
}
// 关闭listener
func listenerClose() {
if listener != nil {
listener.Close()
}
}
func main() {
sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
go tcpServer()
<-sigs
listenerClose()
time.Sleep(time.Millisecond * 200)
}
```
我的写法核心就是让`Accept()`操作设置超时时间,保证可以执行其他代码。
或者关闭代码就在你的程序里面模拟tcp连接发送特殊字符,然后服务端收到特殊字符就走退出逻辑。
```go
package main
import (
"context"
"log"
"net"
"os"
"os/signal"
"syscall"
"time"
)
// tcp服务
func tcpServer(ctx context.Context) {
addr, err := net.ResolveTCPAddr("tcp4", ":6666")
if err != nil {
log.Fatalln(err)
}
listener, err := net.ListenTCP("tcp4", addr)
if err != nil {
log.Fatalln(err)
}
for {
select {
case <-ctx.Done():
err = listener.Close()
if err != nil {
log.Fatalln(err)
}
return
default:
// 控制超时时间,让代码有机会执行ctx.Done()
_ = listener.SetDeadline(time.Now().Add(time.Second))
c, err := listener.AcceptTCP()
if err != nil {
if opErr, ok := err.(*net.OpError); ok {
if opErr.Timeout() {
continue
}
if opErr.Err == net.ErrClosed {
log.Println("listener closed")
return
}
}
log.Println(err)
continue
}
connHandler(c)
}
}
}
// 新连接处理
func connHandler(c *net.TCPConn) {
defer func() {
c.Close()
log.Printf("%s closed", c.RemoteAddr().String())
}()
log.Printf("New connection: %s", c.RemoteAddr().String())
buf := make([]byte, 128)
for {
n, err := c.Read(buf)
if err != nil {
return
}
log.Println(buf[:n])
}
}
func main() {
sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
ctx, cancel := context.WithCancel(context.Background())
go tcpServer(ctx)
<-sigs
cancel()
time.Sleep(time.Millisecond * 200)
}
```
#1
更多评论
net.Listen() 的时候就支持传入一个 context,demo 如下:
```go
package main
import (
"context"
"net"
)
var cancelFunc context.CancelFunc
func main() {
ctx, cancel := context.WithCancel(context.Background())
cancelFunc = cancel
lc := new(net.ListenConfig)
lis, _ := lc.Listen(ctx, "tcp4", ":6666") // demo 忽略 error
for {
conn, _ := lis.Accept() // demo 忽略 error
tcpConn := conn.(*net.TCPConn)
// TODO handle tcpConn
}
}
func Close() {
if cancelFunc != nil {
cancelFunc()
}
}
```
#3