6824988676

NEW 1	import { Inject, Injectable, OnModuleInit } from '@nestjs/common';	×
NEW 2	import { ILoggingService, LoggingService } from '@/logging/logging.interface';	×
NEW 3	import * as shift from 'postgres-shift';	×
NEW 4	import postgres from 'postgres';	×
5
6	/**
7	* The {@link PostgresDatabaseMigrationHook} is a Module Init hook meaning
8	* that it will be executed once the dependencies are resolved.
9	*
10	* This happens before the Application Bootstraps, so route listeners are not
11	* initiated and potentially generating queries.
12	*/
13	@Injectable({})
NEW 14	export class PostgresDatabaseMigrationHook implements OnModuleInit {	×
NEW 15	private static LOCK_MAGIC_NUMBER = 132;	×
16
17	constructor(
NEW 18	@Inject('DB_INSTANCE') private readonly sql: postgres.Sql,	×
NEW 19	@Inject(LoggingService) private readonly loggingService: ILoggingService,	×
20	) {}
21
22	async onModuleInit() {
NEW 23	this.loggingService.info('Checking migrations');	×
NEW 24	try {	×
25	// Acquire lock to perform a migration.
26	// If the lock is not acquired, then a migration is being executed by another instance.
27	// Migrations should strive to be idempotent as they can be executed by multiple instances
28	// on the same database.
NEW 29	await this	×
30	.sql`SELECT pg_advisory_lock(${PostgresDatabaseMigrationHook.LOCK_MAGIC_NUMBER})`;
31	// Perform migration
NEW 32	await shift({ sql: this.sql });	×
NEW 33	this.loggingService.info('Pending migrations executed');	×
34	} catch (e) {
35	// If there's an error performing a migration, we should throw the error
36	// and prevent the service from starting
NEW 37	this.loggingService.error(e);	×
NEW 38	throw e;	×
39	} finally {
40	// the lock should be released if the migration completed (successfully or not)
NEW 41	await this	×
42	.sql`SELECT pg_advisory_unlock(${PostgresDatabaseMigrationHook.LOCK_MAGIC_NUMBER})`;
43	}
44	}
45	}

safe-global / safe-client-gateway / 6824988676

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous