speckle-server/packages/server/modules/multiregion/index.ts

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'
import { scheduleAutoSyncRegions } from '@/modules/multiregion/tasks/regionSync'

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(),
        await scheduleAutoSyncRegions()
      ]
    }
  },
  async shutdown() {
    await shutdownQueue()
    scheduledTasks.forEach((task) => {
      task.stop()
    })
  }
}

export default multiRegion