db: fix merge skew

Add WrapChildren method to probeKeyspanIterator.
cockroachdb · Jan 10, 2024 · 136f75c · 136f75c
1 parent 3171a93
commit 136f75c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/keyspan_probe_test.go b/keyspan_probe_test.go
@@ -370,6 +370,10 @@ func (p *probeKeyspanIterator) Prev() *keyspan.Span {
 	return p.handleOp(op)
 }
 
+func (p *probeKeyspanIterator) WrapChildren(wrap keyspan.WrapFn) {
+	p.iter = wrap(p.iter)
+}
+
 func (p *probeKeyspanIterator) Error() error {
 	return p.err
 }