diff --git a/core/jsonx/json.go b/core/jsonx/json.go index 34cf92ea..abe21340 100644 --- a/core/jsonx/json.go +++ b/core/jsonx/json.go @@ -13,6 +13,16 @@ func Marshal(v interface{}) ([]byte, error) { return json.Marshal(v) } +// MarshalToString marshals v into a string. +func MarshalToString(v interface{}) (string, error) { + data, err := Marshal(v) + if err != nil { + return "", err + } + + return string(data), nil +} + // Unmarshal unmarshals data bytes into v. func Unmarshal(data []byte, v interface{}) error { decoder := json.NewDecoder(bytes.NewReader(data)) @@ -53,12 +63,3 @@ func unmarshalUseNumber(decoder *json.Decoder, v interface{}) error { func formatError(v string, err error) error { return fmt.Errorf("string: `%s`, error: `%w`", v, err) } - -// MarshalToString marshals v into string. -func MarshalToString(v interface{}) (string, error) { - data, err := Marshal(v) - if err != nil { - return "", formatError(string(data), err) - } - return string(data), nil -} diff --git a/core/jsonx/json_test.go b/core/jsonx/json_test.go index d11ce992..016acf20 100644 --- a/core/jsonx/json_test.go +++ b/core/jsonx/json_test.go @@ -20,6 +20,22 @@ func TestMarshal(t *testing.T) { assert.Equal(t, `{"name":"John","age":30}`, string(bs)) } +func TestMarshalToString(t *testing.T) { + var v = struct { + Name string `json:"name"` + Age int `json:"age"` + }{ + Name: "John", + Age: 30, + } + toString, err := MarshalToString(v) + assert.Nil(t, err) + assert.Equal(t, `{"name":"John","age":30}`, toString) + + _, err = MarshalToString(make(chan int)) + assert.NotNil(t, err) +} + func TestUnmarshal(t *testing.T) { const s = `{"name":"John","age":30}` var v struct { @@ -85,16 +101,3 @@ func TestUnmarshalFromReaderError(t *testing.T) { err := UnmarshalFromReader(strings.NewReader(s), &v) assert.NotNil(t, err) } - -func TestMarshalToString(t *testing.T) { - var v = struct { - Name string `json:"name"` - Age int `json:"age"` - }{ - Name: "John", - Age: 30, - } - toString, err := MarshalToString(v) - assert.Nil(t, err) - assert.Equal(t, `{"name":"John","age":30}`, toString) -} diff --git a/go.mod b/go.mod index 1872ef8e..34e65e4d 100644 --- a/go.mod +++ b/go.mod @@ -40,7 +40,7 @@ require ( k8s.io/api v0.22.9 k8s.io/apimachinery v0.22.9 k8s.io/client-go v0.22.9 - k8s.io/utils v0.0.0-20220210201930-3a6ce19ff2f9 + k8s.io/utils v0.0.0-20220706174534-f6158b442e7c ) require ( diff --git a/go.sum b/go.sum index 41959d60..60401c1d 100644 --- a/go.sum +++ b/go.sum @@ -868,6 +868,8 @@ k8s.io/kube-openapi v0.0.0-20211109043538-20434351676c/go.mod h1:vHXdDvt9+2spS2R k8s.io/utils v0.0.0-20211116205334-6203023598ed/go.mod h1:jPW/WVKK9YHAvNhRxK0md/EJ228hCsBRufyofKtW8HA= k8s.io/utils v0.0.0-20220210201930-3a6ce19ff2f9 h1:HNSDgDCrr/6Ly3WEGKZftiE7IY19Vz2GdbOCyI4qqhc= k8s.io/utils v0.0.0-20220210201930-3a6ce19ff2f9/go.mod h1:jPW/WVKK9YHAvNhRxK0md/EJ228hCsBRufyofKtW8HA= +k8s.io/utils v0.0.0-20220706174534-f6158b442e7c h1:hFZO68mv/0xe8+V0gRT9BAq3/31cKjjeVv4nScriuBk= +k8s.io/utils v0.0.0-20220706174534-f6158b442e7c/go.mod h1:jPW/WVKK9YHAvNhRxK0md/EJ228hCsBRufyofKtW8HA= rsc.io/binaryregexp v0.2.0/go.mod h1:qTv7/COck+e2FymRvadv62gMdZztPaShugOCi3I+8D8= rsc.io/quote/v3 v3.1.0/go.mod h1:yEA65RcK8LyAZtP9Kv3t0HmxON59tX3rD+tICJqUlj0= rsc.io/sampler v1.3.0/go.mod h1:T1hPZKmBbMNahiBKFy5HrXp6adAjACjK9JXDnKaTXpA= diff --git a/zrpc/internal/serverinterceptors/statinterceptor.go b/zrpc/internal/serverinterceptors/statinterceptor.go index 16302c14..a0997878 100644 --- a/zrpc/internal/serverinterceptors/statinterceptor.go +++ b/zrpc/internal/serverinterceptors/statinterceptor.go @@ -3,8 +3,10 @@ package serverinterceptors import ( "context" "encoding/json" + "sync" "time" + "github.com/zeromicro/go-zero/core/lang" "github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/stat" "github.com/zeromicro/go-zero/core/syncx" @@ -15,7 +17,15 @@ import ( const defaultSlowThreshold = time.Millisecond * 500 -var slowThreshold = syncx.ForAtomicDuration(defaultSlowThreshold) +var ( + notLoggingContentMethods sync.Map + slowThreshold = syncx.ForAtomicDuration(defaultSlowThreshold) +) + +// DontLogContentForMethod disable logging content for given method. +func DontLogContentForMethod(method string) { + notLoggingContentMethods.Store(method, lang.Placeholder) +} // SetSlowThreshold sets the slow threshold. func SetSlowThreshold(threshold time.Duration) { @@ -45,13 +55,24 @@ func logDuration(ctx context.Context, method string, req interface{}, duration t if ok { addr = client.Addr.String() } + + logger := logx.WithContext(ctx).WithDuration(duration) + _, ok = notLoggingContentMethods.Load(method) + if ok { + if duration > slowThreshold.Load() { + logger.Slowf("[RPC] slowcall - %s - %s - %s", addr, method) + } else { + logger.Infof("%s - %s - %s", addr, method) + } + } + content, err := json.Marshal(req) if err != nil { logx.WithContext(ctx).Errorf("%s - %s", addr, err.Error()) } else if duration > slowThreshold.Load() { - logx.WithContext(ctx).WithDuration(duration).Slowf("[RPC] slowcall - %s - %s - %s", + logger.Slowf("[RPC] slowcall - %s - %s - %s", addr, method, string(content)) } else { - logx.WithContext(ctx).WithDuration(duration).Infof("%s - %s - %s", addr, method, string(content)) + logger.Infof("%s - %s - %s", addr, method, string(content)) } } diff --git a/zrpc/internal/serverinterceptors/statinterceptor_test.go b/zrpc/internal/serverinterceptors/statinterceptor_test.go index b9a4706c..c4545851 100644 --- a/zrpc/internal/serverinterceptors/statinterceptor_test.go +++ b/zrpc/internal/serverinterceptors/statinterceptor_test.go @@ -83,3 +83,58 @@ func TestLogDuration(t *testing.T) { }) } } + +func TestLogDurationWithoutContent(t *testing.T) { + addrs, err := net.InterfaceAddrs() + assert.Nil(t, err) + assert.True(t, len(addrs) > 0) + + tests := []struct { + name string + ctx context.Context + req interface{} + duration time.Duration + }{ + { + name: "normal", + ctx: context.Background(), + req: "foo", + }, + { + name: "bad req", + ctx: context.Background(), + req: make(chan lang.PlaceholderType), // not marshalable + }, + { + name: "timeout", + ctx: context.Background(), + req: "foo", + duration: time.Second, + }, + { + name: "timeout", + ctx: peer.NewContext(context.Background(), &peer.Peer{ + Addr: addrs[0], + }), + req: "foo", + }, + { + name: "timeout", + ctx: context.Background(), + req: "foo", + duration: slowThreshold.Load() + time.Second, + }, + } + + DontLogContentForMethod("foo") + for _, test := range tests { + test := test + t.Run(test.name, func(t *testing.T) { + t.Parallel() + + assert.NotPanics(t, func() { + logDuration(test.ctx, "foo", test.req, test.duration) + }) + }) + } +} diff --git a/zrpc/server.go b/zrpc/server.go index 1409e6de..c7386184 100644 --- a/zrpc/server.go +++ b/zrpc/server.go @@ -97,6 +97,11 @@ func (rs *RpcServer) Stop() { logx.Close() } +// DontLogContentForMethod disable logging content for given method. +func DontLogContentForMethod(method string) { + serverinterceptors.DontLogContentForMethod(method) +} + // SetServerSlowThreshold sets the slow threshold on server side. func SetServerSlowThreshold(threshold time.Duration) { serverinterceptors.SetSlowThreshold(threshold) diff --git a/zrpc/server_test.go b/zrpc/server_test.go index f8df2aec..67d1c2f9 100644 --- a/zrpc/server_test.go +++ b/zrpc/server_test.go @@ -35,6 +35,7 @@ func TestServer_setupInterceptors(t *testing.T) { } func TestServer(t *testing.T) { + DontLogContentForMethod("foo") SetServerSlowThreshold(time.Second) svr := MustNewServer(RpcServerConf{ ServiceConf: service.ServiceConf{ @@ -121,7 +122,7 @@ type mockedServer struct { streamInterceptors []grpc.StreamServerInterceptor } -func (m *mockedServer) AddOptions(options ...grpc.ServerOption) { +func (m *mockedServer) AddOptions(_ ...grpc.ServerOption) { } func (m *mockedServer) AddStreamInterceptors(interceptors ...grpc.StreamServerInterceptor) { @@ -132,9 +133,9 @@ func (m *mockedServer) AddUnaryInterceptors(interceptors ...grpc.UnaryServerInte m.unaryInterceptors = append(m.unaryInterceptors, interceptors...) } -func (m *mockedServer) SetName(s string) { +func (m *mockedServer) SetName(_ string) { } -func (m *mockedServer) Start(register internal.RegisterFn) error { +func (m *mockedServer) Start(_ internal.RegisterFn) error { return nil }