如何在nestjs服务中关闭数据库连接？

Question

在我的数据库模块中，我正在创建与我的数据库的连接。有了这个设置，我确实得到了错误

工作进程未能正常退出，已被强制退出。这可能是由于不正确的拆卸导致测试泄漏造成的。尝试使用 --runInBand --detectOpenHandles 运行以查找泄漏。

在我的 e2e 测试中。

我想我必须关闭连接。但是如何在我的数据库模块中的onModuleDestroy() 中运行client.close？

database.module.ts

import { Module } from '@nestjs/common'
import { MongoClient, Db, Logger } from 'mongodb'

@Module({
    providers: [
        {
            provide: 'DATABASE_CONNECTION',
            useFactory: async (): Promise<Db> => {
                const mongo = 'mongodb://localhost:27017'
                const database = 'testing'

                try {
                    Logger.setLevel('debug')

                    const client = await MongoClient.connect(mongo, {
                        useNewUrlParser: true,
                        useUnifiedTopology: true,
                    });

                    const db = client.db(database)
                    return db
                } catch (error) {
                    throw error
                }
            }
        },
        {
            provide: 'DATABASE_CLIENT',
            useFactory: () => true // how do I get client of the above provider?
        }
    ],
    exports: ['DATABASE_CONNECTION', 'DATABASE_CLIENT']
})

export class DatabaseModule {
    constructor(
        @Inject('DATABASE_CLIENT')
        private client: Db
    ) {}

    async onModuleDestroy() {
        console.log(this.client);
        // await this.client.close()
    }
}

Answer 1

打开的数据库连接是导致此问题的常见原因。一个模块需要自己清理，这尤其适用于一个模块可以被多次实例化的测试。

client 需要保留引用以进行清理。这可以通过单类服务来完成：

@Injectable()
class DatabaseConnection {
  async onModuleInit() {
    // ...same as DATABASE_CONNECTION factory
    this.client = client;
    this.db = client.db(database);
  }
  async onModuleDestroy() {
    await this.client.close()
  } 
}

或者使用两个工厂提供者，其中一个保留对client 的引用：

@Module({
  providers: [
    {
      provide: 'DATABASE_CLIENT',
      useFactory: () => ({ client: null })
    },
    {
      provide: 'DATABASE_CONNECTION',
      inject: ['DATABASE_CLIENT'],
      useFactory: async (dbClient) => {
        ...
        dbClient.client = client
        const db = client.db(database)
        return db;
      }
    }
  ]
  ...
})
export class DatabaseModule {
  constructor(@Inject('DATABASE_CLIENT') private dbClient) {}

  async onModuleDestroy() {
    await this.dbClient.client.close()
  } 
}