avoid memory leak when reconnecting to a server

bluenviron · Aug 23, 2022 · b7cbaec · b7cbaec
1 parent ba2ceac
commit b7cbaec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/endpointclient.go b/endpointclient.go
@@ -116,11 +116,8 @@ func (t *endpointClient) run() {
 	for {
 		t.runInner()
 
-		timer := time.NewTimer(netReconnectPeriod)
-		defer timer.Stop()
-
 		select {
-		case <-timer.C:
+		case <-time.After(netReconnectPeriod):
 		case <-t.ctx.Done():
 			return
 		}

diff --git a/pkg/udplistener/udplistener.go b/pkg/udplistener/udplistener.go
@@ -99,11 +99,8 @@ func (c *conn) Read(byt []byte) (int, error) {
 	var ok bool
 
 	if !c.readDeadline.IsZero() {
-		readTimer := time.NewTimer(time.Until(c.readDeadline))
-		defer readTimer.Stop()
-
 		select {
-		case <-readTimer.C:
+		case <-time.After(time.Until(c.readDeadline)):
 			return 0, errTimeout
 		case buf, ok = <-c.read:
 		}