diff options
author | Richard van der Hoff <richard@matrix.org> | 2020-01-16 22:26:34 +0000 |
---|---|---|
committer | Richard van der Hoff <richard@matrix.org> | 2020-01-16 22:26:34 +0000 |
commit | acc7820574426cf27673d941b1b0362272113351 (patch) | |
tree | 3af507f4d6d93f5200ed6830b0dcadeff01f7e70 /synapse/util | |
parent | move batch_iter to a separate module (diff) | |
download | synapse-acc7820574426cf27673d941b1b0362272113351.tar.xz |
Log saml assertions rather than the whole response
... since the whole response is huge. We even need to break up the assertions, since kibana otherwise truncates them.
Diffstat (limited to 'synapse/util')
-rw-r--r-- | synapse/util/iterutils.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/synapse/util/iterutils.py b/synapse/util/iterutils.py index c10016fbc5..06faeebe7f 100644 --- a/synapse/util/iterutils.py +++ b/synapse/util/iterutils.py @@ -33,3 +33,16 @@ def batch_iter(iterable: Iterable[T], size: int) -> Iterator[Tuple[T]]: sourceiter = iter(iterable) # call islice until it returns an empty tuple return iter(lambda: tuple(islice(sourceiter, size)), ()) + + +ISeq = TypeVar("ISeq", bound=Sequence, covariant=True) + + +def chunk_seq(iseq: ISeq, maxlen: int) -> Iterable[ISeq]: + """Split the given sequence into chunks of the given size + + The last chunk may be shorter than the given size. + + If the input is empty, no chunks are returned. + """ + return (iseq[i : i + maxlen] for i in range(0, len(iseq), maxlen)) |