Changeset 739 in code for trunk/upstream.go

Timestamp:

Dec 2, 2021, 10:27:12 PM (4 years ago)

Author:

contact

Message:

Add context to upstreamConn.handleMessage

File:

• trunk/upstream.go (modified) (7 diffs)

trunk/upstream.go

@@ -413 +413 @@
 }
 
-func (uc *upstreamConn) handleMessage(msg *irc.Message) error {
+func (uc *upstreamConn) handleMessage(ctx context.Context, msg *irc.Message) error {
         var label string
         if l, ok := msg.GetTag("label"); ok {
@@ -492 +492 @@
                         if ch != nil && msg.Command != "TAGMSG" {
                                 if ch.Detached {
-                                        uc.handleDetachedMessage(ch, msg)
+                                        uc.handleDetachedMessage(ctx, ch, msg)
                                 }
 
@@ -663 +663 @@
                 if dc, _ := uc.dequeueCommand("AUTHENTICATE"); dc != nil && dc.sasl != nil {
                         if msg.Command == irc.RPL_SASLSUCCESS {
-                                uc.network.autoSaveSASLPlain(context.TODO(), dc.sasl.plainUsername, dc.sasl.plainPassword)
+                                uc.network.autoSaveSASLPlain(ctx, dc.sasl.plainUsername, dc.sasl.plainPassword)
                         }
 
@@ -685 +685 @@
                                         return err
                                 }
-                                uc.network.autoSaveSASLPlain(context.TODO(), account, password)
+                                uc.network.autoSaveSASLPlain(ctx, account, password)
                         }
 
@@ -1716 +1716 @@
 }
 
-func (uc *upstreamConn) handleDetachedMessage(ch *Channel, msg *irc.Message) {
+func (uc *upstreamConn) handleDetachedMessage(ctx context.Context, ch *Channel, msg *irc.Message) {
         if uc.network.detachedMessageNeedsRelay(ch, msg) {
                 uc.forEachDownstream(func(dc *downstreamConn) {
@@ -1724 +1724 @@
         if ch.ReattachOn == FilterMessage || (ch.ReattachOn == FilterHighlight && uc.network.isHighlight(msg)) {
                 uc.network.attach(ch)
-                if err := uc.srv.db.StoreChannel(context.TODO(), uc.network.ID, ch); err != nil {
+                if err := uc.srv.db.StoreChannel(ctx, uc.network.ID, ch); err != nil {
                         uc.logger.Printf("failed to update channel %q: %v", ch.Name, err)
                 }
@@ -1915 +1915 @@
                 }
 
-                if err := uc.handleMessage(msg); err != nil {
+                if err := uc.handleMessage(context.TODO(), msg); err != nil {
                         if _, ok := err.(registrationError); ok {
                                 return err