请教关闭TCP监听时的争用问题

visli · · 2415 次点击
jan-bar
想要拥有,必定付出。
`Accept()`源码里面没有任何地方用到`ctx`,所以你的`ctx`影响不到`Accept()`。你的`cancelFunc `是全局变量,题主的`listener `也是全局变量,全局变量如果不加锁保护,多个协程访问就会race。除非你的全局变量不会在不同协程里面同时读写。如果所有协程都读全局变量(永远不写时)也不会race的。你可以用`go build -race`开启竟态检查,如果有资源出现race会打印一些信息。
#6
更多评论
jan-bar
想要拥有,必定付出。
我的写法核心就是让`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
<a href="/user/jan-bar" title="@jan-bar">@jan-bar</a> 谢谢。目前只有这个办法了。
#2