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