AWS Lambda连接PostgreSQL令牌过期：服务注入配置疑问

你的核心问题出在连接与Token的创建时机、服务生命周期匹配两个方面：

1. Token生成时机过早：你在DatabaseContext的构造函数中生成Token并创建连接实例，一旦DatabaseContext被复用，Token就不会重新生成，15分钟过期后自然无法连接数据库。
2. Lambda上下文复用导致实例缓存：虽然DatabaseContext注册为Transient，但依赖它的MyDbService是Scoped。AWS Lambda的执行环境会被复用，Scoped服务会在复用的上下文中被长时间保留，导致DatabaseContext实例不会被频繁重建，Token也就不会更新——这就是日志中"Token Generated"只打印一次的原因。
3. 连接管理逻辑缺陷：直接持有固定的NpgsqlConnection实例，复用过期Token的连接，无法在需要时生成新的有效连接。

1. 重构DatabaseContext：按需生成Token与连接 #

去掉构造函数中的连接初始化，改为每次需要数据库操作时生成新的Token和连接：

public class DatabaseContext : IDisposable
{
    private bool disposed = false;
    private readonly string _databaseEndpoint;
    private readonly int _databasePort;
    private readonly string _databaseUsername;
    private readonly string _databaseName;

    // 注入必要的配置参数（建议从环境变量获取）
    public DatabaseContext(string databaseEndpoint, int databasePort, string databaseUsername, string databaseName)
    {
        _databaseEndpoint = databaseEndpoint;
        _databasePort = databasePort;
        _databaseUsername = databaseUsername;
        _databaseName = databaseName;
    }

    // 每次调用都生成新的Token和连接
    public async Task<NpgsqlConnection> GetNewConnectionAsync()
    {
        var token = RDSAuthTokenGenerator.GenerateAuthToken(_databaseEndpoint, _databasePort, _databaseUsername);
        Console.WriteLine("Token Generated: " + token);
        
        var connectionString = $"Host={_databaseEndpoint};Port={_databasePort};Username={_databaseUsername};Password={token};Database={_databaseName};SSL Mode=Require;Trust Server Certificate=true";
        var connection = new NpgsqlConnection(connectionString);
        await connection.OpenAsync();
        return connection;
    }

    // Dispose逻辑无需管理固定连接，仅标记状态
    ~DatabaseContext()
    {
        Dispose(false);
    }

    public void Dispose()
    {
        Dispose(true);
        GC.SuppressFinalize(this);
    }

    protected virtual void Dispose(bool disposing)
    {
        if (disposed) return;
        disposed = true;
    }
}

2. 修改MyDbService：使用using管理连接生命周期 #

每次数据库操作都获取新连接，并通过using自动释放：

public class MyDbService : IMyDbService
{
    private readonly DatabaseContext _db;

    public MyDbService(DatabaseContext db)
    {
        _db = db;
    }

    public async Task<long> GetUserDetails(string userId)
    {
        // 每次操作获取新连接，using块自动关闭释放
        using (var connection = await _db.GetNewConnectionAsync())
        {
            using (var command = new NpgsqlCommand("SELECT id FROM users WHERE user_id = @userId", connection))
            {
                command.Parameters.AddWithValue("@userId", userId);
                var result = await command.ExecuteScalarAsync();
                return result != null ? Convert.ToInt64(result) : 0;
            }
        }
    }
}

3. 调整服务注册：匹配Lambda执行特性 #

将MyDbService改为Transient，避免Lambda上下文复用导致的实例缓存：

// 注册DatabaseContext，注入配置参数（示例从环境变量读取）
services.AddTransient<DatabaseContext>(sp => 
    new DatabaseContext(
        Environment.GetEnvironmentVariable("DB_ENDPOINT"),
        int.Parse(Environment.GetEnvironmentVariable("DB_PORT")),
        Environment.GetEnvironmentVariable("DB_USER"),
        Environment.GetEnvironmentVariable("DB_NAME")
    )
);

// MyDbService改为Transient，确保每次请求都能获取新实例
services.AddTransient<IMyDbService, MyDbService>();

• AWS Lambda的执行环境会复用，Scoped服务在复用周期内不会被销毁，这会导致依赖的实例长期保留。改用Transient可以确保每次Lambda调用都能生成新的DatabaseContext实例，进而生成新的Token。
• 使用using块管理连接可以确保连接及时关闭释放，避免连接泄漏，同时每次操作都使用新的有效Token连接数据库。

内容的提问来源于stack exchange，提问作者Ctrl_Alt_Defeat