small fixes

amirylm · Oct 2, 2023 · fe23c01 · fe23c01
1 parent effdc5b
commit fe23c01
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/core/pubsub.go b/core/pubsub.go
@@ -217,7 +217,9 @@ func (c *Controller) trySubscribe(topic *pubsub.Topic) (sub *pubsub.Subscription
 func (c *Controller) validateMsg(ctx context.Context, p peer.ID, msg *pubsub.Message) pubsub.ValidationResult {
 	res, err := c.valRouter.HandleWait(ctx, p, msg)
 	if err != nil {
-		c.lggr.Warnw("failed to handle msg", "err", err)
+		if ctx.Err() == nil {
+			c.lggr.Warnw("failed to handle msg", "err", err)
+		}
 		return pubsub.ValidationIgnore
 	}
 	return res

diff --git a/core/router.go b/core/router.go
@@ -78,7 +78,6 @@ func (r *msgRouter[T]) Start(pctx context.Context) {
 			}
 			r.dispatchMsg(msg)
 		case <-ctx.Done():
-			close(r.q)
 			return
 		}
 	}