check deadline before sendMsg, because select is randomly, sometimes …

…it will write successfully after deadline Signed-off-by: Asutorufa <[email protected]>
libp2p · Dec 13, 2023 · a773ffa · a773ffa
1 parent 828214b
commit a773ffa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/session.go b/session.go
@@ -490,6 +490,12 @@ func (s *Session) sendMsg(hdr header, body []byte, deadline <-chan struct{}) err
 	default:
 	}
 
+	select {
+	case <-deadline:
+		return ErrTimeout
+	default:
+	}
+
 	// duplicate as we're sending this async.
 	buf := pool.Get(headerSize + len(body))
 	copy(buf[:headerSize], hdr[:])