Kristaps Fabians Geikins
b406d0e32d
* chore(server): node16 & export maps support for server * moar cleanup * lint fixc
61 lines
1.8 KiB
JavaScript
61 lines
1.8 KiB
JavaScript
'use strict'
|
|
const Environment = require('@speckle/shared/environment')
|
|
const {
|
|
loadMultiRegionsConfig,
|
|
configureKnexClient
|
|
} = require('@speckle/shared/environment/multiRegionConfig')
|
|
const { logger } = require('./observability/logging')
|
|
|
|
const { FF_WORKSPACES_MULTI_REGION_ENABLED } = Environment.getFeatureFlags()
|
|
|
|
const isDevEnv = process.env.NODE_ENV !== 'production'
|
|
|
|
let dbClients
|
|
const getDbClients = async () => {
|
|
if (dbClients) return dbClients
|
|
const maxConnections = parseInt(
|
|
process.env.POSTGRES_MAX_CONNECTIONS_WEBHOOK_SERVICE || '1'
|
|
)
|
|
const connectionAcquireTimeoutMillis = parseInt(
|
|
process.env['POSTGRES_CONNECTION_ACQUIRE_TIMEOUT_MILLIS'] || '16000'
|
|
)
|
|
const connectionCreateTimeoutMillis = parseInt(
|
|
process.env['POSTGRES_CONNECTION_CREATE_TIMEOUT_MILLIS'] || '5000'
|
|
)
|
|
|
|
const configArgs = {
|
|
migrationDirs: [],
|
|
isTestEnv: isDevEnv,
|
|
isDevOrTestEnv: isDevEnv,
|
|
logger,
|
|
maxConnections,
|
|
applicationName: 'speckle_webhook_service',
|
|
connectionAcquireTimeoutMillis,
|
|
connectionCreateTimeoutMillis
|
|
}
|
|
if (!FF_WORKSPACES_MULTI_REGION_ENABLED) {
|
|
const mainClient = configureKnexClient(
|
|
{
|
|
postgres: {
|
|
connectionUri:
|
|
process.env.PG_CONNECTION_STRING ||
|
|
'postgres://speckle:speckle@127.0.0.1/speckle'
|
|
}
|
|
},
|
|
configArgs
|
|
)
|
|
dbClients = { main: mainClient }
|
|
} else {
|
|
const configPath = process.env.MULTI_REGION_CONFIG_PATH || 'multiregion.json'
|
|
const config = await loadMultiRegionsConfig({ path: configPath })
|
|
const clients = [['main', configureKnexClient(config.main, configArgs)]]
|
|
Object.entries(config.regions).map(([key, config]) => {
|
|
clients.push([key, configureKnexClient(config, configArgs)])
|
|
})
|
|
dbClients = Object.fromEntries(clients)
|
|
}
|
|
return dbClients
|
|
}
|
|
|
|
module.exports = getDbClients
|