Как установить соединение с Postgres и при потере, восстановить его?

Question

[kondratev-ad]

Есть подписчик на Rabbit, который в логике взаимодействует с Postgres.
Как сделать так, чтобы при запуске скрипта подписчика, он подсоединялся еще и к Postgres и держал это соединение, а при потере соединения, восстанавливал его?

Comments

[Anonymous]

не пиши на python

[kondratev-ad]

А на чем писать?

Answer 1

[Максим Федоров]

Можете пользоваться пулом подключений, если подключения нет — пул автоматом создаст новое (обычно библиотеки, которые позволяют создать pool — хорошо и эффективно создают такие соединения)

Ваша задача только: между сервисами разделить потребности пулов, чтобы размеры пулов в сумме у всех сервисов не были больше количества соединений самого Postgres.

Также посмотрите на pgbouncer или odissey, если у вас нагруженное приложение.

Вообще пересоздать подключение не сложно — вы же создаете первое, также создаете и новое... Предварительно можете пинговать соединение через метод ping() — такой обычно есть у всех библиотек для работы с серверами баз, если оно занято — пересоздать его, Но вам нужно учитывать, что подключение может быть занято долгой транзакцией... потому пул соединений является лучшим решением.

Comments

[kondratev-ad]

У меня еще проблема в том, что я соединение создаю при инициализации приложения, то есть в файле __init__.py, но не понимаю как переподключиться, если в процессе работы postgres перезапустился.
Эксепшен я вылавливаю, но как его перенапрвать в __init__?

[Максим Федоров]

kondratev-ad,

Создание подлючения вынести в отдельный класс, там сделать число попыток — внутри и ловить исключения при каждой попытке

Также при работе дальше вам нужно ловить такое же исключение с потерей соединения и снова в модуле подключения

В целом решение — подключение хранить не в вашем целевом классе, а в своем объекте Connection/DbManager, и вот его уже сохранять в ваш целевой класс...
Через него же делать запросы, при работе запроса он сам ловит исключения и пытается переподключиься