多线程WSGI环境下Flask应用连接Cassandra超时问题求助

WSGI与Cassandra Python Driver本身不存在兼容问题，你的故障主要源于Cluster初始化方式错误和多线程WSGI环境下Session初始化时机不当，以下是具体分析和修复方案：

1. 修正Cluster初始化逻辑 #

你的代码中，创建Cluster实例后再赋值load_balancing_policy的方式不符合Driver规范——负载均衡策略必须在Cluster构造时传入，否则不会生效，这会直接导致连接无法正确建立或请求路由失败。

修正后的连接代码：

def connect_to_cassandra():
    auth_provider = PlainTextAuthProvider(username=config['username'], password=config['password'])
    # 直接在Cluster构造时传入负载均衡策略
    load_balancing_policy = TokenAwarePolicy(DCAwareRoundRobinPolicy(DC))
    cluster = Cluster(
        contact_points=config['contact_points'],
        port=config['port'],
        auth_provider=auth_provider,
        load_balancing_policy=load_balancing_policy
    )
    session = cluster.connect(config['keyspace'])
    # 可选：主动验证连接有效性
    session.execute("SELECT release_version FROM system.local")
    return session

2. 适配多线程WSGI的Session初始化策略 #

在多线程WSGI服务器（如Gunicorn多线程模式、uWSGI多线程模式）中，禁止在模块加载阶段提前创建全局Session——此时WSGI的worker线程尚未完全启动，全局Session无法被子线程正确复用。

正确的初始化方式（以Flask 2.0+为例）：

from flask import Flask

app = Flask(__name__)
cassandra_session = None

def init_cassandra():
    global cassandra_session
    if cassandra_session is None:
        cassandra_session = connect_to_cassandra()

# 确保在第一个请求到来时初始化Session，后续请求复用已创建的实例
@app.before_request
def before_request():
    init_cassandra()

# 示例业务路由
@app.route('/query-data')
def query_cassandra():
    result = cassandra_session.execute("SELECT * FROM your_target_table LIMIT 10")
    return str([row for row in result])

如果使用多进程WSGI服务器（如Gunicorn多worker），需注意每个进程必须独立初始化Cluster和Session——Cassandra Driver的连接池是进程内资源，无法跨进程共享。

3. 额外排查方向 #

• 检查Driver版本：确保使用3.25+的稳定版，旧版本存在多线程环境下的连接池bug。
• 网络连通性：确认WSGI服务器能访问Cassandra集群的所有节点（TokenAwarePolicy会路由到非contact point节点），而非仅配置的contact points。
• 启用Driver日志：添加日志配置排查连接过程中的细节错误：

  import logging
  logging.basicConfig(level=logging.DEBUG)
  

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