Skip to content

Commit

Permalink
kvnemesis: add TestingKnobs.OnRangeSpanningNonTxnalBatch back
Browse files Browse the repository at this point in the history
#103963 accidentally removed a testing knob which caused #104865.
This commit adds the testing knob back.

Fixes: #104865
Release note: none
  • Loading branch information
wenyihu6 committed Jun 14, 2023
1 parent 3572499 commit 5dd8029
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions pkg/kv/kvclient/kvcoord/dist_sender.go
Original file line number Diff line number Diff line change
Expand Up @@ -626,6 +626,10 @@ func NewDistSender(cfg DistSenderConfig) *DistSender {
ds.latencyFunc = ds.rpcContext.RemoteClocks.Latency
}

if cfg.TestingKnobs.OnRangeSpanningNonTxnalBatch != nil {
ds.onRangeSpanningNonTxnalBatch = cfg.TestingKnobs.OnRangeSpanningNonTxnalBatch
}

if cfg.TestingKnobs.BatchRequestInterceptor != nil {
ds.BatchRequestInterceptor = cfg.TestingKnobs.BatchRequestInterceptor
}
Expand Down

0 comments on commit 5dd8029

Please sign in to comment.