访问RDS MySQL最佳实践

步骤二：在函数中访问RDS

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

   from flask import Flask, jsonify
   import pymysql
   import os
   from datetime import datetime
   import logging
   
   app = Flask(__name__)
   
   # 全局变量用于存储 MySQL 单例连接
   _mysql_connection = None
   
   # 创建数据库连接（单例模式）
   def getConnection():
       global _mysql_connection
       try:
           # 如果连接已经存在且未断开，直接返回
           if _mysql_connection is not None:
               try:
                   # 测试连接是否有效（简单命令测试）
                   with _mysql_connection.cursor() as cursor:
                       cursor.execute("SELECT 1")  # 简单查询测试连接状态
                       result = cursor.fetchone()
                       if result and result[0] == 1:
                           return _mysql_connection
               except pymysql.OperationalError:
                   # 如果连接断开，重置连接
                   _mysql_connection = None
   
           # 如果连接不存在或已断开，重新创建连接
           _mysql_connection = pymysql.connect(
               host=os.environ['MYSQL_HOST'],
               port=int(os.environ['MYSQL_PORT']),
               user=os.environ['MYSQL_USER'],
               password=os.environ['MYSQL_PASSWORD'],
               db=os.environ['MYSQL_DBNAME']
           )
           return _mysql_connection
       except Exception as e:
           logging.error(f"Error occurred during database connection: {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. 在函数详情页，依次选择配置 > 环境变量，单击编辑，在环境变量面板配置以下环境变量。

   环境变量名称	环境变量值	说明
   MYSQL_HOST	rm-bp19u8e76ae****.mysql.rds.aliyuncs.com	数据库实例的访问地址。 如果您选择同VPC内的RDS数据库的场景，请将此环境变量值设置为数据库的内网地址。 如果您选择跨VPC或跨地域访问RDS数据库的场景，请将此环境变量值设置为数据库的外网地址。 在RDS实例列表，单击目标RDS实例ID，然后在左侧导航栏，选择数据库连接，在数据库连接页面获取数据库的内网地址或外网地址。
   MYSQL_DBNAME	db_test	RDS实例中创建的数据库名称。
   MYSQL_PASSWORD	*****	数据库密码。
   MYSQL_PORT	3306	数据库实例的私网端口。
   MYSQL_USER	dms_user_****	RDS实例中创建的账号名称。

3. 在函数详情页，选择代码页签，单击测试函数，执行成功后查看返回结果，已成功完成查询表格操作。