1 回答

TA貢獻1934條經驗 獲得超2個贊
在您的 實現中TagRPC,您可以創建一個結構并將指向它的指針添加到上下文中。然后通過對 的連續調用在其中添加信息HandleRPC。因此,如果您需要 Payload 中僅在*stats.InPayload調用中可用的內容,您可以將其拉出并將其存儲在添加到上下文的結構中,然后在HandleRPC再次調用時訪問它*stats.End
type recorderCtxKey struct{}
type recorder struct {
size int64
}
func (sl *statsHandler) TagRPC(ctx context.Context, info *stats.RPCTagInfo) context.Context {
return context.WithValue(ctx, rpcStatCtxKey{}, &recorder{})
}
func (h *statsHandler) HandleRPC(ctx context.Context, rpcStats stats.RPCStats) {
switch stat := rpcStats.(type) {
case *stats.InPayload:
r, _ := ctx.Value(recorderContextKey{}).(*Recorder)
r.size += stat.WireLength
case *stats.End:
durationMs := stat.EndTime.Sub(stat.BeginTime).Seconds() * 1000.0
r, _ := ctx.Value(recorderContextKey{}).(*Recorder)
# use r.size #
}
}
- 1 回答
- 0 關注
- 84 瀏覽
添加回答
舉報