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

visli · · 2376 次点击
<a href="/user/jan-bar" title="@jan-bar">@jan-bar</a> 谢谢。目前只有这个办法了。
#2
更多评论
jan-bar
想要拥有,必定付出。
我的写法核心就是让`Accept()`操作设置超时时间,保证可以执行其他代码。 或者关闭代码就在你的程序里面模拟tcp连接发送特殊字符,然后服务端收到特殊字符就走退出逻辑。 ```go package main import ( &#34;context&#34; &#34;log&#34; &#34;net&#34; &#34;os&#34; &#34;os/signal&#34; &#34;syscall&#34; &#34;time&#34; ) // tcp服务 func tcpServer(ctx context.Context) { addr, err := net.ResolveTCPAddr(&#34;tcp4&#34;, &#34;:6666&#34;) if err != nil { log.Fatalln(err) } listener, err := net.ListenTCP(&#34;tcp4&#34;, addr) if err != nil { log.Fatalln(err) } for { select { case &lt;-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(&#34;listener closed&#34;) return } } log.Println(err) continue } connHandler(c) } } } // 新连接处理 func connHandler(c *net.TCPConn) { defer func() { c.Close() log.Printf(&#34;%s closed&#34;, c.RemoteAddr().String()) }() log.Printf(&#34;New connection: %s&#34;, 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) &lt;-sigs cancel() time.Sleep(time.Millisecond * 200) } ```
#1
net.Listen() 的时候就支持传入一个 context,demo 如下: ```go package main import ( &#34;context&#34; &#34;net&#34; ) var cancelFunc context.CancelFunc func main() { ctx, cancel := context.WithCancel(context.Background()) cancelFunc = cancel lc := new(net.ListenConfig) lis, _ := lc.Listen(ctx, &#34;tcp4&#34;, &#34;:6666&#34;) // demo 忽略 error for { conn, _ := lis.Accept() // demo 忽略 error tcpConn := conn.(*net.TCPConn) // TODO handle tcpConn } } func Close() { if cancelFunc != nil { cancelFunc() } } ```
#3