Examples with CassandraField org.locationtech.geowave.datastore.cassandra.CassandraRow.CassandraField used on opensource projects

Example 1 with CassandraField

use of org.locationtech.geowave.datastore.cassandra.CassandraRow.CassandraField in project geowave by locationtech.

the class CassandraOperations method createTable.

private boolean createTable(final String indexName) {
    synchronized (CREATE_TABLE_MUTEX) {
        try {
            if (!indexExists(indexName)) {
                final String tableName = getCassandraSafeName(indexName);
                CreateTable create = addOptions(CassandraField.GW_PARTITION_ID_KEY.addPartitionKey(getCreateTable(tableName)));
                CassandraField[] fields = CassandraField.values();
                if (DataIndexUtils.isDataIndex(tableName)) {
                    fields = Arrays.stream(fields).filter(f -> f.isDataIndexColumn()).filter(f -> !f.isPartitionKey()).toArray(i -> new CassandraField[i]);
                }
                for (final CassandraField f : fields) {
                    create = f.addColumn(create);
                }
                executeCreateTable(create, tableName);
                return true;
            }
        } catch (final IOException e) {
            LOGGER.error("Unable to create table '" + indexName + "'", e);
        }
    }
    return false;
}

Example 2 with CassandraField

use of org.locationtech.geowave.datastore.cassandra.CassandraRow.CassandraField in project geowave by locationtech.

the class CassandraOperations method getBatchedWrite.

public BatchedWrite getBatchedWrite(final String tableName) {
    PreparedStatement preparedWrite;
    final String safeTableName = getCassandraSafeName(tableName);
    final boolean isDataIndex = DataIndexUtils.isDataIndex(tableName);
    synchronized (state.preparedWritesPerTable) {
        preparedWrite = state.preparedWritesPerTable.get(safeTableName);
        if (preparedWrite == null) {
            final InsertInto insert = getInsert(safeTableName);
            CassandraField[] fields = CassandraField.values();
            if (isDataIndex) {
                fields = Arrays.stream(fields).filter(f -> f.isDataIndexColumn()).toArray(i -> new CassandraField[i]);
            }
            RegularInsert regInsert = null;
            for (final CassandraField f : fields) {
                regInsert = (regInsert != null ? regInsert : insert).value(f.getFieldName(), QueryBuilder.bindMarker(f.getBindMarkerName()));
            }
            preparedWrite = session.prepare(regInsert.build());
            state.preparedWritesPerTable.put(safeTableName, preparedWrite);
        }
    }
    return new BatchedWrite(session, preparedWrite, isDataIndex ? 1 : options.getBatchWriteSize(), isDataIndex, options.isVisibilityEnabled());
}