Adjust POA to adhere to new SeatManager interface

iotaledger · Nov 8, 2023 · 409f2e3 · 409f2e3
1 parent ae8da07
commit 409f2e3
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pkg/protocol/sybilprotection/seatmanager/poa/poa.go b/pkg/protocol/sybilprotection/seatmanager/poa/poa.go
@@ -146,7 +146,14 @@ func (s *SeatManager) OnlineCommittee() ds.Set[account.SeatIndex] {
 	return s.activityTracker.OnlineCommittee()
 }
 
-func (s *SeatManager) SeatCount() int {
+func (s *SeatManager) SeatCountInSlot(slot iotago.SlotIndex) int {
+	s.committeeMutex.RLock()
+	defer s.committeeMutex.RUnlock()
+
+	return s.committee.SeatCount()
+}
+
+func (s *SeatManager) SeatCountInEpoch(epoch iotago.EpochIndex) int {
 	s.committeeMutex.RLock()
 	defer s.committeeMutex.RUnlock()