Make StreamToken and RoomStreamToken methods propagate cancellations (#12366)

`StreamToken.from_string` and `RoomStreamToken.parse` are both async
methods that could be cancelled. These methods must not replace
`CancelledError`s with `SynapseError`s.

Signed-off-by: Sean Quah <seanq@element.io>
This commit is contained in:
Sean Quah 2022-04-05 16:56:52 +01:00 committed by GitHub
parent 9c4c49991d
commit 31c1209c50
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 0 deletions

1
changelog.d/12366.misc Normal file
View File

@ -0,0 +1 @@
Make `StreamToken.from_string` and `RoomStreamToken.parse` propagate cancellations instead of replacing them with `SynapseError`s.

View File

@ -39,6 +39,7 @@ from typing_extensions import TypedDict
from unpaddedbase64 import decode_base64 from unpaddedbase64 import decode_base64
from zope.interface import Interface from zope.interface import Interface
from twisted.internet.defer import CancelledError
from twisted.internet.interfaces import ( from twisted.internet.interfaces import (
IReactorCore, IReactorCore,
IReactorPluggableNameResolver, IReactorPluggableNameResolver,
@ -540,6 +541,8 @@ class RoomStreamToken:
stream=stream, stream=stream,
instance_map=frozendict(instance_map), instance_map=frozendict(instance_map),
) )
except CancelledError:
raise
except Exception: except Exception:
pass pass
raise SynapseError(400, "Invalid room stream token %r" % (string,)) raise SynapseError(400, "Invalid room stream token %r" % (string,))
@ -705,6 +708,8 @@ class StreamToken:
return cls( return cls(
await RoomStreamToken.parse(store, keys[0]), *(int(k) for k in keys[1:]) await RoomStreamToken.parse(store, keys[0]), *(int(k) for k in keys[1:])
) )
except CancelledError:
raise
except Exception: except Exception:
raise SynapseError(400, "Invalid stream token") raise SynapseError(400, "Invalid stream token")