planetmint-go/util/logger.go
Julian Strobl 804a8e5be6
Add AppLogger to test runs (#230)
* [log] Do not create a new object of `GetAppLogger()` every time

* [tests] Enable AppLogger during testing

Signed-off-by: Julian Strobl <jmastr@mailbox.org>
2023-12-12 13:39:37 +01:00

64 lines
1.5 KiB
Go

package util
import (
"sync"
"testing"
sdk "github.com/cosmos/cosmos-sdk/types"
)
type AppLogger struct {
testingLogger *testing.T
}
var (
globalApplicationLoggerTag string
appLogger *AppLogger
initAppLogger sync.Once
)
func init() {
// Initialize the package-level variable
globalApplicationLoggerTag = "[app] "
}
func GetAppLogger() *AppLogger {
initAppLogger.Do(func() {
appLogger = &AppLogger{
testingLogger: nil,
}
})
return appLogger
}
func (logger *AppLogger) SetTestingLogger(testingLogger *testing.T) *AppLogger {
logger.testingLogger = testingLogger
return logger
}
func (logger *AppLogger) testingLog(msg string, keyvals ...interface{}) {
if logger.testingLogger == nil {
return
}
if len(keyvals) == 0 {
logger.testingLogger.Log(msg)
return
}
logger.testingLogger.Log(msg, keyvals)
}
func (logger *AppLogger) Info(ctx sdk.Context, msg string, keyvals ...interface{}) {
logger.testingLog(globalApplicationLoggerTag+msg, keyvals...)
ctx.Logger().Info(globalApplicationLoggerTag+msg, keyvals...)
}
func (logger *AppLogger) Debug(ctx sdk.Context, msg string, keyvals ...interface{}) {
logger.testingLog(globalApplicationLoggerTag+msg, keyvals...)
ctx.Logger().Debug(globalApplicationLoggerTag+msg, keyvals...)
}
func (logger *AppLogger) Error(ctx sdk.Context, msg string, keyvals ...interface{}) {
logger.testingLog(globalApplicationLoggerTag+msg, keyvals...)
ctx.Logger().Error(globalApplicationLoggerTag+msg, keyvals...)
}