Skip to content

Commit 051c8b3

Browse files
committed
rename scheduler
1 parent 9625034 commit 051c8b3

File tree

3 files changed

+9
-8
lines changed

3 files changed

+9
-8
lines changed

gossipsub_test.go

+4-4
Original file line numberDiff line numberDiff line change
@@ -3337,7 +3337,7 @@ func BenchmarkAllocDoDropRPC(b *testing.B) {
33373337
}
33383338
}
33393339

3340-
func TestRarestFirstRPCScheduler(t *testing.T) {
3340+
func TestRoundRobinMessageIDScheduler(t *testing.T) {
33413341
const maxNumPeers = 256
33423342
const maxNumMessages = 1_000
33433343

@@ -3347,7 +3347,7 @@ func TestRarestFirstRPCScheduler(t *testing.T) {
33473347

33483348
output := make([]pendingRPC, 0, numMessages*numPeers)
33493349

3350-
var strategy RarestFirstRPCScheduler
3350+
var strategy RoundRobinMessageIDScheduler
33513351

33523352
peers := make([]peer.ID, numPeers)
33533353
for i := 0; i < int(numPeers); i++ {
@@ -3429,10 +3429,10 @@ func TestRarestFirstRPCScheduler(t *testing.T) {
34293429
}
34303430
}
34313431

3432-
func BenchmarkRarestFirstRPCScheduler(b *testing.B) {
3432+
func BenchmarkRoundRobinMessageIDScheduler(b *testing.B) {
34333433
const numPeers = 1_000
34343434
const numMessages = 1_000
3435-
var strategy RarestFirstRPCScheduler
3435+
var strategy RoundRobinMessageIDScheduler
34363436

34373437
peers := make([]peer.ID, numPeers)
34383438
for i := range int(numPeers) {

messagebatch.go

+4-3
Original file line numberDiff line numberDiff line change
@@ -31,18 +31,19 @@ type pendingRPC struct {
3131
rpc *RPC
3232
}
3333

34-
type RarestFirstRPCScheduler struct {
34+
// RoundRobinMessageIDScheduler schedules outgoing RPCs in round-robin order of message IDs.
35+
type RoundRobinMessageIDScheduler struct {
3536
rpcs map[string][]pendingRPC
3637
}
3738

38-
func (s *RarestFirstRPCScheduler) AddRPC(peer peer.ID, msgID string, rpc *RPC) {
39+
func (s *RoundRobinMessageIDScheduler) AddRPC(peer peer.ID, msgID string, rpc *RPC) {
3940
if s.rpcs == nil {
4041
s.rpcs = make(map[string][]pendingRPC)
4142
}
4243
s.rpcs[msgID] = append(s.rpcs[msgID], pendingRPC{peer: peer, rpc: rpc})
4344
}
4445

45-
func (s *RarestFirstRPCScheduler) All() iter.Seq2[peer.ID, *RPC] {
46+
func (s *RoundRobinMessageIDScheduler) All() iter.Seq2[peer.ID, *RPC] {
4647
return func(yield func(peer.ID, *RPC) bool) {
4748
for len(s.rpcs) > 0 {
4849
for msgID, rpcs := range s.rpcs {

topic.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,7 @@ type BatchPubOpt func(pub *BatchPublishOptions) error
228228

229229
func setDefaultBatchPublishOptions(opts *BatchPublishOptions) {
230230
if opts.Strategy == nil {
231-
opts.Strategy = &RarestFirstRPCScheduler{}
231+
opts.Strategy = &RoundRobinMessageIDScheduler{}
232232
}
233233
}
234234

0 commit comments

Comments
 (0)