Ozone: add option to run migrations from service entrypoint (bluesky-…

…social#2789) ozone: add option to run migrations from service entrypoint
lindner · Sep 5, 2024 · f7cbfa4 · f7cbfa4
1 parent c46dc91
commit f7cbfa4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/services/ozone/api.js b/services/ozone/api.js
@@ -25,6 +25,7 @@ const {
   envToSecrets,
   readEnv,
   httpLogger,
+  Database,
 } = require('@atproto/ozone')
 
 const main = async () => {
@@ -62,6 +63,16 @@ const main = async () => {
       ? new MultiImageInvalidator(imgInvalidators)
       : imgInvalidators[0]
 
+  const migrate = process.env.OZONE_DB_MIGRATE === '1'
+  if (migrate) {
+    const db = new Database({
+      url: cfg.db.postgresUrl,
+      schema: cfg.db.postgresSchema,
+    })
+    await db.migrateToLatestOrThrow()
+    await db.close()
+  }
+
   const ozone = await OzoneService.create(cfg, secrets, { imgInvalidator })
 
   await ozone.start()