@@ -3,7 +3,6 @@ package timecapsule
33import (
44 "time"
55
6- "github.com/sirupsen/logrus"
76 "golang.org/x/net/context"
87)
98
@@ -58,7 +57,6 @@ func mergeTimeCapsuleDiggerOption(original *TimeCapsuleDiggerOption, options ...
5857// once TimeCapsuleDigger.Start() is called, and will stop once TimeCapsuleDigger.Stop()
5958// is called.
6059type TimeCapsuleDigger [P any ] struct {
61- logger TimeCapsuleLogger
6260 dataloader Dataloader [P ]
6361 option TimeCapsuleDiggerOption
6462
@@ -85,7 +83,6 @@ type TimeCapsuleDigger[P any] struct {
8583// digInterval: time.Duration
8684func NewDigger [P any ](dataloader Dataloader [P ], digInterval time.Duration , options ... TimeCapsuleDiggerOption ) * TimeCapsuleDigger [P ] {
8785 digger := & TimeCapsuleDigger [P ]{
88- logger : logrus .New (),
8986 dataloader : dataloader ,
9087 option : DefaultTimeCapsuleDiggerOption (),
9188 diggingTicker : time .NewTicker (digInterval ),
@@ -116,7 +113,7 @@ func (t *TimeCapsuleDigger[P]) dig() *TimeCapsule[P] {
116113
117114 dugCapsule , err := t .dataloader .Dig (ctx )
118115 if err != nil {
119- t .logger .Errorf ("[TimeCapsule] failed to dig time capsule from dataloader %v: %v" , t .dataloader .Type (), err )
116+ t .option . Logger .Errorf ("[TimeCapsule] failed to dig time capsule from dataloader %v: %v" , t .dataloader .Type (), err )
120117 return nil
121118 }
122119
@@ -128,9 +125,9 @@ func (t *TimeCapsuleDigger[P]) destroy(capsule *TimeCapsule[P]) {
128125 defer cancel ()
129126
130127 if err := t .dataloader .Destroy (ctx , capsule ); err != nil {
131- t .logger .Errorf ("[TimeCapsule] failed to burn time capsule: %v" , err )
128+ t .option . Logger .Errorf ("[TimeCapsule] failed to burn time capsule: %v" , err )
132129 } else {
133- t .logger .Debugf ("[TimeCapsule] burned a capsule from dataloader %v" , t .dataloader .Type ())
130+ t .option . Logger .Debugf ("[TimeCapsule] burned a capsule from dataloader %v" , t .dataloader .Type ())
134131 }
135132}
136133
@@ -151,7 +148,7 @@ func (t *TimeCapsuleDigger[P]) Start() {
151148 continue
152149 }
153150
154- t .logger .Debugf ("[TimeCapsule] dug a new capsule from dataloader %v" , t .dataloader .Type ())
151+ t .option . Logger .Debugf ("[TimeCapsule] dug a new capsule from dataloader %v" , t .dataloader .Type ())
155152
156153 t .destroy (dugCapsule )
157154 if t .handlerFunc != nil {
0 commit comments