db: don't pass invalid argument to Uint32N

Fixes #4021
cockroachdb · Oct 9, 2024 · eb01a00 · eb01a00
1 parent 6e41e55
commit eb01a00
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/iterator.go b/iterator.go
@@ -817,8 +817,10 @@ func (i *Iterator) maybeSampleRead() {
 		// every newly-opened iterator, but we do want to sample some of them.
 		if !i.readSampling.initialSamplePassed {
 			i.readSampling.initialSamplePassed = true
-			if rand.Uint32N(uint32(i.readSampling.bytesUntilReadSampling)) > uint32(bytesRead) {
-				continue
+			if i.readSampling.bytesUntilReadSampling > bytesRead {
+				if rand.Uint64N(i.readSampling.bytesUntilReadSampling) > bytesRead {
+					continue
+				}
 			}
 		}
 		i.sampleRead()