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>
code_spécifique_watcha
Sean Quah 3 years ago committed by GitHub
parent 9c4c49991d
commit 31c1209c50
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 1
      changelog.d/12366.misc
  2. 5
      synapse/types.py

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

@ -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")

Loading…
Cancel
Save