mirror of
https://github.com/waku-org/nwaku.git
synced 2025-01-30 16:47:48 +00:00
fd6a71cdd7
* bump_dependencies.md: add nim-results dependency * change imports stew/results to results * switching to Nim 2.0.8 * waku.nimble: reflect the requirement nim 1.6.0 to 2.0.8 Adding --mm:refc as nim 2.0 enables a new garbage collector that we're not yet ready to support * adapt waku code to Nim 2.0 * gcsafe adaptations because Nim 2.0 is more strict
39 lines
1.4 KiB
Nim
39 lines
1.4 KiB
Nim
{.push raises: [].}
|
|
|
|
import chronos, 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 'exec' 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 resetConnPool error: " & error)
|
|
|
|
var numTrial = 0
|
|
while numTrial < MaxNumTrials:
|
|
let res = await connPool.pgQuery(HealthCheckQuery)
|
|
if res.isOk():
|
|
## 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 health check error: " & error)
|
|
|
|
await sleepAsync(CheckConnectivityInterval)
|