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

visli · · 1813 次点击
<a href="/user/buscoop" title="@buscoop">@buscoop</a> 最终方案就是1楼的办法。 另外,我最初的这个方案,似乎也不会报race问题,这是令我困惑的。 ``` func NewSrv() { ln, err := net.ListenTCP() defer ln.Close() ... go func() { &lt;- exitChann ln.Close() } for { c, err:= ln.Accept() .... } } ```
#14
更多评论
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
<a href="/user/jan-bar" title="@jan-bar">@jan-bar</a> 谢谢。目前只有这个办法了。
#2