访问RDS PostgreSQL最佳实践

步骤一：设置数据库白名单

步骤二：在函数中访问PostgreSQL数据库

1. 登录函数计算控制台，在函数列表找到目标函数，在函数详情页，单击代码页签，在代码编辑器根据数据库表编写如下示例代码。

   from flask import Flask, jsonify
   import psycopg2  # 需先安装psycopg2库
   import os
   import sys
   from datetime import datetime
   import logging
   
   
   app = Flask(__name__)
   
   # 全局变量用于存储 PostgreSQL 单例连接
   _postgresql_connection = None
   
   # 创建数据库连接（单例模式）
   def getConnection():
       global _postgresql_connection
       try:
           # 如果连接已经存在且未断开，直接返回
           if _postgresql_connection is not None:
               try:
                   # 测试连接是否有效（简单命令测试）
                   with _postgresql_connection.cursor() as cursor:
                       cursor.execute("SELECT 1")  # 简单查询测试连接状态
                       result = cursor.fetchone()
                       if result and result[0] == 1:
                           return _postgresql_connection
               except psycopg2.OperationalError:
                   # 如果连接断开，重置连接
                   _postgresql_connection = None
   
           # 如果连接不存在或已断开，重新创建连接
           _postgresql_connection = psycopg2.connect(
               database=os.environ['DATABASE'],
               user=os.environ['USER'],
               password=os.environ['PASSWORD'],
               host=os.environ['HOST'],
               port=os.environ['PORT'],
           )
           return _postgresql_connection
       except Exception as e:
           print(f"ERROR: Unexpected error: Could not connect to PostgreSQL instance: {e}")
           raise
   
   
   @app.route('/', defaults={'path': ''})
   @app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE'])
   def hello_world(path):
       conn = getConnection()
       try:
           with conn.cursor() as cursor:
               # 查询 users 表的所有记录,users 表需要根据实际表名进行修改
               sql = "SELECT * FROM users"
               cursor.execute(sql)
               result = cursor.fetchall()
               
               columns = [desc[0] for desc in cursor.description]  # 获取字段名列表
   
               # 将查询结果转换为字典列表
               users = []
               for row in result:
                   user = {}
                   for idx, column_name in enumerate(columns):
                       value = row[idx]
                       if isinstance(value, datetime):  # 处理日期类型字段
                           user[column_name] = value.strftime('%Y-%m-%d %H:%M:%S')
                       else:
                           user[column_name] = value
                   users.append(user)
                   
               if users:
                   # 返回所有用户的 JSON 响应
                   return jsonify(users), 200
               else:
                   # 如果没有找到用户，返回 404 错误
                   return jsonify({'error': 'No users found'}), 404
   
       except Exception as e:
           logging.error(f"Error occurred during database operation: {e}")
           return jsonify({'error': 'Database error'}), 500
   
   
   if __name__ == '__main__':
       app.run(host='0.0.0.0', port=9000)
   

2. 为函数安装所需依赖：pip3 install -t . psycopg2 ，点击部署代码使依赖生效。详细步骤请参见通过控制台Web IDE终端安装依赖。

3. 在函数详情页，依次选择配置 > 环境变量，单击编辑，在环境变量面板配置以下环境变量。

   环境变量名称	环境变量值	说明
   HOST	pgm-*****q.pg.rds.aliyuncs.com	数据库实例的访问地址，如果您选择同VPC内的PostgreSQL数据库的场景，请将此环境变量值设置为数据库的内网地址。如果您选择跨VPC或跨地域访问PostgreSQL数据库的场景，请将此环境变量值设置为数据库的外网地址。 访问实例列表，在上方选择地域，然后单击目标实例ID。在左侧导航栏单击数据库连接，在数据库连接信息区域获取连接数据库的连接地址信息。
   PORT	5432	数据库内网端口号
   USER	*****	PostgreSQL实例中创建的账号名称。
   PASSWORD	*****	数据库密码
   DATABASE	*****	实例中创建的数据库名称。

4. 在函数详情页，选择代码页签，单击测试函数，执行成功后查看返回结果。