From 8183e4fb633d231d39404a5fa344d0d180da19e3 Mon Sep 17 00:00:00 2001 From: Sergey Yakubov <sergey.yakubov@desy.de> Date: Tue, 16 Nov 2021 10:59:51 +0100 Subject: [PATCH] logfields for golang logger --- common/go/src/asapo_common/logger/logger.go | 1 + common/go/src/asapo_common/logger/logrus_logger.go | 7 +++++++ common/go/src/asapo_common/logger/mock_logger.go | 5 +++++ 3 files changed, 13 insertions(+) diff --git a/common/go/src/asapo_common/logger/logger.go b/common/go/src/asapo_common/logger/logger.go index 0f026aa18..ffa2c4ce8 100644 --- a/common/go/src/asapo_common/logger/logger.go +++ b/common/go/src/asapo_common/logger/logger.go @@ -17,6 +17,7 @@ const ( ) type Logger interface { + WithFields(args map[string]interface{}) Logger Info(args ...interface{}) Debug(args ...interface{}) Fatal(args ...interface{}) diff --git a/common/go/src/asapo_common/logger/logrus_logger.go b/common/go/src/asapo_common/logger/logrus_logger.go index 4625f2749..d41432252 100644 --- a/common/go/src/asapo_common/logger/logrus_logger.go +++ b/common/go/src/asapo_common/logger/logrus_logger.go @@ -13,6 +13,13 @@ func (l *logRusLogger) SetSource(source string) { l.source = source } + +func (l *logRusLogger) WithFields(args map[string]interface{}) Logger { + l.logger_entry = l.entry().WithFields(args) + return l +} + + func (l *logRusLogger) entry() *log.Entry { if l.logger_entry != nil { return l.logger_entry diff --git a/common/go/src/asapo_common/logger/mock_logger.go b/common/go/src/asapo_common/logger/mock_logger.go index 484b86cb0..58c1df744 100644 --- a/common/go/src/asapo_common/logger/mock_logger.go +++ b/common/go/src/asapo_common/logger/mock_logger.go @@ -16,6 +16,11 @@ func SetMockLog() { my_logger = &MockLog } +func (l *MockLogger) WithFields(args map[string]interface{}) Logger { + l.Called(args) + return l +} + func UnsetMockLog() { my_logger = &logRusLogger{} } -- GitLab