Skip to content

Commit d2b2290

Browse files
committed
Ensure traverse is cancelable even if it's sending a tNode.
1 parent 35e2d1c commit d2b2290

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

trie/ctrie/ctrie.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -406,7 +406,11 @@ func (c *Ctrie) traverse(i *iNode, ch chan<- *Entry, cancel <-chan struct{}) err
406406
}
407407
}
408408
case main.tNode != nil:
409-
ch <- main.tNode.Entry
409+
select {
410+
case ch <- main.tNode.Entry:
411+
case <-cancel:
412+
return errCanceled
413+
}
410414
}
411415
return nil
412416
}

0 commit comments

Comments
 (0)