mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-01-02 14:03:06 +00:00
4d68e2abd5
* Changes isOk usage into better patterns with e.g. valueOr / isOkOr * Some other refactoring included * This PR partially addresses #1969
38 lines
1.4 KiB
Nim
38 lines
1.4 KiB
Nim
{.push raises: [].}
|
|
|
|
import chronos, chronicles, results
|
|
import ../../../common/databases/db_postgres, ../../../common/error_handling
|
|
|
|
## Simple query to validate that the postgres is working and attending requests
|
|
const HealthCheckQuery = "SELECT version();"
|
|
const CheckConnectivityInterval = 60.seconds
|
|
const MaxNumTrials = 20
|
|
const TrialInterval = 1.seconds
|
|
|
|
proc checkConnectivity*(
|
|
connPool: PgAsyncPool, onFatalErrorAction: OnFatalErrorHandler
|
|
) {.async.} =
|
|
while true:
|
|
(await connPool.pgQuery(HealthCheckQuery)).isOkOr:
|
|
## The connection failed once. Let's try reconnecting for a while.
|
|
## Notice that the 'pgQuery' proc tries to establish a new connection.
|
|
|
|
block errorBlock:
|
|
## Force close all the opened connections. No need to close gracefully.
|
|
(await connPool.resetConnPool()).isOkOr:
|
|
onFatalErrorAction("checkConnectivity legacy resetConnPool error: " & error)
|
|
|
|
var numTrial = 0
|
|
while numTrial < MaxNumTrials:
|
|
(await connPool.pgQuery(HealthCheckQuery)).isErrOr:
|
|
## Connection resumed. Let's go back to the normal healthcheck.
|
|
break errorBlock
|
|
|
|
await sleepAsync(TrialInterval)
|
|
numTrial.inc()
|
|
|
|
## The connection couldn't be resumed. Let's inform the upper layers.
|
|
onFatalErrorAction("postgres legacy health check error: " & error)
|
|
|
|
await sleepAsync(CheckConnectivityInterval)
|