mirror of
https://github.com/pikami/cosmium.git
synced 2025-12-19 17:00:37 +00:00
Fix issues with persist flag; Use custom logger for badger
This commit is contained in:
@@ -15,6 +15,7 @@ type BadgerDataStoreOptions struct {
|
||||
|
||||
func NewBadgerDataStore(options BadgerDataStoreOptions) *BadgerDataStore {
|
||||
badgerOpts := badger.DefaultOptions(options.PersistDataFilePath)
|
||||
badgerOpts = badgerOpts.WithLogger(newBadgerLogger())
|
||||
if options.PersistDataFilePath == "" {
|
||||
badgerOpts = badgerOpts.WithInMemory(true)
|
||||
}
|
||||
|
||||
28
internal/datastore/badger_datastore/badger_logger.go
Normal file
28
internal/datastore/badger_datastore/badger_logger.go
Normal file
@@ -0,0 +1,28 @@
|
||||
package badgerdatastore
|
||||
|
||||
import (
|
||||
"github.com/dgraph-io/badger/v4"
|
||||
"github.com/pikami/cosmium/internal/logger"
|
||||
)
|
||||
|
||||
type badgerLogger struct{}
|
||||
|
||||
func newBadgerLogger() badger.Logger {
|
||||
return &badgerLogger{}
|
||||
}
|
||||
|
||||
func (l *badgerLogger) Errorf(format string, v ...interface{}) {
|
||||
logger.Errorf(format, v...)
|
||||
}
|
||||
|
||||
func (l *badgerLogger) Warningf(format string, v ...interface{}) {
|
||||
logger.Infof(format, v...)
|
||||
}
|
||||
|
||||
func (l *badgerLogger) Infof(format string, v ...interface{}) {
|
||||
logger.Infof(format, v...)
|
||||
}
|
||||
|
||||
func (l *badgerLogger) Debugf(format string, v ...interface{}) {
|
||||
logger.Debugf(format, v...)
|
||||
}
|
||||
Reference in New Issue
Block a user