fix(idleconn): Ensure cancel func is not nill before calling it

gabe565 · Nov 13, 2024 · aa1f9df · aa1f9df
1 parent 9e93ac0
commit aa1f9df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/server/idleconn/conn.go b/internal/server/idleconn/conn.go
@@ -28,7 +28,7 @@ func (c *Conn) Write(p []byte) (int, error) {
 	_ = c.updateDeadline()
 
 	n, err := c.Conn.Write(p)
-	if errors.Is(err, os.ErrDeadlineExceeded) {
+	if c.cancel != nil && errors.Is(err, os.ErrDeadlineExceeded) {
 		c.cancel()
 	}
 	return n, err
@@ -38,7 +38,7 @@ func (c *Conn) Read(b []byte) (int, error) {
 	_ = c.updateDeadline()
 
 	n, err := c.Conn.Read(b)
-	if errors.Is(err, os.ErrDeadlineExceeded) {
+	if c.cancel != nil && errors.Is(err, os.ErrDeadlineExceeded) {
 		c.cancel()
 	}
 	return n, err