refactor(assert): improve error logging with stack traces

Author: kooksee

assert/must_test.go

@@ -2,13 +2,20 @@ package assert_test
 
 import (
 	"fmt"
+	"log/slog"
 	"testing"
 
+	"github.com/stretchr/testify/assert"
+
 	assert1 "github.com/pubgo/funk/assert"
 	"github.com/pubgo/funk/errors"
-	"github.com/stretchr/testify/assert"
+	"github.com/pubgo/funk/log"
 )
 
+func init() {
+	slog.SetDefault(slog.New(log.NewSlog(log.GetLogger("fastcommit"))))
+}
+
 type errBase struct {
 	msg string
 }

assert/util.go

@@ -1,7 +1,6 @@
 package assert
 
 import (
-	"encoding/base64"
 	"fmt"
 	"log/slog"
 	"reflect"
@@ -14,6 +13,7 @@ import (
 	"google.golang.org/protobuf/proto"
 
 	"github.com/pubgo/funk/log/logfields"
+	"github.com/pubgo/funk/stack"
 )
 
 func messageFromMsgAndArgs(msgAndArgs ...any) string {
@@ -30,15 +30,20 @@ func messageFromMsgAndArgs(msgAndArgs ...any) string {
 	return pretty().Sprint(msgAndArgs...)
 }
 
+var assetFile = stack.Caller(0)
+
 func logErr(err error, message string, attrs ...slog.Attr) {
 	if err == nil {
 		return
 	}
 
+	traces := lo.Filter(stack.Trace(), func(item *stack.Frame, index int) bool {
+		return !item.IsRuntime() && item.Pkg != assetFile.Pkg
+	})
 	attrs = append(attrs,
 		slog.String(logfields.Module, "assert"),
 		slog.String(logfields.Error, err.Error()),
-		slog.String(logfields.ErrorStack, base64.StdEncoding.EncodeToString(debug.Stack())),
+		slog.Any(logfields.ErrorStack, lo.Map(traces, func(item *stack.Frame, index int) string { return item.String() })),
 		slog.String(logfields.ErrorDetail, fmt.Sprintf("%v", err)),
 	)
 	slog.Error(message, lo.ToAnySlice(attrs)...)

stack/trace_test.go

@@ -5,9 +5,13 @@ import (
 
 	"github.com/pubgo/funk/pretty"
 	"github.com/pubgo/funk/stack"
+	"github.com/samber/lo"
 )
 
 func TestTrace(t *testing.T) {
 	traces := stack.Trace()
 	t.Log(pretty.Sprint(traces))
+
+	traces = lo.Filter(traces, func(item *stack.Frame, index int) bool { return !item.IsRuntime() })
+	t.Log(pretty.Sprint(lo.Map(traces, func(item *stack.Frame, index int) string { return item.String() })))
 }