running protocol descriptor directly update the connected node map

ChristianMct · Apr 22, 2024 · 2a90505 · 2a90505
1 parent e9e44d3
commit 2a90505
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/protocols/executor.go b/protocols/executor.go
@@ -489,6 +489,17 @@ func (s *Executor) RunDescriptorAsAggregator(ctx context.Context, pd Descriptor)
 		return nil, fmt.Errorf("could not extract session from context")
 	}
 
+	s.connectedNodesMu.Lock()
+	for _, part := range pd.Participants {
+		if pids, has := s.connectedNodes[part]; !has {
+			s.connectedNodesMu.Unlock()
+			return nil, fmt.Errorf("participant %s not registered", part)
+		} else {
+			pids.Add(pd.ID())
+		}
+	}
+	s.connectedNodesMu.Unlock()
+
 	agg := s.runAsAggregator(ctx, sess, pd)
 	return &agg, agg.Error
 }