Mirror/app-MAIL-temp
1
0
Fork 0
mirror of https://github.com/simple-login/app.git synced 2024-11-10 21:27:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

chore: allow to define a different DB_URI for event listener (#2189)

Browse Source
This commit is contained in:
Carlos Quintana 2024-08-20 16:01:48 +02:00 committed by GitHub
parent a72b7bde92
commit 33c418d7c6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/config.py
View File

@ -596,12 +596,16 @@ def read_webhook_enabled_user_ids() -> Optional[List[int]]:
return None return None
ids = [] ids = []
for id in user_ids.split(","): for user_id in user_ids.split(","):
try: try:
ids.append(int(id.strip())) ids.append(int(user_id.strip()))
except ValueError: except ValueError:
pass pass
return ids return ids
EVENT_WEBHOOK_ENABLED_USER_IDS: Optional[List[int]] = read_webhook_enabled_user_ids() EVENT_WEBHOOK_ENABLED_USER_IDS: Optional[List[int]] = read_webhook_enabled_user_ids()
# Allow to define a different DB_URI for the event listener, in case we want to skip the connection pool
# It defaults to the regular DB_URI in case it's needed
EVENT_LISTENER_DB_URI = os.environ.get("EVENT_LISTENER_DB_URI", DB_URI)

4
event_listener.py
View File

@ -2,7 +2,7 @@ import argparse
from enum import Enum from enum import Enum
from sys import argv, exit from sys import argv, exit
from app.config import DB_URI from app.config import EVENT_LISTENER_DB_URI
from app.log import LOG from app.log import LOG
from events.runner import Runner from events.runner import Runner
from events.event_source import DeadLetterEventSource, PostgresEventSource from events.event_source import DeadLetterEventSource, PostgresEventSource
@ -31,7 +31,7 @@ def main(mode: Mode, dry_run: bool, max_retries: int):
source = DeadLetterEventSource(max_retries) source = DeadLetterEventSource(max_retries)
elif mode == Mode.LISTENER: elif mode == Mode.LISTENER:
LOG.i("Using PostgresEventSource") LOG.i("Using PostgresEventSource")
source = PostgresEventSource(DB_URI) source = PostgresEventSource(EVENT_LISTENER_DB_URI)
else: else:
raise ValueError(f"Invalid mode: {mode}") raise ValueError(f"Invalid mode: {mode}")

1
events/event_source.py
View File

@ -46,6 +46,7 @@ class PostgresEventSource(EventSource):
cursor = self.__connection.cursor() cursor = self.__connection.cursor()
cursor.execute(f"LISTEN {NOTIFICATION_CHANNEL};") cursor.execute(f"LISTEN {NOTIFICATION_CHANNEL};")
LOG.info("Starting to listen to events")
while True: while True:
if select.select([self.__connection], [], [], 5) != ([], [], []): if select.select([self.__connection], [], [], 5) != ([], [], []):
self.__connection.poll() self.__connection.poll()