@blackglory/pg-migrations 0.2.1 → 0.2.3

package/lib/es2015/index.js

@@ -1,7 +1,11 @@
 "use strict";
 var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
-    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
 }) : (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
     o[k2] = m[k];

package/lib/es2015/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,4CAAyB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,4CAAyB"}

package/lib/es2018/index.js

@@ -1,7 +1,11 @@
 "use strict";
 var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
-    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
 }) : (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
     o[k2] = m[k];

package/lib/es2018/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,4CAAyB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,4CAAyB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blackglory/pg-migrations",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "A utility for database migrations with pg",
   "keywords": [
     "migration",
@@ -11,7 +11,8 @@
     "sql"
   ],
   "files": [
-    "lib"
+    "lib",
+    "src"
   ],
   "main": "lib/es2018/index.js",
   "types": "lib/es2018/index.d.ts",
@@ -46,26 +47,26 @@
     }
   },
   "devDependencies": {
-    "@commitlint/cli": "^16.2.1",
-    "@commitlint/config-conventional": "^16.2.1",
+    "@commitlint/cli": "^17.0.3",
+    "@commitlint/config-conventional": "^17.0.3",
     "@types/jest": "^27.4.0",
-    "@types/pg": "^8.6.4",
-    "@typescript-eslint/eslint-plugin": "^5.12.0",
-    "@typescript-eslint/parser": "^5.12.0",
-    "eslint": "^8.9.0",
+    "@types/pg": "^8.6.5",
+    "@typescript-eslint/eslint-plugin": "^5.32.0",
+    "@typescript-eslint/parser": "^5.32.0",
+    "eslint": "^8.21.0",
     "husky": "^4.3.0",
     "jest": "^27.5.1",
     "npm-run-all": "^4.1.5",
     "pg": "^8.7.3",
     "rimraf": "^3.0.2",
-    "standard-version": "^9.3.0",
+    "standard-version": "^9.5.0",
     "ts-jest": "^27.1.3",
     "tscpaths": "^0.0.9",
-    "typescript": "^4.5.5"
+    "typescript": "^4.7.4"
   },
   "dependencies": {
-    "@blackglory/errors": "^2.2.0",
-    "@blackglory/types": "^0.6.5"
+    "@blackglory/errors": "^2.2.2",
+    "@blackglory/types": "^1.2.1"
   },
   "peerDependencies": {
     "pg": "8"

package/src/index.ts

@@ -0,0 +1 @@
+export * from './migrate'

package/src/migrate.ts

@@ -0,0 +1,135 @@
+import type { Client } from 'pg'
+import { isFunction } from '@blackglory/types'
+import { assert } from '@blackglory/errors'
+
+export interface IMigration {
+  version: number
+  up: string | ((client: Client) => PromiseLike<void>)
+  down: string | ((client: Client) => PromiseLike<void>)
+}
+
+export async function migrate(
+  client: Client
+, migrations: IMigration[]
+, targetVersion = getMaximumVersion(migrations)
+, migrationsTable: string = 'migrations'
+, advisoryLockKey: bigint = BigInt('-9223372036854775808') // The smallest bigint for postgres
+): Promise<void> {
+  const maxVersion = getMaximumVersion(migrations)
+  await lock(client, advisoryLockKey)
+  try {
+    while (true) {
+      const currentVersion = await getDatabaseVersion(client, migrationsTable)
+      if (maxVersion < currentVersion) {
+        break
+      } else {
+        if (currentVersion === targetVersion) {
+          break
+        } else if (currentVersion < targetVersion) {
+          await upgrade()
+        } else {
+          await downgrade()
+        }
+      }
+    }
+  } finally {
+    await unlock(client, advisoryLockKey)
+  }
+
+  async function upgrade() {
+    await client.query('BEGIN')
+    const currentVersion: number = await getDatabaseVersion(client, migrationsTable)
+    const targetVersion = currentVersion + 1
+    try {
+      const migration = migrations.find(x => x.version === targetVersion)
+      assert(migration, `Cannot find migration for version ${targetVersion}`)
+
+      if (isFunction(migration.up)) {
+        await migration.up(client)
+      } else {
+        await client.query(migration.up)
+      }
+      await setDatabaseVersion(client, migrationsTable, targetVersion)
+      await client.query('COMMIT')
+    } catch (e) {
+      console.error(`Upgrade from version ${currentVersion} to version ${targetVersion} failed.`)
+      await client.query('ROLLBACK')
+      throw e
+    }
+  }
+
+  async function downgrade() {
+    await client.query('BEGIN')
+    const currentVersion = await getDatabaseVersion(client, migrationsTable)
+    const targetVersion = currentVersion - 1
+    try {
+      const migration = migrations.find(x => x.version === currentVersion)
+      assert(migration, `Cannot find migration for version ${targetVersion}`)
+
+      if (isFunction(migration.down)) {
+        await migration.down(client)
+      } else {
+        await client.query(migration.down)
+      }
+      await setDatabaseVersion(client, migrationsTable, targetVersion)
+      await client.query('COMMIT')
+    } catch (e) {
+      console.error(`Downgrade from version ${currentVersion} to version ${targetVersion} failed.`)
+      await client.query('ROLLBACK')
+      throw e
+    }
+  }
+}
+
+function getMaximumVersion(migrations: IMigration[]): number {
+  return migrations.reduce((max, cur) => Math.max(cur.version, max), 0)
+}
+
+async function getDatabaseVersion(client: Client, migrationTable: string): Promise<number> {
+  await ensureMigrationsTable(client, migrationTable)
+
+  const result = await client.query<{ schema_version: number }>(`
+    SELECT schema_version
+      FROM "${migrationTable}";
+  `)
+  if (result.rows.length) {
+    return result.rows[0].schema_version
+  } else {
+    await client.query(`
+      INSERT INTO "${migrationTable}" (schema_version)
+      VALUES (0);
+    `)
+    return 0
+  }
+}
+
+async function ensureMigrationsTable(client: Client, migrationTable: string): Promise<void> {
+  await client.query(`
+    CREATE TABLE IF NOT EXISTS "${migrationTable}" (
+      schema_version INTEGER NOT NULL
+    );
+  `)
+}
+
+async function setDatabaseVersion(
+  client: Client
+, migrationTable: string
+, version: number
+): Promise<void> {
+  await client.query(`
+    UPDATE ${migrationTable}
+       SET schema_version = ${version};
+  `)
+}
+
+async function lock(client: Client, key: bigint): Promise<void> {
+  await client.query(`
+    SELECT pg_advisory_lock(${key});
+  `)
+}
+
+async function unlock(client: Client, key: bigint): Promise<void> {
+  await client.query(`
+    SELECT pg_advisory_unlock(${key});
+  `)
+}