MartenDB 7.x版本中的数据库连接泄漏问题分析

问题背景

MartenDB是一个基于PostgreSQL的.NET文档数据库和事件存储库。在从6.4.1版本升级到7.1.0版本后，用户报告了一个严重的数据库连接泄漏问题。当执行JavaScript转换迁移时，系统会逐渐耗尽所有数据库连接，最终导致"Npgsql.PostgresException 53300: remaining connection slots are reserved for roles with the SUPERUSER attribute"错误。

问题表现

在6.4.1版本中运行正常的代码，在7.1.0版本中会出现以下症状：

1. 数据库连接数持续增加
2. 最终耗尽所有可用连接池
3. 抛出PostgreSQL连接限制异常

问题代码分析

问题出现在以下典型使用模式中：

public override async Task ExecuteAsync(IDatabaseConfiguration databaseConfiguration, CancellationToken cancellationToken)
{
    var store = BuildDocumentStore(
        databaseConfiguration,
        x => x.UseJavascriptTransformsAndPatching(t =>
    {
        t.LoadJavascript(Name, JsScript);
    }));

    await using IDocumentSession session = store.LightweightSession();
    await DoTransformsAsync(session, cancellationToken);
    await session.SaveChangesAsync(cancellationToken);
}

这段代码在循环中被反复调用，每次都会：

1. 创建一个新的DocumentStore实例
2. 打开一个会话
3. 执行转换操作
4. 保存更改

问题根源

在MartenDB 7.x版本中，DocumentStore的生命周期管理变得更加严格。虽然代码中正确使用了await using来处理会话(IDocumentSession)的释放，但却忽略了DocumentStore本身的释放。

正确的做法应该是：

await using var store = BuildDocumentStore(...);

深入理解

DocumentStore的生命周期

DocumentStore是MartenDB中的重量级对象，它：

1. 管理着与PostgreSQL的连接池
2. 维护着模式定义和映射信息
3. 缓存着查询计划

在6.x版本中，DocumentStore的资源管理可能不够严格，导致即使没有显式释放也能正常工作。但在7.x版本中，MartenDB加强了对资源生命周期的控制，要求开发者必须显式管理DocumentStore的释放。

连接泄漏机制

每次创建DocumentStore而不释放时：

1. 会建立新的Npgsql连接池
2. 这些连接不会被自动回收
3. PostgreSQL服务器端的连接数会持续增加
4. 最终达到PostgreSQL的max_connections限制

最佳实践建议

1. DocumentStore应该是单例：在大多数应用中，DocumentStore应该作为单例存在，通过依赖注入容器管理其生命周期。

2. 临时使用时的资源释放：如果确实需要临时创建DocumentStore（如迁移脚本中），必须确保使用using或await using正确释放。

3. 会话管理：会话(IDocumentSession)通常是轻量级的，可以频繁创建和释放，但仍需确保正确释放。

4. 连接池监控：在生产环境中，应该监控数据库连接数，及时发现潜在的泄漏问题。

迁移脚本优化建议

对于需要频繁执行JavaScript转换的场景，可以考虑以下优化：

await using var store = BuildDocumentStore(configuration);

foreach (var transform in transforms)
{
    await using var session = store.LightweightSession();
    // 执行转换
    await session.SaveChangesAsync(cancellationToken);
}

这种方式：

1. 只创建一个DocumentStore
2. 对每个转换创建独立的会话
3. 确保所有资源正确释放

总结

MartenDB 7.x版本对资源管理更加严格，这实际上是框架成熟的表现。开发者需要特别注意DocumentStore的生命周期管理，特别是在迁移脚本等特殊场景中。通过遵循正确的资源释放模式，可以避免连接泄漏问题，确保应用稳定运行。