mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-23 04:03:49 +00:00
84 lines
2.5 KiB
Go
84 lines
2.5 KiB
Go
package heimdall
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
type (
|
|
requestTypeKey struct{}
|
|
requestType string
|
|
|
|
// TODO: Uncomment once metrics are added
|
|
// meter struct {
|
|
// request map[bool]metrics.Meter // map[isSuccessful]metrics.Meter
|
|
// timer metrics.Timer
|
|
// }
|
|
)
|
|
|
|
const (
|
|
stateSyncRequest requestType = "state-sync"
|
|
spanRequest requestType = "span"
|
|
checkpointRequest requestType = "checkpoint"
|
|
checkpointCountRequest requestType = "checkpoint-count"
|
|
)
|
|
|
|
func withRequestType(ctx context.Context, reqType requestType) context.Context {
|
|
return context.WithValue(ctx, requestTypeKey{}, reqType)
|
|
}
|
|
|
|
func getRequestType(ctx context.Context) (requestType, bool) {
|
|
reqType, ok := ctx.Value(requestTypeKey{}).(requestType)
|
|
return reqType, ok
|
|
}
|
|
|
|
// TODO: Uncomment once metrics are added
|
|
// var (
|
|
// requestMeters = map[requestType]meter{
|
|
// stateSyncRequest: {
|
|
// request: map[bool]metrics.Meter{
|
|
// true: metrics.NewRegisteredMeter("client/requests/statesync/valid", nil),
|
|
// false: metrics.NewRegisteredMeter("client/requests/statesync/invalid", nil),
|
|
// },
|
|
// timer: metrics.NewRegisteredTimer("client/requests/statesync/duration", nil),
|
|
// },
|
|
// spanRequest: {
|
|
// request: map[bool]metrics.Meter{
|
|
// true: metrics.NewRegisteredMeter("client/requests/span/valid", nil),
|
|
// false: metrics.NewRegisteredMeter("client/requests/span/invalid", nil),
|
|
// },
|
|
// timer: metrics.NewRegisteredTimer("client/requests/span/duration", nil),
|
|
// },
|
|
// checkpointRequest: {
|
|
// request: map[bool]metrics.Meter{
|
|
// true: metrics.NewRegisteredMeter("client/requests/checkpoint/valid", nil),
|
|
// false: metrics.NewRegisteredMeter("client/requests/checkpoint/invalid", nil),
|
|
// },
|
|
// timer: metrics.NewRegisteredTimer("client/requests/checkpoint/duration", nil),
|
|
// },
|
|
// checkpointCountRequest: {
|
|
// request: map[bool]metrics.Meter{
|
|
// true: metrics.NewRegisteredMeter("client/requests/checkpointcount/valid", nil),
|
|
// false: metrics.NewRegisteredMeter("client/requests/checkpointcount/invalid", nil),
|
|
// },
|
|
// timer: metrics.NewRegisteredTimer("client/requests/checkpointcount/duration", nil),
|
|
// },
|
|
// }
|
|
// )
|
|
|
|
// TODO: Uncomment once metrics is added
|
|
func sendMetrics(ctx context.Context, start time.Time, isSuccessful bool) {
|
|
// reqType, ok := getRequestType(ctx)
|
|
// if !ok {
|
|
// return
|
|
// }
|
|
|
|
// meters, ok := requestMeters[reqType]
|
|
// if !ok {
|
|
// return
|
|
// }
|
|
|
|
// meters.request[isSuccessful].Mark(1)
|
|
// meters.timer.Update(time.Since(start))
|
|
}
|