Skip to content

Commit fe997d9

Browse files
committed
fix: logger incorrectly applied
1 parent 03b03e4 commit fe997d9

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

timecapsule.go

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ package timecapsule
33
import (
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.
6059
type 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
8684
func 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

Comments
 (0)