Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,64 @@ | ||
| package tracer | ||
|
|
||
| import ( | ||
| "context" | ||
| "log/slog" | ||
| "strings" | ||
|
|
||
| "gopkg.in/DataDog/dd-trace-go.v1/internal/log" | ||
| ) | ||
|
|
||
| // slogHandler implements the slog.Handler interface to dispatch messages to our | ||
| // internal logger. | ||
| type slogHandler struct { | ||
| attrs []string | ||
| groups []string | ||
| } | ||
|
|
||
| func (h slogHandler) Enabled(ctx context.Context, lvl slog.Level) bool { | ||
| if lvl <= slog.LevelDebug { | ||
| return log.DebugEnabled() | ||
| } | ||
| // TODO(fg): Implement generic log level checking in the internal logger. | ||
| // But we're we're not concerned with slog perf, so this is okay for now. | ||
| return true | ||
| } | ||
|
|
||
| func (h slogHandler) Handle(ctx context.Context, r slog.Record) error { | ||
| parts := make([]string, 0, 1+len(h.attrs)+r.NumAttrs()) | ||
| parts = append(parts, r.Message) | ||
| parts = append(parts, h.attrs...) | ||
| r.Attrs(func(a slog.Attr) bool { | ||
| parts = append(parts, formatAttr(a, h.groups)) | ||
| return true | ||
| }) | ||
|
|
||
| msg := strings.Join(parts, " ") | ||
| switch r.Level { | ||
| case slog.LevelDebug: | ||
| log.Debug(msg) | ||
| case slog.LevelInfo: | ||
| log.Info(msg) | ||
| case slog.LevelWarn: | ||
| log.Warn(msg) | ||
| case slog.LevelError: | ||
| log.Error(msg) | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (h slogHandler) WithAttrs(attrs []slog.Attr) slog.Handler { | ||
| for _, a := range attrs { | ||
| h.attrs = append(h.attrs, formatAttr(a, h.groups)) | ||
| } | ||
| return h | ||
| } | ||
|
|
||
| func (h slogHandler) WithGroup(name string) slog.Handler { | ||
| h.groups = append(h.groups, name) | ||
| return h | ||
| } | ||
|
|
||
| func formatAttr(a slog.Attr, groups []string) string { | ||
| return strings.Join(append(groups, a.String()), ".") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,33 @@ | ||
| package tracer | ||
|
|
||
| import ( | ||
| "log/slog" | ||
| "testing" | ||
|
|
||
| "github.com/stretchr/testify/require" | ||
| "gopkg.in/DataDog/dd-trace-go.v1/internal/log" | ||
| ) | ||
|
|
||
| func Test_slogHandler(t *testing.T) { | ||
| // Create a record logger to capture the logs and restore the original | ||
| // logger at the end. | ||
| rl := &log.RecordLogger{} | ||
| defer log.UseLogger(rl)() | ||
|
|
||
| // Log a few messages at different levels. The debug message gets discarded | ||
| // because the internal logger does not have debug enabled by default. | ||
| l := slog.New(slogHandler{}) | ||
| l = l.With("foo", "bar") | ||
| l = l.WithGroup("a").WithGroup("b") | ||
| l.Debug("debug test", "n", 0) | ||
| l.Info("info test", "n", 1) | ||
| l.Warn("warn test", "n", 2) | ||
| l.Error("error test", "n", 3) | ||
| log.Flush() // needed to get the error log flushed | ||
|
|
||
| // Check that the logs were written correctly. | ||
| require.Len(t, rl.Logs(), 3) | ||
| require.Contains(t, rl.Logs()[0], "info test foo=bar a.b.n=1") | ||
| require.Contains(t, rl.Logs()[1], "warn test foo=bar a.b.n=2") | ||
| require.Contains(t, rl.Logs()[2], "error test foo=bar a.b.n=3") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters