OBKV-Tableを使用したデータの挿入

@Test
public void testSingleInsert() throws Exception {
    try {
        // Insert data
        // Use setPartitionKey to set the partition key c1
        // Use addMutateColVal to set the value of c1 as well
        MutationResult result = client.insert("test")
            .setPartitionKey(row(colVal("c1", 1000L)))
            .addMutateColVal(colVal("c1", 1000L), colVal("c2", "test1"), colVal("c3", 3000L))
            .execute();

        assertEquals("Insert should affect 1 row", 1, result.getAffectedRows());
        System.out.println("Single insert successful: " + result + " row affected");
    } finally {
        // Clean up data
        truncateTable();
    }
}

@Test
public void testSingleInsertOrUpdate1() throws Exception {
    try {
        // Insert data
        // Use setPartitionKey to set the partition key c1
        // Use addMutateColVal to set the value of c1 as well
        MutationResult result = client.insertOrUpdate(tableName)
            .setPartitionKey(row(colVal("c1", 1000L)))
            .addMutateColVal(colVal("c1", 1000L), colVal("c2", "test1"), colVal("c3", 3000L))
            .execute();

        assertEquals("Insert should affect 1 row", 1, result.getAffectedRows());
        System.out.println("Single insertOrUpdate successful: " + result + " row affected");
    } finally {
        // Clean up data
        truncateTable();
    }
}

@Test
public void testSingleReplace1() throws Exception {
    try {
        // Insert data
        // Use setPartitionKey to set the partition key c1
        // Use addMutateColVal to set the value of c1 as well
        MutationResult result = client.replace(tableName)
            .setPartitionKey(row(colVal("c1", 1000L)))
            .addMutateColVal(colVal("c1", 1000L), colVal("c2", "test1"), colVal("c3", 3000L))
            .execute();

        assertEquals("Insert should affect 1 row", 1, result.getAffectedRows());
        System.out.println("Single replace successful: " + result + " row affected");
    } finally {
        // Clean up data
        truncateTable();
    }
}

@Test
public void testBatchInsert() throws Exception {
    try {
        // Use BatchOperation for batch insert
        BatchOperation batch = client.batchOperation("test");

        // Add multiple insert operations
        // Use setPartitionKey to set the partition key c1
        // Use addMutateColVal to set the value of c1 as well
        for (int i = 1; i <= 3; i++) {
            Insert insert = new Insert();
            insert.setPartitionKey(row(colVal("c1", (long)(1000 + i))))
                .addMutateColVal(colVal("c1", (long)(1000 + i)), colVal("c2", "batch_test" + i), colVal("c3", (long)(3000 + i)));
            batch.addOperation(insert);
        }

        batch.execute();
        System.out.println("Batch insert successful: 3 operations executed");
    } finally {
        // Clean up data
        truncateTable();
    }
}

@Test
public void testBatchInsertOrUpdate1() throws Exception {
    try {
        // Use BatchOperation for batch insert
        BatchOperation batch = client.batchOperation(tableName);

        // Add multiple insert operations
        // Use setPartitionKey to set the partition key c1
        // Use addMutateColVal to set the value of c1 as well
        for (int i = 1; i <= 3; i++) {
            InsertOrUpdate insertOrUpdate = new InsertOrUpdate();
            insertOrUpdate.setPartitionKey(row(colVal("c1", (long)(1000 + i))))
                .addMutateColVal(colVal("c1", (long)(1000 + i)), colVal("c2", "batch_test" + i), colVal("c3", (long)(3000 + i)));
            batch.addOperation(insertOrUpdate);
        }

        batch.execute();
        System.out.println("Batch insertOrUpdate successful: 3 operations executed");
    } finally {
        // Clean up data
        truncateTable();
    }
}

@Test
public void testBatchReplace1() throws Exception {
    try {
        // Use BatchOperation for batch insert
        BatchOperation batch = client.batchOperation(tableName);

        // Add multiple replace operations
        // Use setPartitionKey to set the partition key c1
        // Use addMutateColVal to set the value of c1 as well
        for (int i = 1; i <= 3; i++) {
            Replace replace = new Replace();
            replace.setPartitionKey(row(colVal("c1", (long)(1000 + i))))
                .addMutateColVal(colVal("c1", (long)(1000 + i)), colVal("c2", "batch_test" + i), colVal("c3", (long)(3000 + i)));
            batch.addOperation(replace);
        }

        batch.execute();
        System.out.println("Batch replace successful: 3 operations executed");
    } finally {
        // Clean up data
        truncateTable();
    }
}