chore(metric): Add default structure

appleboy · appleboy · commit 48f76ed58482 · 2021-12-24T22:01:08.000+08:00
Signed-off-by: Bo-Yi Wu &lt;appleboy.tw@gmail.com&gt;
diff --git a/go.mod b/go.mod
@@ -4,6 +4,6 @@ go 1.16
 
 require (
 	github.com/go-redis/redis/v8 v8.11.3
-	github.com/golang-queue/queue v0.0.10
+	github.com/golang-queue/queue v0.0.11
 	github.com/stretchr/testify v1.7.0
 )
diff --git a/go.sum b/go.sum
@@ -13,6 +13,8 @@ github.com/go-redis/redis/v8 v8.11.3/go.mod h1:xNJ9xDG09FsIPwh3bWdk+0oDWHbtF9rPN
 github.com/go-task/slim-sprig v0.0.0-20210107165309-348f09dbbbc0/go.mod h1:fyg7847qk6SyHyPtNmDHnmrv/HOrqktSC+C9fM+CJOE=
 github.com/golang-queue/queue v0.0.10 h1:cGqMgHMf2eamwdd3hmOzGcSQogGu9tMhhVYPQMrMC1g=
 github.com/golang-queue/queue v0.0.10/go.mod h1:ku8iyjYffqYY6Duts+xl+QYfN3/KDK4MEvXMZUkHyio=
+github.com/golang-queue/queue v0.0.11 h1:qaE7rOYUh+11NcO6U9zZ7NazglIwEp0j7Dn0sSoavcI=
+github.com/golang-queue/queue v0.0.11/go.mod h1:ku8iyjYffqYY6Duts+xl+QYfN3/KDK4MEvXMZUkHyio=
 github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=
 github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=
diff --git a/redis.go b/redis.go
@@ -116,6 +116,13 @@ func WithLogger(l queue.Logger) Option {
 	}
 }
 
+// WithMetric set custom Metric
+func WithMetric(m queue.Metric) Option {
+	return func(w *Worker) {
+		w.metric = m
+	}
+}
+
 // NewWorker for struc
 func NewWorker(opts ...Option) *Worker {
 	var err error
@@ -128,6 +135,7 @@ func NewWorker(opts ...Option) *Worker {
 		runFunc: func(context.Context, queue.QueuedMessage) error {
 			return nil
 		},
+		metric: queue.NewMetric(),
 	}
 
 	// Loop through each option

Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,6 @@ go 1.16`
`4`	`4`
`5`	`5`	`require (`
`6`	`6`	`github.com/go-redis/redis/v8 v8.11.3`
`7`		`- github.com/golang-queue/queue v0.0.10`
	`7`	`+ github.com/golang-queue/queue v0.0.11`
`8`	`8`	`github.com/stretchr/testify v1.7.0`
`9`	`9`	`)`
Original file line number	Diff line number	Diff line change
`@@ -116,6 +116,13 @@ func WithLogger(l queue.Logger) Option {`
`116`	`116`	`}`
`117`	`117`	`}`
`118`	`118`
	`119`	`+// WithMetric set custom Metric`
	`120`	`+func WithMetric(m queue.Metric) Option {`
	`121`	`+ return func(w *Worker) {`
	`122`	`+ w.metric = m`
	`123`	`+ }`
	`124`	`+}`
	`125`	`+`
`119`	`126`	`// NewWorker for struc`
`120`	`127`	`func NewWorker(opts ...Option) *Worker {`
`121`	`128`	`var err error`
`@@ -128,6 +135,7 @@ func NewWorker(opts ...Option) *Worker {`
`128`	`135`	`runFunc: func(context.Context, queue.QueuedMessage) error {`
`129`	`136`	`return nil`
`130`	`137`	`},`
	`138`	`+ metric: queue.NewMetric(),`
`131`	`139`	`}`
`132`	`140`
`133`	`141`	`// Loop through each option`