1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
# Copyright (c) 2022 Tulir Asokan
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import annotations
from abc import ABC, abstractmethod
from mautrix.types import SyncToken
class SyncStore(ABC):
"""SyncStore persists information used by /sync."""
@abstractmethod
async def put_next_batch(self, next_batch: SyncToken) -> None:
pass
@abstractmethod
async def get_next_batch(self) -> SyncToken:
pass
class MemorySyncStore(SyncStore):
"""MemorySyncStore is a :class:`SyncStore` implementation that stores the data in memory."""
def __init__(self, next_batch: SyncToken | None = None) -> None:
self._next_batch: SyncToken | None = next_batch
async def put_next_batch(self, next_batch: SyncToken) -> None:
self._next_batch = next_batch
async def get_next_batch(self) -> SyncToken:
return self._next_batch
|
