我一个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()`源码里面没有任何地方用到`ctx`,所以你的`ctx`影响不到`Accept()`。你的`cancelFunc `是全局变量,题主的`listener `也是全局变量,全局变量如果不加锁保护,多个协程访问就会race。除非你的全局变量不会在不同协程里面同时读写。如果所有协程都读全局变量(永远不写时)也不会race的。你可以用`go build -race`开启竟态检查,如果有资源出现race会打印一些信息。
#6
更多评论
我的写法核心就是让`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