File tree 3 files changed +9
-8
lines changed
3 files changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -3337,7 +3337,7 @@ func BenchmarkAllocDoDropRPC(b *testing.B) {
3337
3337
}
3338
3338
}
3339
3339
3340
- func TestRarestFirstRPCScheduler (t * testing.T ) {
3340
+ func TestRoundRobinMessageIDScheduler (t * testing.T ) {
3341
3341
const maxNumPeers = 256
3342
3342
const maxNumMessages = 1_000
3343
3343
@@ -3347,7 +3347,7 @@ func TestRarestFirstRPCScheduler(t *testing.T) {
3347
3347
3348
3348
output := make ([]pendingRPC , 0 , numMessages * numPeers )
3349
3349
3350
- var strategy RarestFirstRPCScheduler
3350
+ var strategy RoundRobinMessageIDScheduler
3351
3351
3352
3352
peers := make ([]peer.ID , numPeers )
3353
3353
for i := 0 ; i < int (numPeers ); i ++ {
@@ -3429,10 +3429,10 @@ func TestRarestFirstRPCScheduler(t *testing.T) {
3429
3429
}
3430
3430
}
3431
3431
3432
- func BenchmarkRarestFirstRPCScheduler (b * testing.B ) {
3432
+ func BenchmarkRoundRobinMessageIDScheduler (b * testing.B ) {
3433
3433
const numPeers = 1_000
3434
3434
const numMessages = 1_000
3435
- var strategy RarestFirstRPCScheduler
3435
+ var strategy RoundRobinMessageIDScheduler
3436
3436
3437
3437
peers := make ([]peer.ID , numPeers )
3438
3438
for i := range int (numPeers ) {
Original file line number Diff line number Diff line change @@ -31,18 +31,19 @@ type pendingRPC struct {
31
31
rpc * RPC
32
32
}
33
33
34
- type RarestFirstRPCScheduler struct {
34
+ // RoundRobinMessageIDScheduler schedules outgoing RPCs in round-robin order of message IDs.
35
+ type RoundRobinMessageIDScheduler struct {
35
36
rpcs map [string ][]pendingRPC
36
37
}
37
38
38
- func (s * RarestFirstRPCScheduler ) AddRPC (peer peer.ID , msgID string , rpc * RPC ) {
39
+ func (s * RoundRobinMessageIDScheduler ) AddRPC (peer peer.ID , msgID string , rpc * RPC ) {
39
40
if s .rpcs == nil {
40
41
s .rpcs = make (map [string ][]pendingRPC )
41
42
}
42
43
s .rpcs [msgID ] = append (s .rpcs [msgID ], pendingRPC {peer : peer , rpc : rpc })
43
44
}
44
45
45
- func (s * RarestFirstRPCScheduler ) All () iter.Seq2 [peer.ID , * RPC ] {
46
+ func (s * RoundRobinMessageIDScheduler ) All () iter.Seq2 [peer.ID , * RPC ] {
46
47
return func (yield func (peer.ID , * RPC ) bool ) {
47
48
for len (s .rpcs ) > 0 {
48
49
for msgID , rpcs := range s .rpcs {
Original file line number Diff line number Diff line change @@ -228,7 +228,7 @@ type BatchPubOpt func(pub *BatchPublishOptions) error
228
228
229
229
func setDefaultBatchPublishOptions (opts * BatchPublishOptions ) {
230
230
if opts .Strategy == nil {
231
- opts .Strategy = & RarestFirstRPCScheduler {}
231
+ opts .Strategy = & RoundRobinMessageIDScheduler {}
232
232
}
233
233
}
234
234
You can’t perform that action at this time.
0 commit comments