From f4850b9e7a9d23878d7fc68178e8c8338660139c Mon Sep 17 00:00:00 2001 From: Ori Newman Date: Wed, 18 Sep 2019 11:47:54 +0300 Subject: [PATCH] [NOD-330] Use BTCD logs for gorm (#410) --- apiserver/database/database.go | 8 ++++++++ apiserver/database/log.go | 9 +++++++++ 2 files changed, 17 insertions(+) create mode 100644 apiserver/database/log.go diff --git a/apiserver/database/database.go b/apiserver/database/database.go index b3790e416..55e3d2be9 100644 --- a/apiserver/database/database.go +++ b/apiserver/database/database.go @@ -23,6 +23,13 @@ func DB() (*gorm.DB, error) { return db, nil } +type gormLogger struct{} + +func (l gormLogger) Print(v ...interface{}) { + str := fmt.Sprint(v...) + log.Errorf(str) +} + // Connect connects to the database mentioned in // config variable. func Connect(cfg *config.Config) error { @@ -40,6 +47,7 @@ func Connect(cfg *config.Config) error { if err != nil { return err } + db.SetLogger(gormLogger{}) return nil } diff --git a/apiserver/database/log.go b/apiserver/database/log.go new file mode 100644 index 000000000..05343395a --- /dev/null +++ b/apiserver/database/log.go @@ -0,0 +1,9 @@ +package database + +import "github.com/daglabs/btcd/util/panics" +import "github.com/daglabs/btcd/apiserver/logger" + +var ( + log = logger.BackendLog.Logger("DTBS") + spawn = panics.GoroutineWrapperFunc(log, logger.BackendLog) +)