Compare commits

...

1 Commits

Author SHA1 Message Date
holger krekel
5cfa2367e3 better preserve notification order, using a queue again 2024-03-30 01:53:41 +01:00
3 changed files with 33 additions and 23 deletions

View File

@@ -2,6 +2,9 @@
## untagged ## untagged
- better preserve notification order
([#263](https://github.com/deltachat/chatmail/pull/263))
- re-enable running the CI in pull requests, but not concurrently - re-enable running the CI in pull requests, but not concurrently
([#258](https://github.com/deltachat/chatmail/pull/258)) ([#258](https://github.com/deltachat/chatmail/pull/258))

View File

@@ -1,7 +1,8 @@
import pwd import pwd
from pathlib import Path from pathlib import Path
from threading import Thread, Event from threading import Thread
from queue import Queue
from socketserver import ( from socketserver import (
UnixStreamServer, UnixStreamServer,
StreamRequestHandler, StreamRequestHandler,
@@ -34,7 +35,7 @@ class Notifier:
self.notification_dir = vmail_dir / "pending_notifications" self.notification_dir = vmail_dir / "pending_notifications"
if not self.notification_dir.exists(): if not self.notification_dir.exists():
self.notification_dir.mkdir() self.notification_dir.mkdir()
self.message_arrived_event = Event() self.notification_queue = Queue()
def get_metadata_dict(self, addr): def get_metadata_dict(self, addr):
return FileDict(self.vmail_dir / addr / "metadata.json") return FileDict(self.vmail_dir / addr / "metadata.json")
@@ -60,31 +61,37 @@ class Notifier:
def new_message_for_addr(self, addr): def new_message_for_addr(self, addr):
self.notification_dir.joinpath(addr).touch() self.notification_dir.joinpath(addr).touch()
self.message_arrived_event.set() self.notification_queue.put(addr)
def thread_run_loop(self): def thread_run_loop(self):
requests_session = requests.Session() requests_session = requests.Session()
# on startup deliver all persisted notifications from last process run
self.notification_queue.put(None)
while 1: while 1:
self.message_arrived_event.wait()
self.message_arrived_event.clear()
self.thread_run_one(requests_session) self.thread_run_one(requests_session)
def thread_run_one(self, requests_session): def thread_run_one(self, requests_session):
for addr_path in self.notification_dir.iterdir(): addr = self.notification_queue.get()
addr = addr_path.name if addr is None:
if "@" not in addr: # startup, notify any "pending" notifications from last run
continue for addr_path in self.notification_dir.iterdir():
for token in self.get_tokens(addr): if "@" in addr_path.name:
response = requests_session.post( self.notify_tokens_for(requests_session, addr_path.name)
"https://notifications.delta.chat/notify", else:
data=token, self.notify_tokens_for(requests_session, addr)
timeout=60,
) def notify_tokens_for(self, requests_session, addr):
if response.status_code == 410: for token in self.get_tokens(addr):
# 410 Gone status code response = requests_session.post(
# means the token is no longer valid. "https://notifications.delta.chat/notify",
self.remove_token(addr, token) data=token,
addr_path.unlink() timeout=60,
)
if response.status_code == 410:
# 410 Gone status code
# means the token is no longer valid.
self.remove_token(addr, token)
self.notification_dir.joinpath(addr).unlink(missing_ok=True)
def handle_dovecot_protocol(rfile, wfile, notifier): def handle_dovecot_protocol(rfile, wfile, notifier):

View File

@@ -84,7 +84,7 @@ def test_handle_dovecot_request_happy_path(notifier, testaddr):
assert handle_dovecot_request(f"B{tx2}\t{testaddr}", transactions, notifier) is None assert handle_dovecot_request(f"B{tx2}\t{testaddr}", transactions, notifier) is None
msg = f"S{tx2}\tpriv/guid00/messagenew" msg = f"S{tx2}\tpriv/guid00/messagenew"
assert handle_dovecot_request(msg, transactions, notifier) is None assert handle_dovecot_request(msg, transactions, notifier) is None
assert notifier.message_arrived_event.is_set() assert notifier.notification_queue.get() == testaddr
assert handle_dovecot_request(f"C{tx2}", transactions, notifier) == "O\n" assert handle_dovecot_request(f"C{tx2}", transactions, notifier) == "O\n"
assert not transactions assert not transactions
assert notifier.notification_dir.joinpath(testaddr).exists() assert notifier.notification_dir.joinpath(testaddr).exists()
@@ -159,8 +159,8 @@ def test_handle_dovecot_protocol_messagenew(notifier):
wfile = io.BytesIO() wfile = io.BytesIO()
handle_dovecot_protocol(rfile, wfile, notifier) handle_dovecot_protocol(rfile, wfile, notifier)
assert wfile.getvalue() == b"O\n" assert wfile.getvalue() == b"O\n"
assert notifier.message_arrived_event.is_set() addr = notifier.notification_queue.get()
assert notifier.notification_dir.joinpath("user@example.org").exists() assert notifier.notification_dir.joinpath(addr).exists()
def test_notifier_thread_run(notifier, testaddr): def test_notifier_thread_run(notifier, testaddr):