gh-108951: add TaskGroup.cancel()#127214
Conversation
sobolevn
left a comment
There was a problem hiding this comment.
Thank you! This is not a full review, just a couple of questions.
There was a problem hiding this comment.
Maybe these tasks should look like this?
async def task(results, num):
results.append(num)
await asyncio.sleep(math.inf)
results.append(-num)There was a problem hiding this comment.
So we can assert what was in results
There was a problem hiding this comment.
For this particular test, I chose a different test approach, which is to wrap in asyncio.timeout().
For the other tests using count, I'm not sure it's much value, since the test code is only a few lines and there is only one possible path through it. So count result of 0, 1, or 2 each have deterministic meaning that's obvious by looking at the code.
Co-authored-by: sobolevn <mail@sobolevn.me>
1st1
left a comment
There was a problem hiding this comment.
Why call it TaskGroup.stop() and not TaskGroup.cancel()? I'd be more in favor of the latter name.
Short-circuiting of task groups is a very common, useful, and normal, so make it a first-class operation.
Any evidence of this statement? I'd like you to write up technical motivation + examples. That will be useful for the docs.
And speaking of the documentation, you should also show some recipies of how this would be used. Like are you supposed to use this API from within the task group async with clause? Or can you pass the task group to some remote task?
I haven't reviewed the actual implementation in detail yet.
|
A Python core developer has requested some changes be made to your pull request before we can consider merging it. If you could please address their requests along with any other requests in other reviews from core developers that would be appreciated. Once you have made the requested changes, please leave a comment on this pull request containing the phrase |
|
This doesn't work in the case that the body of the task group throws an exception, as in this code: async def test_taskgroup_throw_inside(self):
class MyError(RuntimeError):
pass
should_get_here = False
try:
async with asyncio.TaskGroup() as tg:
tg.create_task(asyncio.sleep(0.1))
tg.stop()
self.assertEqual(asyncio.current_task().cancelling(), 1)
raise MyError
self.fail() # <-- reaches here instead of raising ExceptionGroup([MyError()])
except* MyError:
self.assertEqual(asyncio.current_task().cancelling(), 0)
should_get_here = True
self.assertTrue(should_get_here)The problem is that the new code in the if et is not None and not issubclass(et, exceptions.CancelledError):
self._errors.append(exc)One option is move these lines earlier, before the I'd still suggest my original proposal (see the issue) where you just add a single line As a separate point, I'd suggest that the tests could do with a few more checks that |
I'd also prefer
In trio the equivalent is I have years experience developing a non-trivial, production async app, which I've presented at PyCon JP. Anecdotally, I can't imagine how painful and unproductive it would be to not have short circuiting of task groups.
All is on the table: stop from within the TaskGroup body, from a child, from some other entity you've passed the bound stop() method to. |
Well, that's exactly what it does, isn't it? The fact that the cancelled taskgroup catches the Also, trio and anyio already call this operation |
7e25c26 to
f077241
Compare
|
|
||
| Ways to use :meth:`cancel`: | ||
|
|
||
| * call it from the task group body based on some condition or event |
There was a problem hiding this comment.
Probably you want code examples for all of these?
There was a problem hiding this comment.
I have some comments. This is not an endorsement of this PR -- I want to look at @smurfix' version too before deciding. I'm also still torn between naming it stop() or cancel().
Edit: There is no PR by @smurfix -- he opened the issue. Sorry. I think that with the feedback (mine and others') addressed and a decision on the name (stop or cancel) made we can merge this in time before the May 1st beta deadline (in ~10 days).
|
For email readers: Sorry; there is no PR by @smurfix -- he opened the issue. I think that with the feedback (mine and others') addressed (possibly by explaining why no change is needed!) and a decision on the name (stop or cancel) made we can merge this in time before the May 1st beta deadline (in ~10 days). |
Co-authored-by: Guido van Rossum <gvanrossum@gmail.com>
gvanrossum
left a comment
There was a problem hiding this comment.
I count the following TODOs:
- Decide on name: stop vs. cancel (it feels odd that something called cancel() ultimately suppresses the CanceledException)
- Cancel messages (though this could be added in a post-beta-1 bugfix)
- Test comments
|
Also there are merge conflicts. :-( |
gvanrossum
left a comment
There was a problem hiding this comment.
I did the merge and fixed the stray quote in NEWS.
|
[@1st1 / Yury -- on Nov 25, 2024]
Hm. If Yury's intuition goes towards cancel(), I revoke my opposition to it. Decision made: it's cancel().
The main evidence is the huge paragraph that was added to the 3.12/3.13/3.14 docs about how to terminate a TaskGroup.
All of the above. Whoever has a reference to a TaskGroup can call its cancel() and the right thing will happen.
I have and it's fine. |
|
A failing test complains about deleting a label ("Terminating a Task Group"):
I think the latter is best -- though you could also bring the label back with only the body text (paraphrased):
|
|
@belm0 -- I take it you are not planning to add cancel messages and you are not planning to add more examples to the docs? Once you confirm that I will approve and merge. |
belm0
left a comment
There was a problem hiding this comment.
I take it you are not planning to add cancel messages and you are not planning to add more examples to the docs?
I've been working through the PR's items in some priority order and hadn't reached those (you've caught me in the middle of quite a hectic week).
I've just committed a few tweaks to the added docs, but no cancel examples. Task groups are quite powerful and I've seen other async libs build up examples over several pages. The existing docs are lacking here so I don't have the foundation to add the cancel() examples. Expanded TaskGroup docs need to be written by someone who lives and breathes asyncio daily and is intimate with the library's nuances (that was once me for trio, but not asyncio).
For cancel(msg) I've made another comment on that thread-- not being bought into the idea I'm reluctant to take on the extra impl/doc/testing in this PR.
gvanrossum
left a comment
There was a problem hiding this comment.
LGTM; I'll merge it.
I do have some nits on the docs still but they don't need to hold up the merge.

Short-circuiting of task groups is a very common, useful, and normal, so make it a first-class operation. The recommended approach to date-- creating a task just to raise an exception, and then catch and suppress the exception-- is inefficient, prone to races, and requires a lot of boilerplate.
asyncio.TaskGroup.cancelmethod #108951📚 Documentation preview 📚: https://cpython-previews--127214.org.readthedocs.build/