如何高效使用SqlQuerySpec执行Cosmos DB多查询并获取多结果集？

嘿，针对你提出的两个Cosmos DB查询问题，我来给你详细梳理下高效的解决方案！

要高效执行两个独立的SqlQuerySpec查询，核心思路是减少等待时间+优化查询性能，具体可以这么做：

• 异步并行执行查询：Cosmos DB支持并发请求，把两个查询放在异步任务里并行跑，比串行执行能节省不少总耗时。千万别用.Result阻塞线程，改用await配合Task.WhenAll来实现并行，避免线程浪费。
• 确保查询用上合适的索引：你的两个查询都用到了MeterCode和Timestamp，一定要给这两个字段创建组合索引（优先按MeterCode分区，再按Timestamp排序），这样查询能直接走索引，避免全集合扫描，速度会快很多。
• 优化客户端连接配置：调整ConnectionPolicy里的MaxConnectionLimit（比如设到100+），提升并发请求的处理能力。

给你个代码示例，演示异步并行执行的写法：

// 定义两个查询的SqlQuerySpec
var querySpec1 = new SqlQuerySpec(query1, parameters1);
var docQuery1 = docClient.CreateDocumentQuery<Reading>(collectionUri, querySpec1).AsDocumentQuery();

var querySpec2 = new SqlQuerySpec(query2, parameters2);
var docQuery2 = docClient.CreateDocumentQuery<Reading>(collectionUri, querySpec2).AsDocumentQuery();

// 并行执行两个查询任务
var priorTask = FetchAllResults(docQuery1);
var rangeTask = FetchAllResults(docQuery2);
await Task.WhenAll(priorTask, rangeTask);

// 获取最终结果
var priorReading = priorTask.Result.FirstOrDefault();
var rangeReadings = rangeTask.Result;

// 封装一个获取所有分页结果的辅助方法
async Task<List<Reading>> FetchAllResults(IDocumentQuery<Reading> query)
{
    var results = new List<Reading>();
    while (query.HasMoreResults)
    {
        var batch = await query.ExecuteNextAsync<Reading>();
        results.AddRange(batch);
    }
    return results;
}

你想把“取起始点最近的历史读数”和“取时间段内的所有读数”合并成一个请求，有两个靠谱的方案，看你需求选：

方案一：用UNION ALL合并查询（无需存储过程） #

Cosmos DB支持UNION ALL，只要两个查询的列数、类型一致就能合并。我们可以给每个结果加个标识列（比如ResultType），这样客户端拿到结果后就能区分哪部分是历史读数，哪部分是时间段数据。

修改后的查询语句如下：

SELECT 'prior' AS ResultType, n.Timestamp, n.Value 
FROM Readings AS n 
WHERE n.MeterCode = @meterCode AND n.Timestamp < @fromUtc 
ORDER BY n.Timestamp DESC OFFSET 0 LIMIT 1

UNION ALL

SELECT 'range' AS ResultType, n.Timestamp, n.Value 
FROM Readings AS n 
WHERE n.MeterCode = @meterCode AND n.Timestamp >= @fromUtc AND n.Timestamp <= @toUtc 
ORDER BY n.Timestamp ASC

然后客户端拿到结果后，按ResultType分组处理就行：

var combinedQuerySpec = new SqlQuerySpec(combinedQuery, combinedParameters);
var combinedResults = await FetchAllResults(docClient.CreateDocumentQuery<dynamic>(collectionUri, combinedQuerySpec).AsDocumentQuery());

var priorReading = combinedResults.FirstOrDefault(r => r.ResultType == "prior");
var rangeReadings = combinedResults.Where(r => r.ResultType == "range").ToList();

注意：UNION ALL的整体结果排序是子查询各自的排序，如果你需要严格的顺序，可能要在客户端再做一次排序。

方案二：用存储过程一次性返回两个结果集（更高效） #

如果想减少网络往返次数，最推荐的是用Cosmos DB的存储过程：在存储过程里依次执行两个查询，把结果打包成一个对象返回，客户端只需要调用一次存储过程就能拿到所有数据。

存储过程的JavaScript代码示例：

function getCombinedReadings(meterCode, fromUtc, toUtc) {
    const collection = getContext().getCollection();
    const response = getContext().getResponse();
    const result = { prior: [], range: [] };

    // 执行第一个查询：获取起始点前的最近读数
    const query1 = `SELECT TOP 1 n.Timestamp, n.Value FROM Readings AS n WHERE n.MeterCode = @meterCode AND n.Timestamp < @fromUtc ORDER BY n.Timestamp DESC`;
    const params1 = { meterCode, fromUtc };
    
    collection.queryDocuments(collection.getSelfLink(), query1, { parameters: params1 }, (err, docs) => {
        if (err) throw err;
        result.prior = docs;

        // 执行第二个查询：获取时间段内的读数
        const query2 = `SELECT n.Timestamp, n.Value FROM Readings AS n WHERE n.MeterCode = @meterCode AND n.Timestamp >= @fromUtc AND n.Timestamp <= @toUtc ORDER BY n.Timestamp ASC`;
        const params2 = { meterCode, fromUtc, toUtc };
        
        collection.queryDocuments(collection.getSelfLink(), query2, { parameters: params2 }, (err, docs) => {
            if (err) throw err;
            result.range = docs;
            response.setBody(result);
        });
    });
}

客户端调用存储过程的代码：

var sprocUri = UriFactory.CreateStoredProcedureUri(databaseId, collectionId, "getCombinedReadings");
var result = await docClient.ExecuteStoredProcedureAsync<dynamic>(sprocUri, new[] { meterCode, fromUtc, toUtc });

var priorReading = result.Response.prior.FirstOrDefault();
var rangeReadings = result.Response.range.ToList();

这个方案的优势是只有一次网络请求，尤其适合跨区域调用的场景，能显著降低延迟。

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