Skip to content

Commit 55be0bb

Browse files
authored
Add slog logger to BadgerDB (#914)
1 parent 7a95380 commit 55be0bb

1 file changed

Lines changed: 20 additions & 1 deletion

File tree

storage/posix/antispam/badger.go

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,9 @@ func NewAntispam(ctx context.Context, badgerPath string, opts AntispamOpts) (*An
8585
opts.PushbackThreshold = DefaultPushbackThreshold
8686
}
8787

88+
bOpts := badger.DefaultOptions(badgerPath).WithLogger(&slogger{})
8889
// Open the Badger database located at badgerPath, it will be created if it doesn't exist.
89-
db, err := badger.Open(badger.DefaultOptions(badgerPath))
90+
db, err := badger.Open(bOpts)
9091
if err != nil {
9192
return nil, fmt.Errorf("failed to open badger: %v", err)
9293
}
@@ -420,3 +421,21 @@ func (f *follower) EntriesProcessed(ctx context.Context) (uint64, error) {
420421

421422
return nextIdx, err
422423
}
424+
425+
type slogger struct{}
426+
427+
func (s slogger) Errorf(m string, a ...any) {
428+
slog.ErrorContext(context.Background(), fmt.Sprintf("BadgerDB: %s", fmt.Sprintf(m, a...)))
429+
}
430+
431+
func (s slogger) Warningf(m string, a ...any) {
432+
slog.WarnContext(context.Background(), fmt.Sprintf("BadgerDB: %s", fmt.Sprintf(m, a...)))
433+
}
434+
435+
func (s slogger) Infof(m string, a ...any) {
436+
slog.InfoContext(context.Background(), fmt.Sprintf("BadgerDB: %s", fmt.Sprintf(m, a...)))
437+
}
438+
439+
func (s slogger) Debugf(m string, a ...any) {
440+
slog.DebugContext(context.Background(), fmt.Sprintf("BadgerDB: %s", fmt.Sprintf(m, a...)))
441+
}

0 commit comments

Comments
 (0)