Amazon DynamoDB Transactions を使用して複数のアイテムに調整された変更を加える

private static void createTable(String tableName) {
        try {
        	// 文字列を保持する 'Id' という名前のプライマリハッシュキーを持つテーブルを作成します
            CreateTableRequest createTableRequest = new CreateTableRequest().withTableName(tableName)
                .withKeySchema(new KeySchemaElement().withAttributeName("Id").withKeyType(KeyType.HASH))
                .withAttributeDefinitions(new AttributeDefinition().withAttributeName("Id").withAttributeType(ScalarAttributeType.S))
                .withProvisionedThroughput(new ProvisionedThroughput().withReadCapacityUnits(1L).withWriteCapacityUnits(1L));

            // テーブルがまだ存在しない場合は作成します
            TableUtils.createTableIfNotExists(dynamoDB, createTableRequest);

            // テーブルが ACTIVE 状態になるのを待ちます
        	TableUtils.waitUntilActive(dynamoDB, tableName);
        	
            // 新しいテーブルについて記述します
            DescribeTableRequest describeTableRequest = new DescribeTableRequest().withTableName(tableName);
            TableDescription tableDescription = dynamoDB.describeTable(describeTableRequest).getTable();
            System.out.println("Table Created Successfully.Table Description: " + tableDescription);
        	
		} catch (InterruptedException e) {
        	System.out.println("Occupied thread is interrupted");
		}
    }

    private static void createReservation() {
    	
        // 宿泊客の項目を作成します
    	HashMap<String, AttributeValue> guestItem = new HashMap<String, AttributeValue>();
    	guestItem.put("Id", new AttributeValue("John"));
    	guestItem.put("ActiveReservations", new AttributeValue("500"));
    	
        // 部屋の項目を作成します
    	HashMap<String, AttributeValue> roomItem = new HashMap<String, AttributeValue>();
    	roomItem.put("Id", new AttributeValue("R20014"));
    	roomItem.put("RoomStatus", new AttributeValue("FREE"));
        
        // 予約の項目を作成します
    	HashMap<String, AttributeValue> reservationItem = new HashMap<String, AttributeValue>();
    	reservationItem.put("Id", new AttributeValue("500"));
    	reservationItem.put("GuestId", new AttributeValue("John"));
    	reservationItem.put("ReservationStatus", new AttributeValue("PENDING"));

        Put createGuest = new Put().withTableName(TABLE_NAME).withItem(guestItem);
        Put createRoom = new Put().withTableName(TABLE_NAME).withItem(roomItem);
        Put createReservation = new Put().withTableName(TABLE_NAME).withItem(reservationItem);

        Collection<TransactWriteItem> actions = Arrays.asList(
        		new TransactWriteItem().withPut(createGuest),
                new TransactWriteItem().withPut(createRoom),
                new TransactWriteItem().withPut(createReservation));
        
        TransactWriteItemsRequest createReservationTransaction = new TransactWriteItemsRequest()
                .withTransactItems(actions)
                .withReturnConsumedCapacity(ReturnConsumedCapacity.TOTAL);

	    // トランザクションを実行し、結果を処理します。
		// 次のコードスニペットは、前に定義したアクションを単一のオールオアナッシング操作として実行する方法を示しています
        dynamoDB.transactWriteItems(createReservationTransaction);
        System.out.println("Create Reservation transaction is successful");
    }

    private static void checkIn() {
    	
    	//updateGuest オブジェクトを作成します
    	HashMap<String, AttributeValue> guestItemKey = new HashMap<String, AttributeValue>();
    	guestItemKey.put("Id", new AttributeValue("John"));
    	
    	HashMap<String, AttributeValue> guestExpressionAttributeValues = new HashMap<String, AttributeValue>();
    	guestExpressionAttributeValues.put(":occupies_rooms", new AttributeValue("R20014"));
    	
    	Update updateGuest = new Update()
    			.withTableName(TABLE_NAME)
    			.withKey(guestItemKey)
    			.withUpdateExpression("SET OccupiesRooms = :occupies_rooms")
    			.withExpressionAttributeValues(guestExpressionAttributeValues);

    	
    	//updateRoom オブジェクトを作成します
    	HashMap<String, AttributeValue> roomItemKey = new HashMap<String, AttributeValue>();
    	roomItemKey.put("Id", new AttributeValue("R20014"));
    	
    	HashMap<String, AttributeValue> roomExpressionAttributeValues = new HashMap<String, AttributeValue>();
    	roomExpressionAttributeValues.put(":room_status", new AttributeValue("OCCUPIED"));
    	roomExpressionAttributeValues.put(":rented_to_reservation", new AttributeValue("500"));
    	
    	Update updateRoom = new Update()
    			.withTableName(TABLE_NAME)
    			.withKey(roomItemKey)
    			.withUpdateExpression("SET RoomStatus = :room_status, RentedToReservation = :rented_to_reservation")
    			.withExpressionAttributeValues(roomExpressionAttributeValues);

    	//updateReservation オブジェクトを作成します
    	HashMap<String, AttributeValue> reservationItemKey = new HashMap<String, AttributeValue>();
    	reservationItemKey.put("Id", new AttributeValue("500"));
    	
    	HashMap<String, AttributeValue> reservationExpressionAttributeValues = new HashMap<String, AttributeValue>();
    	reservationExpressionAttributeValues.put(":reservation_status", new AttributeValue("FULLFILLED"));
    	reservationExpressionAttributeValues.put(":fullfilled_by_room", new AttributeValue("R20014"));
    	
    	Update updateReservation = new Update()
    			.withTableName(TABLE_NAME)
    			.withKey(reservationItemKey)
    			.withUpdateExpression("SET ReservationStatus = :reservation_status, FullfilledByRoom = :fullfilled_by_room")
    			.withExpressionAttributeValues(reservationExpressionAttributeValues);
    	
    	//Transactions の実行
    	Collection<TransactWriteItem> actions = Arrays.asList(
        		new TransactWriteItem().withUpdate(updateGuest),
                new TransactWriteItem().withUpdate(updateRoom),
                new TransactWriteItem().withUpdate(updateReservation));
        
        TransactWriteItemsRequest createReservationTransaction = new TransactWriteItemsRequest()
                .withTransactItems(actions)
                .withReturnConsumedCapacity(ReturnConsumedCapacity.TOTAL);

        dynamoDB.transactWriteItems(createReservationTransaction);
        System.out.println("Check-in transaction is successful");
    	
    }

    private static void checkOut() {
    	//updateGuest オブジェクトを作成します
    	HashMap<String, AttributeValue> guestItemKey = new HashMap<String, AttributeValue>();
    	guestItemKey.put("Id", new AttributeValue("John"));
    	
    	Update updateGuest = new Update()
    			.withTableName(TABLE_NAME)
    			.withKey(guestItemKey)
    			.withUpdateExpression("REMOVE OccupiesRooms, ActiveReservations");
    	//属性に値がないため、削除されます
    	
    	//updateRoom オブジェクトを作成します
    	HashMap<String, AttributeValue> roomItemKey = new HashMap<String, AttributeValue>();
    	roomItemKey.put("Id", new AttributeValue("R20014"));
    	
    	HashMap<String, AttributeValue> roomExpressionAttributeValues = new HashMap<String, AttributeValue>();
    	roomExpressionAttributeValues.put(":room_status", new AttributeValue("FREE"));
    	
    	Update updateRoom = new Update()
    			.withTableName(TABLE_NAME)
    			.withKey(roomItemKey)
    			.withUpdateExpression("SET RoomStatus = :room_status REMOVE RentedToReservation") //Since there is no value in the attribute, it is removed
    			.withExpressionAttributeValues(roomExpressionAttributeValues);
    	
    	//updateReservation オブジェクトを作成します
    	HashMap<String, AttributeValue> reservationItemKey = new HashMap<String, AttributeValue>();
    	reservationItemKey.put("Id", new AttributeValue("500"));
    	
    	HashMap<String, AttributeValue> reservationExpressionAttributeValues = new HashMap<String, AttributeValue>();
    	reservationExpressionAttributeValues.put(":reservation_status", new AttributeValue("CLOSED"));
    	
    	Update updateReservation = new Update()
    			.withTableName(TABLE_NAME)
    			.withKey(reservationItemKey)
    			.withUpdateExpression("SET ReservationStatus = :reservation_status")
    			.withExpressionAttributeValues(reservationExpressionAttributeValues);
    	
    	//Transactions の実行
    	Collection<TransactWriteItem> actions = Arrays.asList(
        		new TransactWriteItem().withUpdate(updateGuest),
                new TransactWriteItem().withUpdate(updateRoom),
                new TransactWriteItem().withUpdate(updateReservation));
        
        TransactWriteItemsRequest createReservationTransaction = new TransactWriteItemsRequest()
                .withTransactItems(actions)
                .withReturnConsumedCapacity(ReturnConsumedCapacity.TOTAL);

        dynamoDB.transactWriteItems(createReservationTransaction);
        System.out.println("Check-out transaction is successful");
    }