We should also abort when the writer context is cancelled.

go-zeromq · Aug 25, 2023 · 099cec7 · 099cec7
1 parent a7c9a4a
commit 099cec7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rep.go b/rep.go
@@ -258,6 +258,8 @@ func (r *repWriter) rmConn(conn *Conn) {
 func (r *repWriter) write(ctx context.Context, msg Msg) error {
 	conn, preamble := r.state.Get()
 	select {
+	case <-ctx.Done():
+		return ctx.Err()
 	case <-r.ctx.Done(): // repWriter.run() terminates on this, sendCh <- will not complete
 		return r.ctx.Err()
 	case r.sendCh <- repSendPayload{conn, preamble, msg}: