You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
3.0 KiB
Go
99 lines
3.0 KiB
Go
package serverinterceptors
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/zeromicro/go-zero/core/collection"
|
|
"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"
|
|
"github.com/zeromicro/go-zero/core/timex"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/peer"
|
|
)
|
|
|
|
const defaultSlowThreshold = time.Millisecond * 500
|
|
|
|
var (
|
|
notLoggingContentMethods sync.Map
|
|
slowThreshold = syncx.ForAtomicDuration(defaultSlowThreshold)
|
|
)
|
|
|
|
// StatConf defines the static configuration for stat interceptor.
|
|
type StatConf struct {
|
|
SlowThreshold time.Duration
|
|
NotLoggingContentMethods []string
|
|
}
|
|
|
|
// DontLogContentForMethod disable logging content for given method.
|
|
// Deprecated: use StatConf instead.
|
|
func DontLogContentForMethod(method string) {
|
|
notLoggingContentMethods.Store(method, lang.Placeholder)
|
|
}
|
|
|
|
// SetSlowThreshold sets the slow threshold.
|
|
// Deprecated: use StatConf instead.
|
|
func SetSlowThreshold(threshold time.Duration) {
|
|
slowThreshold.Set(threshold)
|
|
}
|
|
|
|
// UnaryStatInterceptor returns a func that uses given metrics to report stats.
|
|
func UnaryStatInterceptor(metrics *stat.Metrics, conf StatConf) grpc.UnaryServerInterceptor {
|
|
staticNotLoggingContentMethods := collection.NewSet()
|
|
staticNotLoggingContentMethods.AddStr(conf.NotLoggingContentMethods...)
|
|
|
|
return func(ctx context.Context, req any, info *grpc.UnaryServerInfo,
|
|
handler grpc.UnaryHandler) (resp any, err error) {
|
|
startTime := timex.Now()
|
|
defer func() {
|
|
duration := timex.Since(startTime)
|
|
metrics.Add(stat.Task{
|
|
Duration: duration,
|
|
})
|
|
logDuration(ctx, info.FullMethod, req, duration,
|
|
staticNotLoggingContentMethods, conf.SlowThreshold)
|
|
}()
|
|
|
|
return handler(ctx, req)
|
|
}
|
|
}
|
|
|
|
func logDuration(ctx context.Context, method string, req any, duration time.Duration,
|
|
staticNotLoggingContentMethods *collection.Set, staticSlowThreshold time.Duration) {
|
|
var addr string
|
|
client, ok := peer.FromContext(ctx)
|
|
if ok {
|
|
addr = client.Addr.String()
|
|
}
|
|
|
|
logger := logx.WithContext(ctx).WithDuration(duration)
|
|
if !shouldLogContent(method, staticNotLoggingContentMethods) {
|
|
if isSlow(duration, staticSlowThreshold) {
|
|
logger.Slowf("[RPC] slowcall - %s - %s", addr, method)
|
|
}
|
|
} else {
|
|
content, err := json.Marshal(req)
|
|
if err != nil {
|
|
logx.WithContext(ctx).Errorf("%s - %s", addr, err.Error())
|
|
} else if duration > slowThreshold.Load() {
|
|
logger.Slowf("[RPC] slowcall - %s - %s - %s", addr, method, string(content))
|
|
} else {
|
|
logger.Infof("%s - %s - %s", addr, method, string(content))
|
|
}
|
|
}
|
|
}
|
|
|
|
func shouldLogContent(method string, staticNotLoggingContentMethods *collection.Set) bool {
|
|
_, ok := notLoggingContentMethods.Load(method)
|
|
return !ok && !staticNotLoggingContentMethods.Contains(method)
|
|
}
|
|
|
|
func isSlow(duration time.Duration, staticSlowThreshold time.Duration) bool {
|
|
return duration > slowThreshold.Load() ||
|
|
(staticSlowThreshold > 0 && duration > staticSlowThreshold)
|
|
}
|