测量每个流的gRPC带宽

Question

I have a gRPC service written in Go which has very long running streams. I want a way to measure the network/bandwidth usage per stream and make that information available to prometheus.

I found grpc.StreamServerInterceptor but from what I gather, it doesn't give you access to the marshalled message (which I need to figure out the size).

There's this third party middleware I've found, but it doesn't seem to export any network related information.

Answer 1

You can set a stats.Handler on both gRPC servers and clients. You can put whatever labels you need in the context you provide it with.

gRPC will then invoke your handler's HandleRPC method with either a stats.OutPayload or a stats.InPayload, containing the message (raw and encoded), its uncompressed and wire lengths and a timestamp.

It's all in the source code.