如何在多次AWS Lambda调用中复用PostgreSQL数据库连接客户端？

Lambda的执行环境会在多次调用间复用（即“warm start”），利用这个特性可以将数据库连接/连接池放在全局作用域，避免每次调用都新建连接。以下是具体实现方案：

核心思路 #

• 将连接池（推荐）或连接对象放在Lambda函数的全局作用域，仅在冷启动时初始化
• 每次调用前验证连接可用性，避免使用失效连接
• 优先使用pg包的Pool而非单连接，更好适配Lambda的并发场景

代码实现（连接池方案） #

const { Pool } = require('pg');

// 全局作用域存储连接池，冷启动时初始化，warm start时复用
let pgPool;

exports.handler = async (event) => {
  // 初始化连接池（仅第一次调用/冷启动时执行）
  if (!pgPool) {
    pgPool = new Pool({
      user: '你的数据库用户名',
      host: '你的数据库主机地址',
      database: '你的数据库名',
      password: '你的数据库密码',
      port: 5432,
      // 连接池参数按需调整：max为最大连接数，需匹配数据库允许的连接上限
      max: 5,
      // 空闲连接超时时间，避免数据库端主动断开闲置连接
      idleTimeoutMillis: 30000,
      // 连接超时时间
      connectionTimeoutMillis: 2000,
    });

    // 验证连接池可用性，避免初始化失败
    try {
      const client = await pgPool.connect();
      await client.query('SELECT 1');
      client.release();
      console.log('连接池初始化成功');
    } catch (err) {
      console.error('连接池初始化失败', err);
      throw err;
    }
  }

  // 执行数据库查询
  try {
    const queryResult = await pgPool.query(
      'SELECT * FROM your_table WHERE id = $1',
      [event.id]
    );
    return {
      statusCode: 200,
      body: JSON.stringify(queryResult.rows),
    };
  } catch (err) {
    console.error('查询执行失败', err);
    // 若连接失效，重置连接池以便下次调用重新初始化
    if (err.message.includes('connection terminated') || err.message.includes('socket hang up')) {
      pgPool = null;
    }
    return {
      statusCode: 500,
      body: JSON.stringify({ error: err.message }),
    };
  }
};

关键注意事项 #

• 并发适配：Lambda并发数较高时，需调整max参数，确保数据库的最大连接数能容纳所有Lambda实例的连接池总连接数，避免连接被拒绝
• 失效处理：数据库可能会主动断开闲置连接，需在捕获到连接错误时重置连接池，让下次调用重新初始化
• 资源回收：无需在handler中手动关闭连接池，Lambda执行环境销毁时会自动回收资源；手动关闭会导致warm start时重复初始化，失去复用意义
• 单连接方案（不推荐）：如果业务并发极低，可使用单连接，但同一时间只能处理一个请求，高并发下会导致性能瓶颈，代码示例如下：

const { Client } = require('pg');

let pgClient;

exports.handler = async (event) => {
  if (!pgClient || !pgClient._connected) {
    pgClient = new Client({ /* 连接参数 */ });
    await pgClient.connect();
    await pgClient.query('SELECT 1');
  }

  try {
    const result = await pgClient.query('SELECT * FROM your_table WHERE id = $1', [event.id]);
    return { statusCode: 200, body: JSON.stringify(result.rows) };
  } catch (err) {
    pgClient = null; // 连接失效时重置
    throw err;
  }
};

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