Changeset 757 in code for trunk/conn.go

Timestamp:

Dec 8, 2021, 5:03:40 PM (4 years ago)

Author:

contact

Message:

Add context to {conn,upstreamConn}.SendMessage

This avoids blocking on upstream message rate limiting for too
long.

File:

• trunk/conn.go (modified) (2 diffs)

trunk/conn.go

@@ -218 +218 @@
 // If the connection is closed before the message is sent, SendMessage silently
 // drops the message.
-func (c *conn) SendMessage(msg *irc.Message) {
+func (c *conn) SendMessage(ctx context.Context, msg *irc.Message) {
         c.lock.Lock()
         defer c.lock.Unlock()
@@ -225 +225 @@
                 return
         }
-        c.outgoing <- msg
+
+        select {
+        case c.outgoing <- msg:
+                // Success
+        case <-ctx.Done():
+                c.logger.Printf("failed to send message: %v", ctx.Err())
+        }
 }