TxBatchSyncTestBed.java

public void testBed(Customer targetCustomer, String[] operation) {

    try (CosmosClient client = new CosmosClientBuilder()
            .endpoint(ENDPOINT)
            .key(KEY)
            .consistencyLevel(ConsistencyLevel.SESSION)
            .contentResponseOnWriteEnabled(true)
            .preferredRegions(List.of("Japan East"))
            .readRequestsFallbackEnabled(true)
            .endpointDiscoveryEnabled(true)
            .gatewayMode()
            .connectionSharingAcrossClientsEnabled(true)
            .buildClient()) {

        CosmosContainer container = client
                .getDatabase(DATABASE)
                .getContainer(CONTAINER);

        TransactionalBatch txBatch = TransactionalBatch
                .createTransactionalBatch(new PartitionKey(targetCustomer.getMyPartitionKey()));
        boolean illegalOperation = false;
        for (String s : operation) {
            switch (s.toUpperCase()) {
                case "CREATE":
                    // add Create Operation
                    txBatch.createItemOperation(targetCustomer);
                    break;
                case "REPLACE":
                    // add Replace Operation
                    targetCustomer.setCity("きょうと");
                    show(s.toUpperCase(), targetCustomer);
                    txBatch.replaceItemOperation(targetCustomer.getId(), targetCustomer);
                    break;
                case "UPSERT":
                    // add Upsert Operation
                    targetCustomer.setRegion("日本のどこか");
                    show(s.toUpperCase(), targetCustomer);
                    txBatch.upsertItemOperation(targetCustomer);
                    break;
                case "DELETE":
                    // add Delete Operation
                    show(s.toUpperCase(), targetCustomer);
                    txBatch.deleteItemOperation(targetCustomer.getId());
                    break;
                case "READ":
                    // add read operation
                    txBatch.readItemOperation(targetCustomer.getId()).getItem();
                    break;
                default:
                    illegalOperation = true;
                    break;
            }
        }
        if (!illegalOperation) {
            TransactionalBatchResponse txResponse = container.executeTransactionalBatch(txBatch);
            txResponse.getResults().forEach(txResult -> logger.info(
                    "Result [" + txResult.getStatusCode() + "] - [" + txResult.getSubStatusCode() + "] Operation: " + txResult.getOperation().getOperationType().name()));
        }
    } catch (CosmosException e) {
        e.printStackTrace();
    }
}