Daniel Gak Anagrov
57 lines
1.6 KiB
TypeScript
57 lines
1.6 KiB
TypeScript
import { moduleLogger } from '@/observability/logging'
|
|
import {
|
|
getValidDefaultProjectRegionKey,
|
|
initializeRegisteredRegionClients as initDb
|
|
} from '@/modules/multiregion/utils/dbSelector'
|
|
import { isMultiRegionEnabled } from '@/modules/multiregion/helpers'
|
|
import type { SpeckleModule } from '@/modules/shared/helpers/typeHelper'
|
|
import {
|
|
initializeRegisteredRegionClients as initBlobs,
|
|
isMultiRegionBlobStorageEnabled
|
|
} from '@/modules/multiregion/utils/blobStorageSelector'
|
|
import {
|
|
initializeQueue,
|
|
shutdownQueue,
|
|
startQueue
|
|
} from '@/modules/multiregion/services/queue'
|
|
import { scheduleStalePreparedTransactionCleanup } from '@/modules/multiregion/tasks/pendingTransactions'
|
|
import type cron from 'node-cron'
|
|
|
|
let scheduledTasks: cron.ScheduledTask[] = []
|
|
|
|
const multiRegion: SpeckleModule = {
|
|
async init({ isInitial }) {
|
|
const isEnabled = isMultiRegionEnabled()
|
|
if (!isEnabled) {
|
|
return
|
|
}
|
|
|
|
moduleLogger.info('🌍 Init multiRegion module')
|
|
|
|
// Init registered region clients
|
|
await initDb()
|
|
// validate default project region key
|
|
await getValidDefaultProjectRegionKey()
|
|
|
|
const isBlobStorageEnabled = isMultiRegionBlobStorageEnabled()
|
|
if (isBlobStorageEnabled) {
|
|
moduleLogger.info('🌍 Init multiRegion blob storage')
|
|
await initBlobs()
|
|
}
|
|
|
|
if (isInitial) {
|
|
await initializeQueue()
|
|
await startQueue()
|
|
scheduledTasks = [await scheduleStalePreparedTransactionCleanup()]
|
|
}
|
|
},
|
|
async shutdown() {
|
|
await shutdownQueue()
|
|
scheduledTasks.forEach((task) => {
|
|
task.stop()
|
|
})
|
|
}
|
|
}
|
|
|
|
export default multiRegion
|