我正在使用 stats/HandleRPC() 發(fā)出一些關(guān)于 RPC 持續(xù)時(shí)間的指標(biāo)，當(dāng)我收到 stats/End 數(shù)據(jù)時(shí)，我想用一些可以從傳入和傳出有效負(fù)載中提取的信息來(lái)標(biāo)記這些指標(biāo)。實(shí)現(xiàn)這一目標(biāo)的最佳方法是什么？func (h *myStatsHandler) HandleRPC(ctx context.Context, rpcStats stats.RPCStats) { switch stat := rpcStats.(type) { case *stats.End: durationMs := stat.EndTime.Sub(stat.BeginTime).Seconds() * 1000.0 // Now before sending this value, I need to know, for example the value of a specific key in the request payload, or whether the response is nil or not }}
1 回答
撒科打諢
TA貢獻(xiàn)1934條經(jīng)驗(yàn) 獲得超2個(gè)贊
在您的 實(shí)現(xiàn)中TagRPC，您可以創(chuàng)建一個(gè)結(jié)構(gòu)并將指向它的指針添加到上下文中。然后通過(guò)對(duì) 的連續(xù)調(diào)用在其中添加信息HandleRPC。因此，如果您需要 Payload 中僅在*stats.InPayload調(diào)用中可用的內(nèi)容，您可以將其拉出并將其存儲(chǔ)在添加到上下文的結(jié)構(gòu)中，然后在HandleRPC再次調(diào)用時(shí)訪問(wèn)它*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 關(guān)注
- 90 瀏覽
添加回答
舉報(bào)
0/150
提交
取消