fix: accesslog

Author: sysulq

cmd/kod/internal/generate_generator.go

@@ -810,7 +810,8 @@ func (g *generator) generateLocalStubs(p printFn) {
 				}
 			`, g.returnsList(mt), m.Name(), g.argList(comp, mt), g.setReturnsList(mt))
 
-			p(`info := kod.CallInfo{
+			p(`info := kod.CallInfo {
+					Impl: s.impl,
 					Component:  s.name,
 					FullMethod: "%s.%s",
 					Method:    "%s",

examples/helloworld/kod_gen.go

@@ -4,6 +4,8 @@ import "context"
 
 // CallInfo contains information about the call.
 type CallInfo struct {
+	// The impl of the called component.
+	Impl any
 	// The component name of the called method.
 	Component string
 	// The full name of the called method, in the format of "package/service.method".

interceptor.go

@@ -25,7 +25,12 @@ func Interceptor() kod.Interceptor {
 			attrs = append(attrs, slog.String("error", err.Error()))
 		}
 
-		kod.FromContext(ctx).L(ctx).LogAttrs(ctx, level, "accesslog", attrs...)
+		// check if impl L(ctx context.Context) method
+		if l, ok := info.Impl.(interface {
+			L(ctx context.Context) *slog.Logger
+		}); ok {
+			l.L(ctx).LogAttrs(ctx, level, "accesslog", attrs...)
+		}
 
 		return err
 	}

interceptor/kaccesslog/accesslog.go

@@ -166,9 +166,13 @@ func TestPanicKod(t *testing.T) {
 }
 
 func BenchmarkCase1(b *testing.B) {
-	kod.RunTest(b, func(ctx context.Context, k *test1Component) {
-		_, err := k.Foo(ctx, &FooReq{})
-		assert.Equal(b, "B", k.Config().A)
-		assert.Equal(b, "test1:B", err.Error())
+	b.Run("case1", func(b *testing.B) {
+		kod.RunTest(b, func(ctx context.Context, k *test1Component) {
+			for i := 0; i < b.N; i++ {
+				_, err := k.Foo(ctx, &FooReq{})
+				assert.Equal(b, "B", k.Config().A)
+				assert.Equal(b, "test1:B", err.Error())
+			}
+		})
 	})
 }