From 58f023a8ef938217844f3ea23abde9e35001d251 Mon Sep 17 00:00:00 2001
From: edwin <edwin@maicoin.com>
Date: Tue, 4 Jun 2024 22:46:56 +0800
Subject: [PATCH] refactor: pass custom log to pubsub

---
 pubsub.go | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/pubsub.go b/pubsub.go
index efc2354a..b43ed524 100644
--- a/pubsub.go
+++ b/pubsub.go
@@ -470,6 +470,12 @@ func (c *PubSub) ChannelWithSubscriptions(_ context.Context, size int) <-chan in
 
 type ChannelOption func(c *channel)
 
+func WithLogger(logger internal.Logging) ChannelOption {
+	return func(c *channel) {
+		c.logger = logger
+	}
+}
+
 // WithChannelSize specifies the Go chan size that is used to buffer incoming messages.
 //
 // The default is 100 messages.
@@ -510,6 +516,8 @@ type channel struct {
 	chanSize        int
 	chanSendTimeout time.Duration
 	checkInterval   time.Duration
+
+	logger internal.Logging
 }
 
 func newChannel(pubSub *PubSub, opts ...ChannelOption) *channel {
@@ -519,6 +527,7 @@ func newChannel(pubSub *PubSub, opts ...ChannelOption) *channel {
 		chanSize:        100,
 		chanSendTimeout: time.Minute,
 		checkInterval:   3 * time.Second,
+		logger:          internal.Logger,
 	}
 	for _, opt := range opts {
 		opt(c)
@@ -602,12 +611,12 @@ func (c *channel) initMsgChan() {
 						<-timer.C
 					}
 				case <-timer.C:
-					internal.Logger.Printf(
-						ctx, "redis: %s channel is full for %s (message is dropped)",
+					c.logger.Printf(
+						ctx, "redis: %+v channel is full for %s (message is dropped)",
 						c, c.chanSendTimeout)
 				}
 			default:
-				internal.Logger.Printf(ctx, "redis: unknown message type: %T", msg)
+				c.logger.Printf(ctx, "redis: unknown message type: %T", msg)
 			}
 		}
 	}()
@@ -656,12 +665,12 @@ func (c *channel) initAllChan() {
 						<-timer.C
 					}
 				case <-timer.C:
-					internal.Logger.Printf(
-						ctx, "redis: %s channel is full for %s (message is dropped)",
+					c.logger.Printf(
+						ctx, "redis: %+v channel is full for %s (message is dropped)",
 						c, c.chanSendTimeout)
 				}
 			default:
-				internal.Logger.Printf(ctx, "redis: unknown message type: %T", msg)
+				c.logger.Printf(ctx, "redis: unknown message type: %T", msg)
 			}
 		}
 	}()