Open dcsommer opened 3 years ago
A PR would be great, but I would like to see a couple constraints:
tee(itr: ..., n: int = 2, *, limit: int = 100)
, where limit=0
would be unbounded. await asyncio.sleep(0)
after each item would be sufficient.By default, this would still cause consumption to be limited by the slowest iterator, but no longer dependent on the first iterator specifically.
The implementation of aioitertools.itertools.tee currently has the caveat "all iterators are dependent on the first iterator." This caveat can cause periodic stalls in my pipeline where the consumers all consume at roughly the same speed, with small imbalances.
A small refactor of
tee
could remove this limitation if the producer was a separate scheduled task (instead of the first consumer). Would you be open to a pull request for a change along these lines?