You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
998 B
36 lines
998 B
2 years ago
|
package prome
|
||
|
2 years ago
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"Open_IM/pkg/common/log"
|
||
|
|
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
"google.golang.org/grpc/peer"
|
||
|
|
)
|
||
|
|
|
||
|
|
func UnaryServerInterceptorProme(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
|
||
|
|
remote, _ := peer.FromContext(ctx)
|
||
|
|
remoteAddr := remote.Addr.String()
|
||
|
|
|
||
|
|
in, _ := json.Marshal(req)
|
||
|
|
inStr := string(in)
|
||
|
|
log.NewInfo("ip", remoteAddr, "access_start", info.FullMethod, "in", inStr)
|
||
|
|
|
||
|
|
start := time.Now()
|
||
|
|
defer func() {
|
||
|
|
out, _ := json.Marshal(resp)
|
||
|
|
outStr := string(out)
|
||
|
|
duration := int64(time.Since(start) / time.Millisecond)
|
||
|
|
if duration >= 500 {
|
||
|
|
log.NewInfo("ip", remoteAddr, "access_end", info.FullMethod, "in", inStr, "out", outStr, "err", err, "duration/ms", duration)
|
||
|
|
} else {
|
||
|
|
log.NewInfo("ip", remoteAddr, "access_end", info.FullMethod, "in", inStr, "out", outStr, "err", err, "duration/ms", duration)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
resp, err = handler(ctx, req)
|
||
|
|
return
|
||
|
|
}
|