👔 up: update log write logic, add more unit tests

issues_test.go

@@ -36,7 +36,7 @@ func TestIssues_31(t *testing.T) {
 	h2 := handler.MustFileHandler("testdata/info_issue31.log", handler.WithLogLevels(infoLevels))
 
 	slog.PushHandler(h1)
-	slog.PushHandler(h2)
+	slog.PushHandlers(h2)
 
 	// add logs
 	slog.Info("info message text")

logger.go

@@ -81,7 +81,7 @@ func NewWithName(name string) *Logger {
 		exitHandlers: []func(){},
 		// options
 		ReportCaller: true,
-		CallerSkip:   7,
+		CallerSkip:   6,
 		TimeClock:    DefaultClockFn,
 	}
 

logger_test.go

@@ -2,6 +2,7 @@ package slog_test
 
 import (
 	"bytes"
+	"context"
 	"errors"
 	"testing"
 
@@ -82,6 +83,22 @@ func TestLogger_Log(t *testing.T) {
 	l.WithTime(timex.NowHourStart()).Panicln("a panic message")
 }
 
+func TestLogger_WithContext(t *testing.T) {
+	var buf bytes.Buffer
+	h := handler.NewIOWriterHandler(&buf, slog.AllLevels)
+
+	l := newLogger()
+	l.AddHandlers(h)
+
+	ctx := context.Background()
+
+	r := l.WithCtx(ctx)
+	r.Info("with context")
+
+	str := buf.String()
+	assert.Contains(t, str, `with context`)
+}
+
 func TestLogger_panic(t *testing.T) {
 	h := newTestHandler()
 	h.errOnFlush = true
@@ -160,6 +177,13 @@ func TestLogger_logf_allLevel(t *testing.T) {
 	printfAllLevelLogs(l, "this a log %s", "message")
 }
 
+func newLogger() *slog.Logger {
+	return slog.NewWithConfig(func(l *slog.Logger) {
+		l.ReportCaller = true
+		l.DoNothingOnPanicFatal()
+	})
+}
+
 func printAllLevelLogs(l gsr.Logger, args ...interface{}) {
 	l.Debug(args...)
 	l.Info(args...)

processor_test.go

@@ -37,7 +37,7 @@ func TestLogger_AddProcessor(t *testing.T) {
 	assert.Contains(t, str, `"memoryUsage":`)
 
 	l.ResetProcessors()
-	l.PushProcessor(slog.AddUniqueID("requestId"))
+	l.SetProcessors([]slog.Processor{slog.AddUniqueID("requestId")})
 	l.Info("message3")
 	str = buf.String()
 	buf.Reset()

record.go

@@ -293,15 +293,19 @@ func (r *Record) SetFields(fields M) *Record {
 //
 
 func (r *Record) log(level Level, args []any) {
+	r.Level = level
 	// will reduce memory allocation once
 	// r.Message = strutil.Byte2str(formatArgsWithSpaces(args))
 	r.Message = formatArgsWithSpaces(args)
-	r.logWrite(level)
+	// r.logWrite(level)
+	r.logger.writeRecord(level, r)
 }
 
 func (r *Record) logf(level Level, format string, args []any) {
+	r.Level = level
 	r.Message = fmt.Sprintf(format, args...)
-	r.logWrite(level)
+	// r.logWrite(level)
+	r.logger.writeRecord(level, r)
 }
 
 // Log a message with level

sugared.go

@@ -93,8 +93,7 @@ func (sl *SugaredLogger) Close() error {
 	_ = sl.Logger.VisitAll(func(handler Handler) error {
 		// TIP: must exclude self
 		if _, ok := handler.(*SugaredLogger); !ok {
-			err := handler.Close()
-			if err != nil {
+			if err := handler.Close(); err != nil {
 				sl.err = err
 			}
 		}

write.go

@@ -6,21 +6,20 @@ package slog
 // ---------------------------------------------------------------------------
 //
 
-func (r *Record) logWrite(level Level) {
-	// Will reduce memory allocation once
-	// r.Message = strutil.Byte2str(message)
-
-	// var buf *bytes.Buffer
-	// buf = bufferPool.Get().(*bytes.Buffer)
-	// defer bufferPool.Put(buf)
-	// r.Buffer = buf
-
-	// TODO release on here ??
-	// defer r.logger.releaseRecord(r)
-	r.Level = level
-	r.logger.writeRecord(level, r)
-	// r.Buffer = nil
-}
+// func (r *Record) logWrite(level Level) {
+// Will reduce memory allocation once
+// r.Message = strutil.Byte2str(message)
+
+// var buf *bytes.Buffer
+// buf = bufferPool.Get().(*bytes.Buffer)
+// defer bufferPool.Put(buf)
+// r.Buffer = buf
+
+// TODO release on here ??
+// defer r.logger.releaseRecord(r)
+// r.logger.writeRecord(level, r)
+// r.Buffer = nil
+// }
 
 // Init something for record.
 func (r *Record) Init(lowerLevelName bool) {