com.stratio.connector.cassandra.statements.CreateTableStatement Maven / Gradle / Ivy
The newest version!
/*
* Licensed to STRATIO (C) under one or more contributor license agreements.
* See the NOTICE file distributed with this work for additional information
* regarding copyright ownership. The STRATIO (C) licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package com.stratio.connector.cassandra.statements;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import com.stratio.crossdata.common.exceptions.ExecutionException;
import com.stratio.crossdata.common.data.ColumnName;
import com.stratio.crossdata.common.metadata.ColumnMetadata;
import com.stratio.crossdata.common.metadata.TableMetadata;
/**
* Class that models a {@code CREATE TABLE} statement of the META language.
*/
public class CreateTableStatement {
private static final int PRIMARY_SINGLE = 1;
private static final int PRIMARY_COMPOSED = 2;
private static final int PRIMARY_AND_CLUSTERING_SPECIFIED = 3;
/**
* The name of the target table.
*/
private String tableName;
/**
* A map with the name of the columns in the table and the associated data type.
*/
private Map tableColumns;
/**
* The list of columns that are part of the primary key.
*/
private List primaryKey;
/**
* The list of columns that are part of the clustering key.
*/
private List clusterKey;
/**
* The list of properties of the table.
*/
private String properties;
/**
* The type of primary key. Accepted values are:
*
* - 1: If the primary key contains a single column.
* - 2: If the primary key is composed of several columns but it does not contain a clustering
* key.
* - 3: If both the primary key and clustering key are specified.
*
*/
private int primaryKeyType;
/**
* Whether the table should be created only if not exists.
*/
private boolean ifNotExists;
/**
* Whether the table should be created with a set of properties.
*/
private boolean withProperties = false;
/**
* Catalog.
*/
private String catalog;
/**
* Indicates if there is a catalog specified in the table name.
*/
private boolean catalogInc;
/**
* Class Constructor.
* @param tableMetadata The metadata of the table.
* @param primaryKey The primary key of the table.
* @param partitionKey The partition key of the table.
* @param clusterKey The cluster key of the table.
* @param primaryKeyType The type of the primary key of the table.
* @param properties The specific properties of the table that will be created.
* @param ifNotExists the condition of creation of the table.
* @throws ExecutionException
*/
public CreateTableStatement(TableMetadata tableMetadata,
List primaryKey, List partitionKey, List clusterKey,
int primaryKeyType, String properties, boolean ifNotExists) throws ExecutionException {
this.tableName = tableMetadata.getName().getName();
this.catalog = tableMetadata.getName().getCatalogName().getName();
this.catalogInc = true;
this.tableColumns = tableMetadata.getColumns();
this.primaryKey = primaryKey;
this.clusterKey = clusterKey;
this.primaryKeyType = primaryKeyType;
this.ifNotExists = ifNotExists;
if (properties.length() > 0) {
this.withProperties = true;
this.properties = properties;
}
if (partitionKey == null || partitionKey.size() == 0) {
throw new ExecutionException("PrimaryKey must contain partition Key");
} else if (clusterKey == null && primaryKeyType == PRIMARY_AND_CLUSTERING_SPECIFIED) {
throw new ExecutionException(
"ClusterKey must exist with the type og primaryKey specified");
}
}
/**
* Obtain the single primary key.
* @return a string with the primary key.
*/
public String getSinglePKString() {
StringBuilder sb = new StringBuilder(" (");
Set keySet = tableColumns.keySet();
int i = 0;
for (ColumnName column : keySet) {
if (i != 0) {
sb.append(", ");
}
i = 1;
String key = column.getName().substring(column.getName().lastIndexOf(".") + 1);
String vp = tableColumns.get(column).getColumnType().toString();
sb.append(key).append(" ").append(vp);
if (key.equals(primaryKey.get(0).getName())) {
sb.append(" PRIMARY KEY");
}
}
sb.append(")");
return sb.toString();
}
/**
* Obtain the composite primary key.
* @return a string with the primary key.
*/
public String getCompositePKString() {
StringBuilder sb = new StringBuilder("PRIMARY KEY");
if (primaryKeyType == PRIMARY_COMPOSED) {
sb.append("(");
Iterator pks = primaryKey.iterator();
while (pks.hasNext()) {
sb.append(pks.next().getName());
if (pks.hasNext()) {
sb.append(", ");
}
}
sb.append(")");
}
if (primaryKeyType == PRIMARY_AND_CLUSTERING_SPECIFIED) {
sb.append("((");
Iterator pks = primaryKey.iterator();
while (pks.hasNext()) {
sb.append(pks.next().getName());
if (pks.hasNext()) {
sb.append(", ");
}
}
sb.append(")");
for (ColumnName key : clusterKey) {
sb.append(", ").append(key.getName());
}
sb.append(")");
}
return sb.toString();
}
/**
* Get the query of create table in Cassandra language.
* @return th String with the query.
*/
@Override
public String toString() {
StringBuilder sb = new StringBuilder("Create table ");
if (ifNotExists) {
sb.append("IF NOT EXISTS ");
}
if (catalogInc) {
sb.append(catalog).append(".");
}
sb.append(tableName);
if (primaryKeyType == PRIMARY_SINGLE) {
sb.append(getSinglePKString());
} else {
Set keySet = tableColumns.keySet();
sb.append(" (");
for (ColumnName key : keySet) {
String vp = tableColumns.get(key).getColumnType().toString();
sb.append(key.getName()).append(" ").append(vp).append(", ");
}
sb.append(getCompositePKString()).append(")");
}
if (withProperties) {
sb.append(" WITH ").append(properties);
}
return sb.toString();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy