Spring Data Cosmos是否支持Azure Cosmos DB事务批量操作？

支持状态说明 #

目前官方稳定版Spring Data Cosmos未原生封装Cosmos Java SDK 4.7.0+提供的事务批量操作能力，因此官方文档暂无相关说明，属于正常情况。
虽然Spring Data Cosmos底层依赖的核心SDK版本已经兼容事务批量特性，但高层Spring Data抽象层未做对应的API暴露，因此无法直接通过Repository或者CosmosTemplate的原生方法调用事务批量能力。

最优实现路径 #

推荐直接复用Spring容器中已初始化的CosmosClient实例调用原生SDK的事务批量API，该方案无需额外引入依赖、无需重复配置连接资源，完全复用Spring Data Cosmos已有的连接池、权限配置、序列化规则等能力，无额外性能损耗，是当前阶段的最优选择。
示例代码如下：

import com.azure.cosmos.CosmosClient;
import com.azure.cosmos.CosmosContainer;
import com.azure.cosmos.models.PartitionKey;
import com.azure.cosmos.models.TransactionalBatch;
import com.azure.cosmos.models.TransactionalBatchResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

@Component
public class CosmosTransactionalBatchService {

    // 直接注入Spring托管的CosmosClient实例，无需手动创建
    @Autowired
    private CosmosClient cosmosClient;

    public boolean executeBatchOp(String databaseName, String containerName, String partitionKeyVal) {
        CosmosContainer container = cosmosClient.getDatabase(databaseName)
                .getContainer(containerName);
        
        // 构造事务批量操作，所有操作必须属于同一个分区键值
        TransactionalBatch batch = TransactionalBatch.createTransactionalBatch(new PartitionKey(partitionKeyVal))
                // 按业务需要添加增删改操作，最多支持100个操作，总大小不超过2MB
                .createItem(new TestEntity("item001", partitionKeyVal, "新增内容"))
                .replaceItem("item002", new TestEntity("item002", partitionKeyVal, "更新内容"))
                .deleteItem("item003", new PartitionKey(partitionKeyVal));
        
        TransactionalBatchResponse batchResponse = batch.execute();
        return batchResponse.isSuccessStatusCode();
    }

    // 示例实体类
    public static class TestEntity {
        private String id;
        private String partitionKey;
        private String content;

        public TestEntity(String id, String partitionKey, String content) {
            this.id = id;
            this.partitionKey = partitionKey;
            this.content = content;
        }
        // 省略getter/setter
    }
}

扩展建议 #

如果团队需要统一使用Spring Data风格的操作接口，可以基于上述原生调用的逻辑自行封装通用工具类，上层封装符合Spring Data使用习惯的API即可，底层依然复用原生SDK的事务批量能力。
注意：Cosmos DB事务批量本身存在约束，所有操作必须归属同一个逻辑分区键值，单个批量请求最多支持100个操作，请求总大小不可超过2MB。

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