@lenne.tech/nest-server 11.4.6 → 11.4.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lenne.tech/nest-server",
-  "version": "11.4.6",
+  "version": "11.4.7",
   "description": "Modern, fast, powerful Node.js web framework in TypeScript based on Nest with a GraphQL API and a connection to MongoDB (or other databases).",
   "keywords": [
     "node",

package/src/core/modules/migrate/cli/migrate-cli.ts

@@ -91,7 +91,7 @@ async function listMigrations(options: CliOptions) {
   const stateStore = loadStateStore(options.store);
 
   const runner = new MigrationRunner({
-    migrationsDirectory: options.migrationsDir,
+    migrationsDirectory: path.resolve(process.cwd(), options.migrationsDir),
     stateStore,
   });
 
@@ -255,7 +255,7 @@ async function runDown(options: CliOptions) {
   const stateStore = loadStateStore(options.store);
 
   const runner = new MigrationRunner({
-    migrationsDirectory: options.migrationsDir,
+    migrationsDirectory: path.resolve(process.cwd(), options.migrationsDir),
     stateStore,
   });
 
@@ -270,7 +270,7 @@ async function runUp(options: CliOptions) {
   const stateStore = loadStateStore(options.store);
 
   const runner = new MigrationRunner({
-    migrationsDirectory: options.migrationsDir,
+    migrationsDirectory: path.resolve(process.cwd(), options.migrationsDir),
     stateStore,
   });
 

package/src/core/modules/migrate/helpers/migration.helper.ts

@@ -6,20 +6,64 @@ import * as path from 'path';
  * Migration helper functions for database operations
  */
 
+// Store active connections for auto-cleanup
+const activeConnections = new Set<MongoClient>();
+
+// Track if we're in a migration context
+let inMigrationContext = false;
+
+/**
+ * Mark the start of a migration
+ * @internal Used by migration runner
+ */
+export const _startMigration = () => {
+  inMigrationContext = true;
+};
+
+/**
+ * Mark the end of a migration and close all connections
+ * @internal Used by migration runner
+ */
+export const _endMigration = async () => {
+  inMigrationContext = false;
+  // Close all active connections
+  const promises = Array.from(activeConnections).map((client) => client.close());
+  activeConnections.clear();
+  await Promise.all(promises);
+};
+
 /**
  * Get database connection
  *
+ * When used in migrations, connections are automatically closed after the migration completes.
+ * For manual usage outside migrations, you must close the connection manually.
+ *
  * @param mongoUrl - MongoDB connection URI
  * @returns Promise with database instance
  *
  * @example
  * ```typescript
+ * // In migrations - connection auto-closes after migration
  * const db = await getDb('mongodb://localhost/mydb');
  * await db.collection('users').updateMany(...);
+ *
+ * // Outside migrations - must close manually
+ * const db = await getDb('mongodb://localhost/mydb');
+ * try {
+ *   await db.collection('users').updateMany(...);
+ * } finally {
+ *   await db.client.close();
+ * }
  * ```
  */
 export const getDb = async (mongoUrl: string): Promise<Db> => {
   const client: MongoClient = await MongoClient.connect(mongoUrl);
+
+  // Track connection for auto-cleanup in migrations
+  if (inMigrationContext) {
+    activeConnections.add(client);
+  }
+
   return client.db();
 };
 

package/src/core/modules/migrate/migration-runner.ts

@@ -106,6 +106,8 @@ export class MigrationRunner {
    * Run all pending migrations (up)
    */
   async up(): Promise<void> {
+    const { _endMigration, _startMigration } = await import('./helpers/migration.helper');
+
     const allMigrations = await this.loadMigrationFiles();
     const state = await this.options.stateStore.loadAsync();
     const completedMigrations = (state.migrations || []).map((m) => m.title);
@@ -121,23 +123,32 @@ export class MigrationRunner {
 
     for (const migration of pendingMigrations) {
       console.log(`Running migration: ${migration.title}`);
-      await migration.up();
-
-      // Update state
-      const newState = await this.options.stateStore.loadAsync();
-      const migrations = newState.migrations || [];
-      migrations.push({
-        timestamp: migration.timestamp,
-        title: migration.title,
-      });
-
-      await this.options.stateStore.saveAsync({
-        lastRun: migration.title,
-        migrations,
-        up: () => {},
-      } as any);
 
-      console.log(`✓ Migration completed: ${migration.title}`);
+      // Mark start of migration for auto-cleanup
+      _startMigration();
+
+      try {
+        await migration.up();
+
+        // Update state
+        const newState = await this.options.stateStore.loadAsync();
+        const migrations = newState.migrations || [];
+        migrations.push({
+          timestamp: migration.timestamp,
+          title: migration.title,
+        });
+
+        await this.options.stateStore.saveAsync({
+          lastRun: migration.title,
+          migrations,
+          up: () => {},
+        } as any);
+
+        console.log(`✓ Migration completed: ${migration.title}`);
+      } finally {
+        // Always close connections, even on error
+        await _endMigration();
+      }
     }
 
     console.log('All migrations completed successfully');
@@ -147,6 +158,8 @@ export class MigrationRunner {
    * Rollback the last migration (down)
    */
   async down(): Promise<void> {
+    const { _endMigration, _startMigration } = await import('./helpers/migration.helper');
+
     const state = await this.options.stateStore.loadAsync();
     const completedMigrations = state.migrations || [];
 
@@ -168,17 +181,26 @@ export class MigrationRunner {
     }
 
     console.log(`Rolling back migration: ${migrationToRollback.title}`);
-    await migrationToRollback.down();
 
-    // Update state
-    const newMigrations = completedMigrations.slice(0, -1);
-    await this.options.stateStore.saveAsync({
-      lastRun: newMigrations.length > 0 ? newMigrations[newMigrations.length - 1].title : undefined,
-      migrations: newMigrations,
-      up: () => {},
-    } as any);
+    // Mark start of migration for auto-cleanup
+    _startMigration();
+
+    try {
+      await migrationToRollback.down();
 
-    console.log(`✓ Migration rolled back: ${migrationToRollback.title}`);
+      // Update state
+      const newMigrations = completedMigrations.slice(0, -1);
+      await this.options.stateStore.saveAsync({
+        lastRun: newMigrations.length > 0 ? newMigrations[newMigrations.length - 1].title : undefined,
+        migrations: newMigrations,
+        up: () => {},
+      } as any);
+
+      console.log(`✓ Migration rolled back: ${migrationToRollback.title}`);
+    } finally {
+      // Always close connections, even on error
+      await _endMigration();
+    }
   }
 
   /**