fix: memory leak (#279)

xtaci · Oct 4, 2024 · da3a437 · da3a437
1 parent 2ffcdaf
commit da3a437
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/sess.go b/sess.go
@@ -1035,7 +1035,10 @@ func (l *Listener) Accept() (net.Conn, error) {
 func (l *Listener) AcceptKCP() (*UDPSession, error) {
 	var timeout <-chan time.Time
 	if tdeadline, ok := l.rd.Load().(time.Time); ok && !tdeadline.IsZero() {
-		timeout = time.After(time.Until(tdeadline))
+		timer := time.NewTimer(time.Until(tdeadline))
+		defer timer.Stop()
+
+		timeout = timer.C
 	}
 
 	select {

diff --git a/timedsched.go b/timedsched.go
@@ -83,8 +83,10 @@ func NewTimedSched(parallel int) *TimedSched {
 
 // sched is a goroutine to schedule and execute timed tasks.
 func (ts *TimedSched) sched() {
-	var tasks timedFuncHeap
 	timer := time.NewTimer(0)
+	defer timer.Stop()
+
+	var tasks timedFuncHeap
 	drained := false
 	for {
 		select {