Adapt groupable to logsink

cisco-open · Jan 18, 2022 · 8ecf704 · 8ecf704
1 parent 7edb7e7
commit 8ecf704
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/pkg/logger/logger.go b/pkg/logger/logger.go
@@ -141,9 +141,17 @@ func (log *logger) Grouped(state bool) {
 }
 
 func EnableGroupSession(logger interface{}) func() {
-	if l, ok := logger.(interface{ Grouped(state bool) }); ok {
-		l.Grouped(true)
-		return func() { l.Grouped(false) }
+	var l interface{}
+
+	if lwrap, ok := logger.(interface{ GetSink() logr.LogSink }); ok {
+		l = lwrap.GetSink()
+	} else {
+		l = logger
+	}
+
+	if lgroupable, ok := l.(interface{ Grouped(state bool) }); ok {
+		lgroupable.Grouped(true)
+		return func() { lgroupable.Grouped(false) }
 	}
 	return func() {}
 }