Locust中@events.test_stop.add_listener不执行数据库请求问题咨询

针对你遇到的Locust test_stop 事件无法执行数据库请求的问题，常见原因和解决办法如下：

1. 数据库连接已失效或提前关闭 #

test_start 中创建的数据库连接，可能在测试过程中被意外关闭（比如测试用例里的代码释放了连接），或者因超时被数据库主动断开，导致test_stop时调用的是无效连接，自然得不到响应结果。

• 解决办法：在test_stop事件处理函数内重新建立数据库连接，不要复用test_start创建的连接；或者给连接设置合理的超时参数，确保其在整个测试周期内保持活跃。

2. Locust资源清理时机冲突 #

Locust在测试停止阶段会逐步清理运行时资源（比如线程池、进程池），如果你的数据库请求依赖这些已被清理的资源（比如用了Locust的异步客户端或共享线程），就会导致请求无法执行。另外如果是分布式测试，test_stop默认在worker节点执行，若worker节点未正确配置数据库连接信息，也会失败。

• 解决办法：
  • 分布式测试场景下，通过判断environment.master标识，让数据库逻辑在master节点执行；
  • 在test_stop中使用独立资源执行数据库操作，不依赖Locust的测试资源池。

3. 异常被静默吞掉 #

test_stop事件中的代码如果没有异常捕获逻辑，数据库请求的报错（比如连接超时、权限不足、SQL语法错误）不会被输出，看起来就像没有执行。

• 解决办法：给数据库操作代码加上完整的异常捕获并打印错误信息，示例：

  @events.test_stop.add_listener
  def on_test_stop(environment, **kwargs):
      try:
          # 重新建立数据库连接
          conn = psycopg2.connect(
              dbname="your_db", user="user", password="pass", host="localhost"
          )
          cursor = conn.cursor()
          cursor.execute("SELECT COUNT(*) FROM test_results")
          result = cursor.fetchone()
          print(f"测试结束统计结果: {result}")
          cursor.close()
          conn.close()
      except Exception as e:
          print(f"测试停止时数据库操作失败: {str(e)}")
  

4. 进程退出过快导致异步操作未完成 #

如果你的数据库请求是异步执行的，Locust在测试结束时可能直接终止进程，导致异步请求还没得到响应就被中断，看起来没有结果。

• 解决办法：改用同步的数据库客户端执行请求；如果必须用异步，需要在test_stop中等待异步操作完成后再退出（比如使用asyncio.run()或await关键字确保执行完成）。

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