Cosmos DB存储过程动态输入与输出实现方案咨询

很高兴能帮到你！针对你提到的动态查询场景，我们可以利用Cosmos DB存储过程的JavaScript特性，通过传入参数动态构建SQL语句来实现需求。下面是一个完整的示例方案，完全适配你给出的三个查询场景，同时具备扩展能力。

核心思路 #

我们将创建一个存储过程，接受两个关键参数：

• selectedFields：需要查询的字段数组（比如["FirstName", "LastName"]）
• filterConditions：过滤条件对象（比如{"FirstName": "Jack"}）

通过这两个参数，我们可以动态拼接SELECT子句和WHERE子句，同时使用参数化查询避免SQL注入风险。

完整存储过程代码 #

function dynamicStudentQuery(selectedFields, filterConditions) {
    const collection = getContext().getCollection();
    const response = getContext().getResponse();

    // 校验输入参数
    if (!selectedFields || selectedFields.length === 0) {
        throw new Error("请至少指定一个要查询的字段");
    }
    if (!filterConditions || Object.keys(filterConditions).length === 0) {
        throw new Error("请至少指定一个过滤条件");
    }

    // 构建SELECT子句
    const selectClause = `SELECT ${selectedFields.map(field => `c.${field}`).join(', ')}`;

    // 构建WHERE子句（支持多条件AND拼接）
    const whereConditions = Object.entries(filterConditions).map(([field, value]) => 
        `c.${field} = @${field}`
    ).join(' AND ');
    const whereClause = `WHERE ${whereConditions}`;

    // 拼接完整SQL语句
    const sqlQuery = `${selectClause} FROM c ${whereClause}`;

    // 准备参数化查询的参数，避免SQL注入
    const parameters = Object.entries(filterConditions).map(([field, value]) => ({
        name: `@${field}`,
        value: value
    }));

    // 执行查询（enableCrossPartitionQuery根据你的分区键配置调整）
    const queryOptions = { enableCrossPartitionQuery: true };
    collection.queryDocuments(collection.getSelfLink(), sqlQuery, queryOptions, (err, documents) => {
        if (err) throw err;
        if (documents.length === 0) {
            response.setBody("未找到匹配的文档");
        } else {
            // 转换结果，移除c.前缀，返回干净的字段结构
            const transformedResults = documents.map(doc => {
                const result = {};
                selectedFields.forEach(field => {
                    result[field] = doc[field];
                });
                return result;
            });
            response.setBody(transformedResults);
        }
    });
}

针对你的场景调用示例 #

场景1：查询FirstName和LastName，过滤条件为FirstName="Jack" #

调用存储过程时传入参数：

["FirstName", "LastName"], {"FirstName": "Jack"}

场景2：查询FirstName和ZipCode，过滤条件为LastName="Louis" #

调用参数：

["FirstName", "ZipCode"], {"LastName": "Louis"}

场景3：查询ZipCode，过滤条件为ZipCode="12345" #

调用参数：

["ZipCode"], {"ZipCode": "12345"}

注意事项与扩展建议 #

• 分区键优化：如果你的集合有分区键，且过滤条件中包含分区键字段，建议将enableCrossPartitionQuery设为false，以提升查询性能。
• SQL注入防护：代码中使用了参数化查询，绝对不要直接将用户输入的值拼接进SQL字符串，避免安全风险。
• 扩展过滤运算符：如果需要支持>、<、LIKE等运算符，可以修改filterConditions的结构，比如传入{"FirstName": {"operator": "=", "value": "Jack"}}，然后在构建WHERE子句时动态替换运算符。
• 空结果处理：代码中已经处理了无匹配文档的情况，你可以根据需求调整返回内容。

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