AddCallerSkip in RedirectStdLog (#320)

global.go

@@ -59,11 +59,17 @@ func ReplaceGlobals(logger *Logger) func() {
 // It returns a function to restore the original prefix and flags and reset the
 // standard library's output to os.Stdout.
 func RedirectStdLog(l *Logger) func() {
+	const (
+		stdLogDefaultDepth = 4
+		loggerWriterDepth  = 1
+	)
 	flags := log.Flags()
 	prefix := log.Prefix()
 	log.SetFlags(0)
 	log.SetPrefix("")
-	log.SetOutput(&loggerWriter{l})
+	log.SetOutput(&loggerWriter{l.WithOptions(
+		AddCallerSkip(stdLogDefaultDepth + loggerWriterDepth),
+	)})
 	return func() {
 		log.SetFlags(flags)
 		log.SetPrefix(prefix)

global_test.go

@@ -25,6 +25,7 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 
 	"go.uber.org/zap/internal/observer"
 	"go.uber.org/zap/zapcore"
@@ -76,3 +77,13 @@ func TestRedirectStdLog(t *testing.T) {
 	assert.Equal(t, initialFlags, log.Flags(), "Expected to reset initial flags.")
 	assert.Equal(t, initialPrefix, log.Prefix(), "Expected to reset initial prefix.")
 }
+
+func TestRedirectStdLogCaller(t *testing.T) {
+	withLogger(t, DebugLevel, []Option{AddCaller()}, func(l *Logger, logs *observer.ObservedLogs) {
+		defer RedirectStdLog(l)()
+		log.Print("redirected")
+		entries := logs.All()
+		require.Len(t, entries, 1, "Unexpected number of logs.")
+		assert.Contains(t, entries[0].Entry.Caller.File, "global_test.go", "Unexpected caller annotation.")
+	})
+}