From 7147cec83028e0c5bea06e31c6fea2896f691451 Mon Sep 17 00:00:00 2001
From: brucexc <108378464+brucexc@users.noreply.github.com>
Date: Wed, 20 Nov 2024 23:56:36 -0500
Subject: [PATCH] fix: add federated worker index count (#656)

---
 internal/node/monitor/monitor.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/internal/node/monitor/monitor.go b/internal/node/monitor/monitor.go
index 554d49c21..b76112a52 100644
--- a/internal/node/monitor/monitor.go
+++ b/internal/node/monitor/monitor.go
@@ -108,6 +108,17 @@ func (m *Monitor) processDecentralizedWorker(ctx context.Context, w *config.Modu
 
 // processFederatedWorker processes the federated worker status.
 func (m *Monitor) processFederatedWorker(ctx context.Context, w *config.Module) error {
+	// get checkpoint info from database
+	indexCount, _, err := m.getCheckpointState(ctx, w.ID, w.Network, w.Worker.Name())
+	if err != nil {
+		zap.L().Error("get checkpoint info", zap.Error(err))
+		return err
+	}
+
+	if err = m.UpdateWorkerProgress(ctx, w.ID, ConstructWorkerProgress(0, 0, 0, indexCount)); err != nil {
+		return fmt.Errorf("update worker progress: %w", err)
+	}
+
 	client, ok := m.clients[w.Network]
 	if !ok {
 		return fmt.Errorf("client not exist")