Npgsql.EntityFrameworkCore.PostgreSQL 中共享 DbContext 连接与类型映射的实践

在使用 Npgsql.EntityFrameworkCore.PostgreSQL 进行数据库操作时，开发者可能会遇到需要共享数据库连接和事务的场景。本文将深入探讨如何正确地在多个 DbContext 实例之间共享连接，同时确保自定义类型映射（如枚举类型）能够正常工作。

连接共享的基本原理

在 Entity Framework Core 中，可以通过 UseTransaction 方法在多个 DbContext 实例间共享事务。要实现这一点，首先需要确保这些 DbContext 实例使用的是同一个数据库连接。

Npgsql 提供了 NpgsqlDataSource 类来管理数据库连接池和类型映射。理论上，通过共享 NpgsqlDataSource 实例，可以确保所有创建的连接都具备相同的配置。

两种共享方式的对比

方式一：直接共享 NpgsqlDataSource

var builder = new NpgsqlDataSourceBuilder(connectionString);
builder.MapEnum<ClientType>("client_type");  // 配置枚举映射
var sharedDataSource = builder.Build();

// 在 DbContext 配置中使用共享的 DataSource
services.AddDbContext<MainDbContext>(options => 
    options.UseNpgsql(sharedDataSource));

这种方式能够确保类型映射正常工作，生成的 SQL 会正确使用枚举类型（如 'old'::client_type）。但是，它不能自动共享连接，每个 DbContext 会从连接池获取独立的连接。

方式二：共享 NpgsqlConnection

// 创建共享连接
var sharedConnection = sharedDataSource.CreateConnection();

// 在 DbContext 配置中使用共享连接
services.AddDbContext<MainDbContext>(options => 
    options.UseNpgsql(sharedConnection));

这种方式确实实现了连接共享，但会出现类型映射失效的问题。生成的 SQL 会使用枚举的整数值（如 1）而不是类型化的表示，可能导致数据库操作失败。

问题根源分析

这个问题的根本原因在于 EF Core 的类型映射系统与 Npgsql 的类型映射系统的交互方式：

1. 当使用 NpgsqlDataSource 配置 DbContext 时，EF Core 会从 DataSource 中获取类型映射信息
2. 当直接使用 NpgsqlConnection 时，EF Core 无法获取到 DataSource 中的映射配置
3. 虽然底层的 Npgsql 连接确实包含了正确的映射（因为它来自配置好的 DataSource），但 EF Core 层面缺少这些信息

解决方案与最佳实践

目前推荐的解决方案是：

1. 对于类型映射：使用全局类型映射器进行配置（尽管它被标记为过时，但在当前版本中仍是可靠的选择）

// 在应用程序启动时配置全局映射
NpgsqlConnection.GlobalTypeMapper.MapEnum<ClientType>("client_type");

2. 对于连接共享：继续使用 NpgsqlDataSource.CreateConnection() 创建共享连接

3. 对于事务管理：通过共享的连接创建事务，并在多个 DbContext 实例间共享

using var connection = sharedDataSource.CreateConnection();
connection.Open();
using var transaction = connection.BeginTransaction();

using var dbContext1 = new MainDbContext(/* 使用共享连接的配置 */);
dbContext1.Database.UseTransaction(transaction);

using var dbContext2 = new ReadOnlyMainDbContext(/* 使用共享连接的配置 */);
dbContext2.Database.UseTransaction(transaction);

未来改进方向

Npgsql 团队已经意识到这个问题，并计划在未来版本中改进类型映射系统。主要改进方向包括：

1. 使 EF Core 不再依赖从 NpgsqlDataSource 获取类型映射
2. 提供更优雅的方式来处理共享连接场景下的类型映射
3. 最终淘汰全局类型映射器，提供更好的替代方案

总结

在 Npgsql.EntityFrameworkCore.PostgreSQL 中实现 DbContext 间的连接共享需要注意类型映射的配置问题。当前可以通过全局类型映射器作为临时解决方案，同时期待未来版本能提供更优雅的集成方式。开发者应当根据实际需求权衡连接共享的必要性与类型映射的复杂性，选择最适合自己应用场景的方案。