如何使用SQLAlchemy为Trino设置客户端标签

方法一：通过连接参数直接配置（针对支持会话属性的驱动） #

如果你的数据库驱动支持通过会话属性配置客户端标签，可直接在connect_args的session_props中添加对应参数。修改后的引擎代码示例如下：

engine = create_engine(url,
                       connect_args={'protocol': 'https',
                                     'requests_kwargs': {
                                         'cert': (cert_file_path, key_file_path),
                                         'verify': False
                                     },
                                     'session_props': {
                                         'query_max_execution_time': '6h',
                                         'query_max_run_time': '8h',
                                         'client_tag': '你的客户端标签名称'  # 新增客户端标签配置
                                     }
                                     })

方法二：通过SQLAlchemy事件监听全局添加标签 #

若驱动不支持会话属性配置，或需要更灵活的控制逻辑，可使用SQLAlchemy的事件监听机制，在每次查询执行前自动执行设置标签的语句：

from sqlalchemy import event

engine = create_engine(url,
                       connect_args={'protocol': 'https',
                                     'requests_kwargs': {
                                         'cert': (cert_file_path, key_file_path),
                                         'verify': False
                                     },
                                     'session_props': {
                                         'query_max_execution_time': '6h',
                                         'query_max_run_time': '8h'
                                     }
                                     })

# 定义事件监听函数，执行前设置客户端标签
@event.listens_for(engine, "before_execute")
def set_client_label(conn, clauseelement, multiparams, params):
    # 根据数据库类型替换对应SQL语句
    # 示例：PostgreSQL用 SET application_name = '你的客户端标签名称'
    # 示例：MySQL用 SET @client_tag = '你的客户端标签名称'
    conn.execute("SET application_name = '你的客户端标签名称'")

注意：不同数据库设置客户端标签的SQL语法存在差异，需根据实际使用的数据库类型调整监听函数中的执行语句。

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