如何解除Datastore模式下Firestore事务25个实体组的限制？

问题描述 #

我的应用在部分边缘场景中需要对大量实体（当前为163个）执行事务，此时会触发如下异常：

File ".venv/lib/python3.10/site-packages/google/api_core/grpc_helpers.py", line 67, in error_remapped_callable
raise exceptions.from_grpc_error(exc) from exc
google.api_core.exceptions.InvalidArgument: 400 operating on too many entity groups in a single transaction.

此前使用旧版Cloud Datastore时，25个实体组的限制符合预期，但应用已自动迁移至Firestore in Datastore模式（通过gcloud alpha firestore databases describe命令确认）。根据官方文档，该模式下事务不再受25个实体组限制，我已将Python客户端库升级至最新版google-cloud-datastore==2.20.1，但栈追踪显示错误来自后端。测试发现25个实体的事务可正常执行，26个则失败，说明限制仍在生效。请问迁移后的旧数据库是否存在问题？是否需要更换客户端库？

解决方案 #

• 核心问题在于事务类型的选择：Firestore in Datastore模式确实支持跨任意数量实体组的事务，但必须使用分区事务（Partitioned Transactions），而非传统的Datastore事务。传统事务的25实体组限制在该模式下依然保留，只有分区事务才突破了这个限制。
• 检查代码调用方式：在Python客户端库中，普通的client.transaction()仍受25实体组限制，需要改为调用client.transaction(partitioned=True)创建分区事务，才能处理超过25个实体组的操作。
• 迁移后的数据库本身无问题：自动迁移到Firestore in Datastore模式的数据库完全支持分区事务，不存在遗留限制导致的异常。
• 客户端库无需更换：你使用的google-cloud-datastore==2.20.1已经支持分区事务API，版本符合要求，不需要更换其他库。

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