Sikring af stabilitet i langvarige PostgreSQL-notifikationslyttere med Psycopg3

Sikring af pålidelige meddelelser i langvarige PostgreSQL-lyttere

Backend-implementering ved hjælp af Python og psycopg3 til at håndtere langvarige databaseforbindelser

import psycopg
from psycopg import sql
import time
CONN_STR = "postgresql://user:password@localhost/dbname"
def listen_notifications():
    try:
        with psycopg.connect(CONN_STR, autocommit=True) as conn:
            listen_sql = sql.SQL("LISTEN {};").format(sql.Identifier("scheduler_test"))
            conn.execute(listen_sql)
            print("Listening for notifications...")
            gen = conn.notifies(timeout=5)
            for notification in gen:
                print("Received notification:", notification)
                perform_health_check(conn, listen_sql)
    except Exception as e:
        print("Error:", e)
def perform_health_check(conn, listen_sql):
    try:
        print("Performing health check...")
        conn.execute("SELECT 1")
        conn.execute(listen_sql)
    except Exception as e:
        print("Health check failed:", e)
if __name__ == "__main__":
    listen_notifications()

Alternativ tilgang: Brug af asynkron psycopg3 til forbedret lydhørhed

Asynkron implementering ved hjælp af Pythons asyncio og psycopg3

import asyncio
from psycopg import AsyncConnection, sql
CONN_STR = "postgresql://user:password@localhost/dbname"
async def listen_notifications():
    try:
        async with AsyncConnection.connect(CONN_STR, autocommit=True) as conn:
            listen_sql = sql.SQL("LISTEN {};").format(sql.Identifier("scheduler_test"))
            await conn.execute(listen_sql)
            print("Listening for notifications...")
            gen = conn.notifies(timeout=5)
            async for notification in gen:
                print("Received notification:", notification)
                await perform_health_check(conn, listen_sql)
    except Exception as e:
        print("Error:", e)
async def perform_health_check(conn, listen_sql):
    try:
        print("Performing health check...")
        await conn.execute("SELECT 1")
        await conn.execute(listen_sql)
    except Exception as e:
        print("Health check failed:", e)
if __name__ == "__main__":
    asyncio.run(listen_notifications())

Enhedstest for robusthed

Python-enhedstests for backend-logik ved hjælp af unittest

import unittest
from unittest.mock import patch, MagicMock
class TestNotificationListener(unittest.TestCase):
    @patch("psycopg.connect")
    def test_listen_notifications(self, mock_connect):
        mock_conn = MagicMock()
        mock_connect.return_value.__enter__.return_value = mock_conn
        mock_conn.notifies.return_value = iter(["test_notification"])
        listen_notifications()
        mock_conn.execute.assert_called_with("LISTEN scheduler_test;")
        mock_conn.notifies.assert_called_once()
if __name__ == "__main__":
    unittest.main()

Ofte stillede spørgsmål om administration af PostgreSQL-notifikationslyttere

1. Hvad er formålet med conn.notifies() i psychopg3?
2. conn.notifies() bruges til at hente meddelelser sendt af PostgreSQL-serveren, hvilket muliggør hændelseshåndtering i realtid i applikationer.
3. Kan LISTEN kommandoer mister beskeder under genforbindelse?
4. Nej, PostgreSQL buffer meddelelser, så meddelelser ikke går tabt under genforbindelse. Men korrekt håndtering af notifies generator er påkrævet for at sikre problemfri behandling.
5. Hvorfor skal jeg bruge autocommit=True?
6. Indstilling autocommit=True tillader forbindelsen straks at anvende kommandoer som LISTEN uden at vente på en eksplicit forpligtelse, hvilket forbedrer reaktionsevnen.
7. Hvordan kan jeg udføre sundhedstjek under en langvarig notifies behandle?
8. Du kan med jævne mellemrum udføre lette forespørgsler som f.eks SELECT 1 for at sikre, at forbindelsen forbliver lydhør.
9. Hvad er den bedste praksis for at rydde op i databaseforbindelser?
10. Ved hjælp af en with statement eller Pythons kontekstmanager sikrer, at forbindelsen er ordentligt lukket, og undgår ressourcelækage.
11. Hvordan håndterer jeg timeout-undtagelser i conn.notifies()?
12. Indpakning conn.notifies() i en try-except-blok for at fange timeout-undtagelser og håndtere dem elegant, f.eks. ved at logge eller prøve igen.
13. Understøtter psycopg3 asynkrone operationer for meddelelser?
14. Ja, psycopg3 tilbyder en asynkron API via AsyncConnection, som er ideel til ikke-blokerende, skalerbare applikationer.
15. Hvad sker der, hvis jeg ikke lukker notifies generator?
16. Hvis generatoren ikke lukkes, kan det resultere i hukommelseslækager eller hængende ressourcer, især i langvarige processer.
17. Kan notifikationer gå glip af i løbet af en pg_sleep() operation?
18. Ja, meddelelser, der genereres i dvaleperioden, kan gå glip af, hvis de ikke bufferes, hvilket er grunden til korrekt håndtering af LISTEN kommandoer er afgørende.
19. Er det sikkert at genbruge den samme forbindelse til flere meddelelser?
20. Ja, så længe sundhedstjek og korrekte genforbindelser administreres, er genbrug af den samme forbindelse effektivt og ressourcevenligt.
21. Hvordan kan jeg teste pålideligheden af ​​mit meddelelsessystem?
22. Skriv enhedstest ved hjælp af biblioteker som unittest.mock at simulere meddelelser og databaseadfærd uden at være afhængig af en live server.

Sikring af pålidelig notifikationslytning

Opretholdelse af forbindelsessundhed for langvarige processer er afgørende for uafbrudte operationer. Med psycopg3's værktøjer som conn.notifies(), kan udviklere implementere robuste notifikationssystemer. Regelmæssige sundhedstjek hjælper med at undgå, at forbindelser ikke reagerer. Eksempler omfatter overvågning af lagersystemer for live-opdateringer for at forhindre udfald.

Lukning og genåbning af notifikationsgeneratoren kombineret med lette SQL-kommandoer sikrer både ydeevne og pålidelighed. Disse teknikker gælder for forskellige use cases, fra logistikopdateringer til økonomiske advarsler. Sådanne strategier hjælper med at beskytte kritiske applikationer mod nedetid, hvilket sikrer en problemfri brugeroplevelse. ⚡