adding logging to server
This commit is contained in:
@@ -1,31 +1,23 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"io"
|
||||
|
||||
logger "github.com/apsdehal/go-logger"
|
||||
"github.com/asdine/storm"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
type DB struct {
|
||||
*storm.DB
|
||||
*logger.Logger
|
||||
*zerolog.Logger
|
||||
}
|
||||
|
||||
// NewDB returns a new database object,
|
||||
// it configures the database for you.
|
||||
func NewDB(dbPath, format string, logLevel int, logWriter io.WriteCloser) (*DB, error) {
|
||||
//Note! Do not use logger as you have no idea if logWriter has been configured for output yet
|
||||
func NewDB(dbPath string, log *zerolog.Logger) (*DB, error) {
|
||||
var db DB
|
||||
log, err := logger.New("db logger", 1, logWriter)
|
||||
if err != nil {
|
||||
return &db, err
|
||||
}
|
||||
log.SetLogLevel(logger.LogLevel(logLevel))
|
||||
log.SetFormat(format)
|
||||
db.Logger = log
|
||||
databaseLogger := log.With().Str("module", "database").Logger() // Setting up a sub logger for the database module
|
||||
db.Logger = &databaseLogger
|
||||
if err := db.ConfigureDB(dbPath); err != nil {
|
||||
log.ErrorF("Error configuring the database ", err)
|
||||
db.Err(err).Msg("unable to configure database")
|
||||
return &db, err
|
||||
}
|
||||
return &db, nil
|
||||
|
||||
Reference in New Issue
Block a user