For my graph library petgraph I found a new pattern for visitor callbacks that hits the sweet spot for both convenience and the “pay for what you use” principle.
It’s a visitor callback based graph traversal function, and of course it should support breaking the traversal early.
One common solution is to use a boolean return value for the visitor closure.
true to continue, and
false to break — or is it the other way around? At
this point in my Rust experience much prefer enums to booleans, they make the
code easier to both read and write.
Enums are one of Rust’s absolute best features, and we can quickly express what we want. We allow the user to break with a value:
At the same time we don’t want to burden the user with verbose enum imports for the common case of never breaking at all. Traits to the rescue!
The default return type of a closure is
(); this is returned if you just leave
off with a semicolon. Uniting both
() with a common trait
will be really useful!
The result is an API where the default case (no break) is seamlessly available. Breaking with a value is opt-in, and what’s even better is that the compiler knows statically whether it needs to check for early return or not, which is the “pay for what you use” factor.
The finished graph traversal function is
As a bonus, we let errors break through as well (which makes some algorithms beautiful):
In the usage site we use a macro similar to
try!() to simplify the usage of