Axum中服务错误转客户端错误时扩展字段丢失问题排查

核心原因推测 #

自定义Error类型无法在中间件中被获取，大概率是以下两类问题导致：

• 类型路径不一致：IntoResponse实现中插入的Error类型，和中间件main_response_mapper中尝试获取的类型并非同一个（比如存在模块路径差异、重复定义的同名类型）
• 缺少Clone实现：Axum处理响应时可能会克隆响应对象，而String默认实现了Clone，自定义Error若未实现该trait，克隆后extensions中的值会丢失

具体排查与修复步骤 #

• 校验类型一致性：确保IntoResponse实现和中间件中引用的Error是完全相同的类型。比如两处都使用crate::error::Error这种绝对路径，避免相对路径或不同模块的同名类型混淆。
• 为Error派生Clone：如果Error枚举的所有变体都支持克隆，直接派生该trait：

  #[derive(Debug, Clone)]
  enum Error {
      LoginFail,
      // 其他变体
  }
  

• 改用类型擦除存储：若不想实现Clone，可将Error包装为 trait 对象存入extensions，中间件按对应类型获取：

  // IntoResponse实现中插入
  response.extensions_mut().insert(Box::new(self) as Box<dyn std::error::Error + Send + Sync>);
  
  // 中间件中获取
  let err = response.extensions().get::<Box<dyn std::error::Error + Send + Sync>>();
  

• 验证响应克隆行为：在IntoResponse插入后打印response.extensions().get::<Error>()，再在中间件中打印响应的内存地址，确认是否为同一对象。若为克隆后的实例，必须保证Error可被克隆。

额外检查点 #

检查中间件执行顺序，若有其他中间件替换了整个Response对象，也会导致extensions中的值丢失，需确保map_response(main_response_mapper)在响应修改类中间件之后执行。

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