org.openl.rules.ext.cassandra.CassandraOperations Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of org.openl.rules.ext.cassandra Show documentation

OpenL - Cassandra Support

There is a newer version: 5.27.9

Show newest version

package org.openl.rules.ext.cassandra;

import java.util.Map;

import com.datastax.oss.driver.api.core.CqlSession;
import com.datastax.oss.driver.api.core.cql.ResultSet;
import com.datastax.oss.driver.api.core.cql.SimpleStatement;
import com.datastax.oss.driver.api.core.cql.SimpleStatementBuilder;
import com.datastax.oss.driver.api.core.cql.Statement;

public class CassandraOperations {

    private static class CassandraOperationsHolder {
        private static final CassandraOperations INSTANCE = new CassandraOperations();
    }

    private static final String CASSANDRA_PROPERTIES_FILE_NAME = "cassandra.properties";

    public static CassandraOperations getInstance() {
        return CassandraOperationsHolder.INSTANCE;
    }

    private volatile CqlSession session;

    public CqlSession getSession() {
        if (session == null) {
            synchronized (this) {
                if (session == null) {
                    session = CqlSession.builder()
                        .withConfigLoader(ConfigLoader.fromProjectResource(CASSANDRA_PROPERTIES_FILE_NAME))
                        .build();
                }
            }
        }
        return session;
    }

    public static ResultSet cassandraExecute(String query) {
        return CassandraOperations.getInstance().getSession().execute(query);
    }

    public static ResultSet cassandraExecute(String query, Object... values) {
        SimpleStatement statement = SimpleStatement.builder(query).addPositionalValues(values).build();
        return CassandraOperations.getInstance().getSession().execute(statement);
    }

    public static ResultSet cassandraExecute(String query, Map values) {
        SimpleStatementBuilder statementBuilder = SimpleStatement.builder(query);
        if (values != null) {
            values.forEach(statementBuilder::addNamedValue);
        }
        return CassandraOperations.getInstance().getSession().execute(statementBuilder.build());
    }

    public static ResultSet cassandraExecute(Statement statement) {
        return CassandraOperations.getInstance().getSession().execute(statement);
    }
}