Iain Sproat
7e6f31ad66
* chore(health): add dedicated liveness and readiness endpoints - provides more flexibility in defining what constitutes readiness and liveness * add tests for /liveness and /readiness * Prevent liveness and readiness endpoints being exposed externally by redirecting their routes
85 lines
2.4 KiB
TypeScript
85 lines
2.4 KiB
TypeScript
import { registerOrUpdateScope, registerOrUpdateRole } from '@/modules/shared'
|
|
import { moduleLogger } from '@/logging/logging'
|
|
import {
|
|
setupResultListener,
|
|
shutdownResultListener
|
|
} from '@/modules/core/utils/dbNotificationListener'
|
|
import * as mp from '@/modules/shared/utils/mixpanel'
|
|
import { Optional, SpeckleModule } from '@/modules/shared/helpers/typeHelper'
|
|
|
|
import staticRest from '@/modules/core/rest/static'
|
|
import uploadRest from '@/modules/core/rest/upload'
|
|
import downloadRest from '@/modules/core/rest/download'
|
|
import diffUpload from '@/modules/core/rest/diffUpload'
|
|
import diffDownload from '@/modules/core/rest/diffDownload'
|
|
import healthRest from '@/modules/core/rest/health'
|
|
import scopes from '@/modules/core/scopes'
|
|
import roles from '@/modules/core/roles'
|
|
import Redis from 'ioredis'
|
|
import { createRedisClient } from '@/modules/shared/redis/redis'
|
|
import { getRedisUrl } from '@/modules/shared/helpers/envHelper'
|
|
import { UninitializedResourceAccessError } from '@/modules/shared/errors'
|
|
|
|
let genericRedisClient: Optional<Redis> = undefined
|
|
|
|
const coreModule: SpeckleModule<{
|
|
getGenericRedis: () => Redis
|
|
}> = {
|
|
async init(app, isInitial) {
|
|
moduleLogger.info('💥 Init core module')
|
|
|
|
// Initialize the static route
|
|
staticRest(app)
|
|
|
|
// Initialize the health check route
|
|
healthRest(app)
|
|
|
|
// Initialises the two main bulk upload/download endpoints
|
|
uploadRest(app)
|
|
downloadRest(app)
|
|
|
|
// Initialises the two diff-based upload/download endpoints
|
|
diffUpload(app)
|
|
diffDownload(app)
|
|
|
|
// Register core-based scoeps
|
|
for (const scope of scopes) {
|
|
await registerOrUpdateScope(scope)
|
|
}
|
|
|
|
// Register core-based roles
|
|
for (const role of roles) {
|
|
await registerOrUpdateRole(role)
|
|
}
|
|
|
|
if (isInitial) {
|
|
// Setup global pg notification listener
|
|
setupResultListener()
|
|
|
|
// Init mp
|
|
mp.initialize()
|
|
|
|
// Generic redis client
|
|
genericRedisClient = createRedisClient(getRedisUrl(), {})
|
|
}
|
|
},
|
|
async shutdown() {
|
|
await shutdownResultListener()
|
|
|
|
if (genericRedisClient) {
|
|
await genericRedisClient.quit()
|
|
}
|
|
},
|
|
/**
|
|
* A general purpose redis client that can be used after safely all modules are initialized
|
|
*/
|
|
getGenericRedis() {
|
|
if (!genericRedisClient) {
|
|
throw new UninitializedResourceAccessError('Generic redis client not initialized')
|
|
}
|
|
return genericRedisClient
|
|
}
|
|
}
|
|
|
|
export = coreModule
|