fix: cannot call SetPrimaryCore after using a Tee logger

Signed-off-by: Manfred Touron <[email protected]>

Author: moul

core.go

@@ -1,6 +1,7 @@
 package log
 
 import (
+	"reflect"
 	"sync"
 
 	"go.uber.org/multierr"
@@ -94,7 +95,7 @@ func (l *lockedMultiCore) ReplaceCore(original, replacement zapcore.Core) {
 	defer l.mu.Unlock()
 
 	for i := 0; i < len(l.cores); i++ {
-		if l.cores[i] == original {
+		if reflect.DeepEqual(l.cores[i], original) {
 			l.cores[i] = replacement
 		}
 	}

setup_test.go

@@ -7,6 +7,9 @@ import (
 	"os"
 	"strings"
 	"testing"
+
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
 )
 
 func TestGetLoggerDefault(t *testing.T) {
@@ -240,3 +243,18 @@ func TestCustomCore(t *testing.T) {
 		t.Errorf("got %q, wanted it to contain log output", buf2.String())
 	}
 }
+
+func TestTeeCore(t *testing.T) {
+	// configure to use a tee logger
+	tee := zap.New(zapcore.NewTee(
+		zap.NewNop().Core(),
+		zap.NewNop().Core(),
+	), zap.AddCaller())
+	SetPrimaryCore(tee.Core())
+	log := getLogger("test")
+	log.Error("scooby")
+
+	// replaces the tee logger with a simple one
+	SetPrimaryCore(zap.NewNop().Core())
+	log.Error("doo")
+}