Memastikan Stabilitas pada Pendengar Notifikasi PostgreSQL yang Berjalan Lama dengan Psycopg3

Memastikan Notifikasi yang Andal pada Listener PostgreSQL yang Sudah Berjalan Lama

Implementasi backend menggunakan Python dan psycopg3 untuk menangani koneksi database yang berjalan lama

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()

Pendekatan Alternatif: Menggunakan Asynchronous psycopg3 untuk Meningkatkan Responsif

Implementasi asinkron menggunakan asyncio dan psycopg3 Python

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())

Pengujian Unit untuk Kekokohan

Unit Python menguji logika backend menggunakan 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()

FAQ tentang Mengelola Pendengar Notifikasi PostgreSQL

1. Apa tujuannya conn.notifies() di psikopg3?
2. conn.notifies() digunakan untuk mengambil notifikasi yang dikirim oleh server PostgreSQL, memungkinkan penanganan kejadian real-time dalam aplikasi.
3. Bisa LISTEN perintah kehilangan pesan selama koneksi ulang?
4. Tidak, PostgreSQL mem-buffer notifikasi, sehingga pesan tidak hilang selama koneksi ulang. Namun penanganan yang tepat terhadapnya notifies generator diperlukan untuk memastikan pemrosesan yang lancar.
5. Mengapa saya harus menggunakan autocommit=True?
6. Pengaturan autocommit=True memungkinkan koneksi untuk segera menerapkan perintah seperti LISTEN tanpa menunggu komitmen eksplisit, meningkatkan daya tanggap.
7. Bagaimana saya bisa melakukan pemeriksaan kesehatan dalam jangka panjang notifies proses?
8. Anda dapat secara berkala menjalankan kueri ringan seperti SELECT 1 untuk memastikan koneksi tetap responsif.
9. Apa praktik terbaik untuk membersihkan koneksi database?
10. Menggunakan a with pernyataan atau manajer konteks Python memastikan bahwa koneksi ditutup dengan benar, menghindari kebocoran sumber daya.
11. Bagaimana cara menangani pengecualian batas waktu di conn.notifies()?
12. Membungkus conn.notifies() di blok coba-kecuali untuk menangkap pengecualian batas waktu dan menanganinya dengan baik, seperti dengan mencatat atau mencoba ulang.
13. Apakah psycopg3 mendukung operasi asinkron untuk notifikasi?
14. Ya, psycopg3 menawarkan API asinkron melalui AsyncConnection, yang ideal untuk aplikasi non-pemblokiran dan dapat diskalakan.
15. Apa yang terjadi jika saya tidak menutupnya notifies generator?
16. Gagal menutup generator dapat mengakibatkan kebocoran memori atau sumber daya terhenti, terutama pada proses yang berjalan lama.
17. Bisakah notifikasi terlewatkan selama a pg_sleep() operasi?
18. Ya, notifikasi yang dihasilkan selama periode tidur mungkin terlewatkan jika tidak di-buffer, itulah sebabnya diperlukan penanganan yang tepat LISTEN perintah sangatlah penting.
19. Apakah aman menggunakan kembali koneksi yang sama untuk beberapa notifikasi?
20. Ya, selama pemeriksaan kondisi dan koneksi ulang yang tepat dikelola, penggunaan kembali koneksi yang sama akan efisien dan hemat sumber daya.
21. Bagaimana cara menguji keandalan sistem notifikasi saya?
22. Tulis pengujian unit menggunakan perpustakaan seperti unittest.mock untuk mensimulasikan notifikasi dan perilaku database tanpa bergantung pada server langsung.

Memastikan Mendengarkan Notifikasi yang Andal

Menjaga kesehatan koneksi untuk proses yang berjalan lama sangat penting agar pengoperasian tidak terganggu. Dengan alat psycopg3 seperti samb.notify(), pengembang dapat menerapkan sistem notifikasi yang kuat. Pemeriksaan kesehatan rutin membantu menghindari koneksi yang tidak responsif. Contohnya termasuk memantau sistem inventaris untuk pembaruan langsung guna mencegah pemadaman.

Menutup dan membuka kembali generator notifikasi, dikombinasikan dengan perintah SQL ringan, memastikan kinerja dan keandalan. Teknik-teknik ini berlaku untuk berbagai kasus penggunaan, mulai dari pembaruan logistik hingga peringatan keuangan. Strategi tersebut membantu melindungi aplikasi penting dari downtime, sehingga memastikan pengalaman pengguna yang lancar. ⚡