node/db: use provided logger for badgerDb
This commit is contained in:
parent
36fc45d87b
commit
fe09eb5099
|
@ -1,6 +1,7 @@
|
||||||
package db
|
package db
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
|
|
||||||
|
@ -8,8 +9,28 @@ import (
|
||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type badgerZapLogger struct {
|
||||||
|
*zap.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l badgerZapLogger) Errorf(f string, v ...interface{}) {
|
||||||
|
l.Error(fmt.Sprintf(f, v...))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l badgerZapLogger) Warningf(f string, v ...interface{}) {
|
||||||
|
l.Warn(fmt.Sprintf(f, v...))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l badgerZapLogger) Infof(f string, v ...interface{}) {
|
||||||
|
l.Info(fmt.Sprintf(f, v...))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l badgerZapLogger) Debugf(f string, v ...interface{}) {
|
||||||
|
l.Debug(fmt.Sprintf(f, v...))
|
||||||
|
}
|
||||||
|
|
||||||
func OpenDb(logger *zap.Logger, dataDir *string) *Database {
|
func OpenDb(logger *zap.Logger, dataDir *string) *Database {
|
||||||
var options badger.Options
|
options := badger.DefaultOptions(dbPath)
|
||||||
|
|
||||||
if dataDir != nil {
|
if dataDir != nil {
|
||||||
dbPath := path.Join(*dataDir, "db")
|
dbPath := path.Join(*dataDir, "db")
|
||||||
|
@ -22,6 +43,8 @@ func OpenDb(logger *zap.Logger, dataDir *string) *Database {
|
||||||
options = badger.DefaultOptions("").WithInMemory(true)
|
options = badger.DefaultOptions("").WithInMemory(true)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
options = options.WithLogger(badgerZapLogger{logger})
|
||||||
|
|
||||||
db, err := badger.Open(options)
|
db, err := badger.Open(options)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatal("failed to open database", zap.Error(err))
|
logger.Fatal("failed to open database", zap.Error(err))
|
||||||
|
|
Loading…
Reference in New Issue