Spring Data CosmosDB分页实现咨询：如何按页获取结果？

我明白你现在的困扰——Spring Data JPA里用Pageable分页顺手得很，但Spring Data CosmosDB的相关支持没找到，还不能用@Query，确实有点头疼。别担心，咱们可以直接用Azure Cosmos DB的原生Java SDK来实现分页，下面给你详细步骤和代码示例：

核心思路 #

Azure Cosmos DB原生SDK靠FeedResponse和**续令牌（Continuation Token）**处理分页：每次查询返回一批结果的同时，会生成一个续令牌，下一次查询带上这个令牌，就能精准获取下一页数据，逻辑和JPA分页类似，只是需要手动处理续令牌的传递。

具体实现步骤 #

1. 初始化Cosmos客户端与容器 #

先确保你已经配置好CosmosClient和CosmosContainer实例，基础示例如下：

// 替换成你的Cosmos DB配置
CosmosClient cosmosClient = new CosmosClientBuilder()
        .endpoint("你的Cosmos DB端点地址")
        .key("你的访问密钥")
        .buildClient();
CosmosContainer container = cosmosClient.getDatabase("你的数据库名").getContainer("你的容器名");

2. 编写分页查询逻辑 #

用QueryDefinition定义查询语句，通过QueryRequestOptions设置每页大小，再循环处理续令牌来逐页获取数据：

// 替换成你的业务查询语句
String query = "SELECT * FROM c WHERE c.status = @status";
QueryDefinition queryDefinition = new QueryDefinition(query)
        .withParameter("@status", "active");

int pageSize = 10; // 每页10条数据，对应你需求的分页大小
String continuationToken = null; // 第一页查询时传null，后续页用前一次返回的续令牌

do {
    QueryRequestOptions requestOptions = new QueryRequestOptions();
    requestOptions.setMaxItemCount(pageSize); // 设置每页最大返回条目数

    // 执行查询，获取分页结果
    FeedResponse<YourEntity> feedResponse = container.queryItems(
            queryDefinition,
            requestOptions,
            YourEntity.class // 替换成你的实体类
    );

    // 处理当前页的数据，比如返回给前端或执行业务逻辑
    List<YourEntity> currentPageData = feedResponse.getResults();
    System.out.println("当前页数据量：" + currentPageData.size());

    // 更新续令牌，用于下一页查询
    continuationToken = feedResponse.getContinuationToken();

    // 续令牌为null时，说明没有更多数据了，退出循环
} while (continuationToken != null);

3. 封装成贴近Spring风格的分页工具（可选） #

如果你想更贴近Spring Data的分页体验，可以封装一个分页结果类，让调用更简洁：

// 自定义分页结果类，类似Spring的Page
public class CosmosPage<T> {
    private List<T> content;
    private int pageSize;
    private String continuationToken;
    private boolean hasNext;

    // 构造函数、getter、setter省略
}

// 封装分页查询方法
public CosmosPage<YourEntity> getPage(String query, List<SqlParameter> params, int pageSize, String continuationToken) {
    QueryDefinition queryDefinition = new QueryDefinition(query);
    params.forEach(queryDefinition::withParameter);

    QueryRequestOptions options = new QueryRequestOptions();
    options.setMaxItemCount(pageSize);

    FeedResponse<YourEntity> response = container.queryItems(queryDefinition, options, YourEntity.class);

    CosmosPage<YourEntity> page = new CosmosPage<>();
    page.setContent(response.getResults());
    page.setPageSize(pageSize);
    page.setContinuationToken(response.getContinuationToken());
    page.setHasNext(response.getContinuationToken() != null);

    return page;
}

这样你就能像调用JPA分页方法一样，传入页大小和续令牌，拿到封装好的分页对象。

注意事项 #

• 续令牌是Cosmos DB标记查询位置的专属标识，不要自行解析或修改，直接在前后端之间传递即可。
• 如果查询涉及排序，务必在查询语句中明确排序字段，否则续令牌可能无法正常工作（分页依赖查询结果的固定顺序）。
• Cosmos DB不支持实时返回总页数/总条数（分布式场景下计算成本极高），如果需要总条数，可单独执行SELECT VALUE COUNT(1) FROM c WHERE ...的统计查询，但要注意性能开销。

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