Spring Cosmos部分请求响应缓慢，卡在Unsafe.park等待数秒

可能原因 #

1. 结果集阻塞等待
   collect(Collectors.toList())是终端操作，会阻塞线程直到所有查询结果返回。如果单次查询结果集过大，或Cosmos服务端因分区热点、跨区域同步延迟（开启了multipleWriteRegionsEnabled+preferredRegions）导致结果返回缓慢，线程就会进入Unsafe.park的等待状态。

2. 连接/线程池资源不足
   默认DirectConnectionConfig的连接池、Reactor线程池参数可能不匹配实际负载。即使是1TPS的低负载，若线程池被阻塞请求占满，新请求会等待可用线程，表现为超时。

3. 一致性级别与跨区域开销
   若设置了强一致性级别，结合多写区域开启，跨区域数据同步需要额外时间。当请求路由到非主区域时，需等待数据同步完成，导致延迟飙升。

4. 查询与索引缺陷
   部分慢查询可能未配置合适索引，引发Cosmos服务端全表扫描。平时数据量小时表现正常，一旦遇到大分区或热点数据，查询耗时剧增，客户端线程持续等待结果。

优化方案 #

1. 优化查询与索引 #

• 检查慢查询语句，确保过滤、排序字段配置对应复合索引或范围索引。可在Cosmos DB门户查看查询执行统计，确认是否存在索引缺失。
• 避免一次性拉取全量数据，改用分页查询（如Spring Data的Pageable），减少collect()的阻塞等待时间。

2. 调整Cosmos客户端配置 #

• 优化连接池参数：修改DirectConnectionConfig的连接池相关参数，确保资源充足：

  DirectConnectionConfig directConnectionConfig = DirectConnectionConfig.getDefaultConfig()
      .setMaxConnectionsPerEndpoint(60)
      .setConnectionPoolSize(200);
  

• 设置请求超时：缩短requestTimeout避免线程长时间阻塞，同时开启超时重试：

  cosmosClient = new CosmosClientBuilder()
      .endpoint(uri)
      .key(key)
      .contentResponseOnWriteEnabled(true)
      .directMode(directConnectionConfig)
      .multipleWriteRegionsEnabled(true)
      .consistencyLevel(ConsistencyLevel.valueOf(consistencyLevel.trim().toUpperCase()))
      .preferredRegions(Arrays.asList(preferredRegion.split("\\s*,\\s*")))
      .requestTimeout(Duration.ofSeconds(3)) // 设置超时时间
      .buildClient();
  

• 优化区域路由：如果不需要多写能力，关闭multipleWriteRegionsEnabled；调整preferredRegions优先选择主写区域，减少跨区域同步延迟。

3. 异步处理结果 #

避免主线程阻塞等待，改用异步订阅处理查询结果：

cosmosTemplate.find(query, Entity.class)
    .subscribe(
        results -> { /* 业务逻辑处理 */ },
        throwable -> { /* 异常处理 */ }
    );

4. 监控与排查 #

• 开启Cosmos客户端详细日志，跟踪慢请求的ID、路由区域、执行耗时，定位是客户端还是服务端问题。
• 使用JVM工具（如jstack、Arthas）分析Unsafe.park线程的栈信息，确认是等待响应结果还是等待资源。
• 在Cosmos DB门户查看核心指标：请求延迟、分区键分布、索引利用率，排查服务端瓶颈。

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