forked from ava-labs/avalanchego
-
Notifications
You must be signed in to change notification settings - Fork 4
/
requests.go
47 lines (38 loc) · 1.02 KB
/
requests.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
// Copyright (C) 2019-2024, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package bootstrapper
import (
"context"
"github.com/MetalBlockchain/metalgo/ids"
"github.com/MetalBlockchain/metalgo/utils/set"
)
type requests struct {
maxOutstanding int
pendingSend set.Set[ids.NodeID]
outstanding set.Set[ids.NodeID]
}
func (r *requests) GetPeers(context.Context) set.Set[ids.NodeID] {
numPending := r.outstanding.Len()
if numPending >= r.maxOutstanding {
return nil
}
numToSend := min(
r.maxOutstanding-numPending,
r.pendingSend.Len(),
)
nodeIDs := set.NewSet[ids.NodeID](numToSend)
for i := 0; i < numToSend; i++ {
nodeID, _ := r.pendingSend.Pop()
nodeIDs.Add(nodeID)
}
r.outstanding.Union(nodeIDs)
return nodeIDs
}
func (r *requests) recordResponse(nodeID ids.NodeID) bool {
wasOutstanding := r.outstanding.Contains(nodeID)
r.outstanding.Remove(nodeID)
return wasOutstanding
}
func (r *requests) finished() bool {
return r.pendingSend.Len() == 0 && r.outstanding.Len() == 0
}