Instrumentation: Report the size of plugin request (#66149)

* Report the size of the plugin request

* Remove middleware, report directly

* PR review updates
This commit is contained in:
ismail simsek
2023-04-11 22:04:32 +02:00
committed by GitHub
parent 1791c6043f
commit a1bc227228
2 changed files with 34 additions and 8 deletions

View File

@ -40,11 +40,16 @@ func (s *Service) QueryData(ctx context.Context, req *backend.QueryDataRequest)
return nil, plugins.ErrPluginNotRegistered.Errorf("%w", backendplugin.ErrPluginNotRegistered)
}
var totalBytes float64
for _, v := range req.Queries {
totalBytes += float64(len(v.JSON))
}
var resp *backend.QueryDataResponse
err := instrumentation.InstrumentQueryDataRequest(ctx, &req.PluginContext, instrumentation.Cfg{
LogDatasourceRequests: s.cfg.LogDatasourceRequests,
Target: p.Target(),
}, func() (innerErr error) {
}, totalBytes, func() (innerErr error) {
resp, innerErr = p.QueryData(ctx, req)
return
})
@ -86,10 +91,12 @@ func (s *Service) CallResource(ctx context.Context, req *backend.CallResourceReq
if !exists {
return backendplugin.ErrPluginNotRegistered
}
totalBytes := float64(len(req.Body))
err := instrumentation.InstrumentCallResourceRequest(ctx, &req.PluginContext, instrumentation.Cfg{
LogDatasourceRequests: s.cfg.LogDatasourceRequests,
Target: p.Target(),
}, func() error {
}, totalBytes, func() error {
removeConnectionHeaders(req.Headers)
removeHopByHopHeaders(req.Headers)