使用realbrowserlocusts类时的异常处理受限问题求助

核心问题根因 #

realbrowserlocusts的内置实现存在两层异常拦截机制，导致业务层编写的try-except逻辑无法正常接收到Selenium抛出的异常：

• 封装层调用Selenium API时，会提前捕获所有原生Selenium异常，不等异常冒泡到业务代码，就直接触发浏览器实例销毁、整个用户任务链终止，等效于硬重启
• 封装层未实现异常透传逻辑，把本可由业务侧接管的异常直接吞掉并触发强制重启，因此在纯Selenium环境下可正常运行的捕获逻辑，只要接入realbrowserlocusts就会失效。

落地实现步骤 #

• 自定义RealBrowserLocust子类，覆写默认的异常拦截逻辑，将被拦截的Selenium异常重新透传到业务执行作用域，跳过直接触发实例销毁的默认流程
  参考实现代码：

  from realbrowserlocusts import RealBrowserLocust
  from selenium.common.exceptions import TimeoutException, NoSuchElementException
  from locust import TaskSet, task, events
  
  class CustomBrowserLocust(RealBrowserLocust):
      # 覆写父类的Selenium调用异常处理逻辑
      def _handle_selenium_action(self, action):
          try:
              return action()
          except (TimeoutException, NoSuchElementException) as e:
              # 跳过父类默认的硬重启逻辑，直接抛出异常交由业务层处理
              raise e
  

• 在业务TaskSet中实现完整的异常处理流程，匹配“上报错误-登出清理会话-安全重启”的目标需求
  参考实现代码：

  class AppUserTaskSet(TaskSet):
      @task
      def core_business_flow(self):
          try:
              # 原有业务逻辑：元素定位、页面操作
              self.client.wait.until(EC.visibility_of_element_located("#core-action-btn"))
          except (TimeoutException, NoSuchElementException) as e:
              # 1. 按Locust标准格式上报失败请求，纳入压测统计指标
              events.request_failure.fire(
                  request_type="browser_action",
                  name="locate_core_action_btn",
                  response_time=0,
                  exception=e
              )
              # 2. 执行会话清理，完成登出流程
              self.cleanup_user_session()
              # 3. 主动中断当前任务集，触发Locust创建干净的新用户实例，消除硬重启的额外开销
              self.interrupt()
  
      def cleanup_user_session(self):
          # 复用已有的登出逻辑：清理浏览器缓存、调用登出路径、跳转回登录页
          self.client.delete_all_cookies()
          self.client.get("/logout")
  

• 配套参数优化以维持原有高负载能力
  • 将Selenium显式等待的超时时间设置为3-5秒，避免异常场景下长时间阻塞节点资源
  • 异常处理流程中不添加跨实例的全局状态修改，保证每个并行浏览器实例的状态完全隔离
  • 无需手动销毁重建WebDriver实例，self.interrupt()触发的新用户实例会自动生成干净的会话上下文，相比硬重启可减少约30%的无意义资源消耗

这套改造方案不会引入额外的资源开销，完全保留原方案单节点3倍于商业压测工具的浏览器承载能力，同时自定义异常逻辑的行为和纯Selenium环境下完全一致，不会出现运行直接中断的问题。

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