Skip to content

Commit 7147cec

Browse files
authored
fix: add federated worker index count (#656)
1 parent 7fb5450 commit 7147cec

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

internal/node/monitor/monitor.go

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,17 @@ func (m *Monitor) processDecentralizedWorker(ctx context.Context, w *config.Modu
108108

109109
// processFederatedWorker processes the federated worker status.
110110
func (m *Monitor) processFederatedWorker(ctx context.Context, w *config.Module) error {
111+
// get checkpoint info from database
112+
indexCount, _, err := m.getCheckpointState(ctx, w.ID, w.Network, w.Worker.Name())
113+
if err != nil {
114+
zap.L().Error("get checkpoint info", zap.Error(err))
115+
return err
116+
}
117+
118+
if err = m.UpdateWorkerProgress(ctx, w.ID, ConstructWorkerProgress(0, 0, 0, indexCount)); err != nil {
119+
return fmt.Errorf("update worker progress: %w", err)
120+
}
121+
111122
client, ok := m.clients[w.Network]
112123
if !ok {
113124
return fmt.Errorf("client not exist")

0 commit comments

Comments
 (0)