如何通过Firestore Query Protos自动生成多语言支持查询接口？

本文共计790个文字，预计阅读时间需要4分钟。

原文：

在 Java 中，可以使用 Firestore SDK 构建查询，例如：

Query query= db.collection("col2").whereGreaterThanOrEqualTo("name", "a").orderBy("name", Query.Direction.ASCENDING).limit(50);

为了将其转换为其他语言，首先需要将该查询转换为 Protobuf 格式的 StructuredQuery 对象：

import com.google.cloud.firestore.Query; import com.google.cloud.firestore.v1.RunQueryRequest; import com.google.firestore.v1.StructuredQuery; // Assuming you have a Firestore 'db' instance RunQueryRequest runQueryRequest = RunQueryRequest.newBuilder() .setStructuredQuery(query.toProto()) .build(); StructuredQuery structuredQuery = runQueryRequest.getStructuredQuery(); System.out.println("structuredQuery: " + structuredQuery);

这段代码会输出 StructuredQuery 对象的文本表示，包含查询的各种属性，如集合 ID、过滤条件、排序规则和限制数量。

现在，关键在于如何利用这个 StructuredQuery 对象生成其他语言的查询代码。这可以通过解析 StructuredQuery 对象的属性，并将其转换为目标语言的语法来实现。 例如，可以利用 Java 的 String.format 方法，根据 StructuredQuery 的属性动态生成目标语言的代码字符串。

下面是一个示例，展示如何使用 String.format 生成 Python 代码片段：

String collectionId = structuredQuery.getFrom(0).getCollectionId(); String fieldPath = structuredQuery.getWhere().getFieldFilter().getField().getFieldPath(); String operator = structuredQuery.getWhere().getFieldFilter().getOp().toString(); String stringValue = structuredQuery.getWhere().getFieldFilter().getValue().getStringValue(); String orderByField = structuredQuery.getOrderBy(0).getField().getFieldPath(); String direction = structuredQuery.getOrderBy(0).getDirection().toString(); long limit = structuredQuery.getLimit().getValue(); String pythonCode = String.format( "db.collection('%s').where('%s', '%s', '%s').order_by('%s', direction='%s').limit(%d)", collectionId, fieldPath, convertOperator(operator), stringValue, orderByField, convertDirection(direction), limit ); System.out.println("Python code: " + pythonCode);

在这个例子中，convertOperator 和 convertDirection 是自定义函数，用于将 Java 中的 Firestore 操作符和排序方向转换为 Python 中对应的表示。

public static String convertOperator(String javaOperator) { switch (javaOperator) { case "GREATER_THAN_OR_EQUAL": return ">="; // Add more cases for other operators default: return javaOperator; } } public static String convertDirection(String javaDirection) { switch (javaDirection) { case "ASCENDING": return "ASCENDING"; case "DESCENDING": return "DESCENDING"; default: return javaDirection; } }

注意事项：

• 错误处理： 在解析 StructuredQuery 对象时，需要进行错误处理，以防止由于缺少某些属性而导致的异常。
• 类型转换： StructuredQuery 中 value 字段可能是不同类型的，需要根据实际类型进行转换。
• 语言差异： 不同语言的 Firestore SDK 在语法和功能上可能存在差异，需要根据目标语言的特点进行调整。例如，Python的排序方向需要指定 direction 参数，而Java则直接使用 Query.Direction.ASCENDING。
• 安全性： 确保生成的代码是安全的，避免 SQL 注入等安全问题。

总结：

通过结合 Firestore 的 StructuredQuery 类和字符串格式化技术，可以实现从 Java Firestore 查询自动生成其他编程语言查询代码。这种方法能够显著提高多语言 Firestore 应用的开发效率，并确保不同平台查询逻辑的一致性。 虽然示例代码只展示了 Python 的转换，但该方法可以扩展到其他支持 Protobuf 的编程语言。