fix: close and release a socket when error occurs

Fixed #179
panjf2000 · Jan 30, 2021 · 9d86c92 · 9d86c92
1 parent 83b96ed
commit 9d86c92
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/acceptor_unix.go b/acceptor_unix.go
@@ -46,13 +46,19 @@ func (svr *server) acceptNewConnection(fd int) error {
 	el := svr.lb.next(netAddr)
 	c := newTCPConn(nfd, el, sa, netAddr)
 
-	_ = el.poller.Trigger(func() (err error) {
+	err = el.poller.Trigger(func() (err error) {
 		if err = el.poller.AddRead(nfd); err != nil {
+			_ = unix.Close(nfd)
+			c.releaseTCP()
 			return
 		}
 		el.connections[nfd] = c
 		err = el.loopOpen(c)
 		return
 	})
+	if err != nil {
+		_ = unix.Close(nfd)
+		c.releaseTCP()
+	}
 	return nil
 }