How to batch insert in YCQL API in Java?

General
1

The docs at Batch operations in YCQL | YugabyteDB Docs say:

BatchStatement batch = new BatchStatement();

But this does not work, as the BatchStatement class is abstract. There are no imports mentioned in the docs (that should be fixed), so I can only assume that the correct import is:

import com.datastax.oss.driver.api.core.cql.BatchStatement;

And this is coming from the (AFAIK) newest library:

com.yugabyte:java-driver-core:4.6.0-yb-11

There is a class DefaultBatchStatement and a BatchStatementBuilder, which all require additional parameters and appear to implement a different API than is used in the Yugabyte docs.

So am I doing something wrong or are the docs out of date?

2

Hi, here are some example with the import:

another one
There’s also this test:

3

Ok thanks. So it seems these are examples for an older driver (3.8.0-yb-x). The docs refer to driver 4.6.0-yb-x, which apparently has a newer API. I’ll look into whether I can figure out the new API and get it to work.

1 Like
4

Thanks, notified the doc team to look at it

5

The new API works like this:

        BatchStatementBuilder builder = new BatchStatementBuilder(BatchType.UNLOGGED);
        PreparedStatement stmt = session.prepare("...some cql...");
        Object[] values1 = ...;
        Object[] values2 = ...;
        builder.addStatement(stmt.bind(values1));
        builder.addStatement(stmt.bind(values2));
        ...
        session.execute(builder.build());

I am not sure what those BatchTypes really mean in the context of YugabyteDB, but I hope UNLOGGED is ok for mass data (like sensor data).

1 Like
6

Here is a sample code snippet with all the imports. There are no changes in how the client API behaves from 3.x to 4.x driver versions. However, the client APIs have moved under different packages -

package com.yugabyte.sample.apps;

import java.net.InetSocketAddress;
import java.util.List;

import com.datastax.oss.driver.api.core.CqlSession;
import com.datastax.oss.driver.api.core.DefaultConsistencyLevel;
import com.datastax.oss.driver.api.core.cql.BatchStatement;
import com.datastax.oss.driver.api.core.cql.BoundStatement;
import com.datastax.oss.driver.api.core.cql.DefaultBatchType;
import com.datastax.oss.driver.api.core.cql.PreparedStatement;
import com.datastax.oss.driver.api.core.cql.ResultSet;
import com.datastax.oss.driver.api.core.cql.Row;
import com.datastax.oss.driver.api.core.cql.SimpleStatement;


public class YcqlDemoApp {
	public static void main(String[] args) {

		try {
			
			CqlSession session = CqlSession.builder()
					.addContactPoint(new InetSocketAddress("127.0.0.1", 9042))
					.build();
			
			// Create keyspace 'ybdemo' if it does not exist.
			String createKeyspace = "CREATE KEYSPACE IF NOT EXISTS ybdemo;";
			session.execute(createKeyspace);
			System.out.println("Created keyspace ybdemo");


			// Create table 'employee', if it does not exist.
			String createTable = "CREATE TABLE IF NOT EXISTS ybdemo.employee (id int PRIMARY KEY, " + "name varchar, "
					+ "age int, " + "language varchar);";
			session.execute(createTable);
			
			
			SimpleStatement insertStmt1 = SimpleStatement
					.builder("INSERT INTO ybdemo.employee (id, name, age, language) VALUES (1, 'John', 35, 'Java');")
					.build();
			
			SimpleStatement insertStmt2 = SimpleStatement
					.builder("INSERT INTO ybdemo.employee (id, name, age, language) VALUES (3453453, 'John', 35, 'Java');")
					.build();
			
			PreparedStatement preparedInsrt1 = session.prepare(insertStmt1);
			PreparedStatement preparedInsrt2 = session.prepare(insertStmt2);
			BoundStatement boundSelectStmt1 = preparedInsrt1.bind();
			BoundStatement boundSelectStmt2 = preparedInsrt2.bind();
			BatchStatement batch = BatchStatement.newInstance(DefaultBatchType.UNLOGGED,
					boundSelectStmt1, boundSelectStmt2);
			session.execute(batch);
			
			
			SimpleStatement simpleSelect = SimpleStatement
					.builder("SELECT name, age, language FROM ybdemo.employee WHERE id = ?")
					.setConsistencyLevel(DefaultConsistencyLevel.YB_CONSISTENT_PREFIX).build();

			PreparedStatement preparedSelect = session.prepare(simpleSelect);
			BoundStatement boundSelectStatement = preparedSelect.bind().setInt(0, 1);
			ResultSet selectResult = session.execute(boundSelectStatement);
			List<Row> rows = selectResult.all();
			String name = rows.get(0).getString(0);
			int age = rows.get(0).getInt(1);
			String language = rows.get(0).getString(2);
			System.out.println("Query returned " + rows.size() + " row: " + "name=" + name + ", age=" + age
					+ ", language: " + language);
			

			session.close();
		} catch (Exception e) {
			System.err.println("Error: " + e.getMessage());
		}
	}
}