Added metrics for observed waypoints (#12785)

polygon/heimdall/metrics.go

@@ -86,6 +86,9 @@ var (
 			timer: metrics.GetOrCreateSummary("client_requests_checkpointcount_duration"),
 		},
 	}
+
+	waypointCheckpointLength = metrics.NewGauge(`waypoint_length{type="checkpoint"}`)
+	waypointMilestoneLength  = metrics.NewGauge(`waypoint_length{type="milestone"}`)
 )
 
 func sendMetrics(ctx context.Context, start time.Time, isSuccessful bool) {
@@ -102,3 +105,11 @@ func sendMetrics(ctx context.Context, start time.Time, isSuccessful bool) {
 	meters.request[isSuccessful].Set(1)
 	meters.timer.ObserveDuration(start)
 }
+
+func UpdateObservedWaypointCheckpointLength(length uint64) {
+	waypointCheckpointLength.SetUint64(length)
+}
+
+func UpdateObservedWaypointMilestoneLength(length uint64) {
+	waypointMilestoneLength.SetUint64(length)
+}

polygon/heimdall/service.go

@@ -228,6 +228,15 @@ func (s *Service) RegisterMilestoneObserver(callback func(*Milestone), opts ...O
 	})
 }
 
+func (s *Service) RegisterCheckpointObserver(callback func(*Checkpoint), opts ...ObserverOption) polygoncommon.UnregisterFunc {
+	options := NewObserverOptions(opts...)
+	return s.checkpointScraper.RegisterObserver(func(entities []*Checkpoint) {
+		for _, entity := range libcommon.SliceTakeLast(entities, options.eventsLimit) {
+			callback(entity)
+		}
+	})
+}
+
 func (s *Service) RegisterSpanObserver(callback func(*Span), opts ...ObserverOption) polygoncommon.UnregisterFunc {
 	options := NewObserverOptions(opts...)
 	return s.spanScraper.RegisterObserver(func(entities []*Span) {
@@ -304,6 +313,16 @@ func (s *Service) Run(ctx context.Context) error {
 		s.spanBlockProducersTracker.ObserveSpanAsync(span)
 	})
 
+	milestoneObserver := s.RegisterMilestoneObserver(func(milestone *Milestone) {
+		UpdateObservedWaypointMilestoneLength(milestone.Length())
+	})
+	defer milestoneObserver()
+
+	checkpointObserver := s.RegisterCheckpointObserver(func(checkpoint *Checkpoint) {
+		UpdateObservedWaypointCheckpointLength(checkpoint.Length())
+	}, WithEventsLimit(5))
+	defer checkpointObserver()
+
 	eg, ctx := errgroup.WithContext(ctx)
 	eg.Go(func() error {
 		if err := s.checkpointScraper.Run(ctx); err != nil {