com.datastax.oss.driver.internal.querybuilder.schema.DefaultAlterTable Maven / Gradle / Ivy
The newest version!
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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.datastax.oss.driver.internal.querybuilder.schema;
import static com.datastax.oss.driver.internal.querybuilder.schema.Utils.appendSet;
import com.datastax.oss.driver.api.core.CqlIdentifier;
import com.datastax.oss.driver.api.core.type.DataType;
import com.datastax.oss.driver.api.querybuilder.BuildableQuery;
import com.datastax.oss.driver.api.querybuilder.schema.AlterTableAddColumnEnd;
import com.datastax.oss.driver.api.querybuilder.schema.AlterTableDropColumnEnd;
import com.datastax.oss.driver.api.querybuilder.schema.AlterTableRenameColumnEnd;
import com.datastax.oss.driver.api.querybuilder.schema.AlterTableStart;
import com.datastax.oss.driver.api.querybuilder.schema.AlterTableWithOptionsEnd;
import com.datastax.oss.driver.internal.querybuilder.CqlHelper;
import com.datastax.oss.driver.internal.querybuilder.ImmutableCollections;
import com.datastax.oss.driver.shaded.guava.common.collect.ImmutableMap;
import com.datastax.oss.driver.shaded.guava.common.collect.ImmutableSet;
import edu.umd.cs.findbugs.annotations.NonNull;
import edu.umd.cs.findbugs.annotations.Nullable;
import java.util.Map;
import net.jcip.annotations.Immutable;
@Immutable
public class DefaultAlterTable
implements AlterTableStart,
AlterTableAddColumnEnd,
AlterTableDropColumnEnd,
AlterTableRenameColumnEnd,
AlterTableWithOptionsEnd,
BuildableQuery {
private final CqlIdentifier keyspace;
private final CqlIdentifier tableName;
private final ImmutableMap allColumnsToAddInOrder;
private final ImmutableSet columnsToAddRegular;
private final ImmutableSet columnsToAddStatic;
private final ImmutableSet columnsToDrop;
private final ImmutableMap columnsToRename;
private final CqlIdentifier columnToAlter;
private final DataType columnToAlterType;
private final ImmutableMap options;
private final boolean dropCompactStorage;
public DefaultAlterTable(@NonNull CqlIdentifier tableName) {
this(null, tableName);
}
public DefaultAlterTable(@Nullable CqlIdentifier keyspace, @NonNull CqlIdentifier tableName) {
this(
keyspace,
tableName,
false,
ImmutableMap.of(),
ImmutableSet.of(),
ImmutableSet.of(),
ImmutableSet.of(),
ImmutableMap.of(),
null,
null,
ImmutableMap.of());
}
public DefaultAlterTable(
@Nullable CqlIdentifier keyspace,
@NonNull CqlIdentifier tableName,
boolean dropCompactStorage,
@NonNull ImmutableMap allColumnsToAddInOrder,
@NonNull ImmutableSet columnsToAddRegular,
@NonNull ImmutableSet columnsToAddStatic,
@NonNull ImmutableSet columnsToDrop,
@NonNull ImmutableMap columnsToRename,
@Nullable CqlIdentifier columnToAlter,
@Nullable DataType columnToAlterType,
@NonNull ImmutableMap options) {
this.keyspace = keyspace;
this.tableName = tableName;
this.dropCompactStorage = dropCompactStorage;
this.allColumnsToAddInOrder = allColumnsToAddInOrder;
this.columnsToAddRegular = columnsToAddRegular;
this.columnsToAddStatic = columnsToAddStatic;
this.columnsToDrop = columnsToDrop;
this.columnsToRename = columnsToRename;
this.columnToAlter = columnToAlter;
this.columnToAlterType = columnToAlterType;
this.options = options;
}
@NonNull
@Override
public AlterTableAddColumnEnd addColumn(
@NonNull CqlIdentifier columnName, @NonNull DataType dataType) {
return new DefaultAlterTable(
keyspace,
tableName,
dropCompactStorage,
ImmutableCollections.append(allColumnsToAddInOrder, columnName, dataType),
appendSet(columnsToAddRegular, columnName),
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
options);
}
@NonNull
@Override
public AlterTableAddColumnEnd addStaticColumn(
@NonNull CqlIdentifier columnName, @NonNull DataType dataType) {
return new DefaultAlterTable(
keyspace,
tableName,
dropCompactStorage,
ImmutableCollections.append(allColumnsToAddInOrder, columnName, dataType),
columnsToAddRegular,
appendSet(columnsToAddStatic, columnName),
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
options);
}
@NonNull
@Override
public BuildableQuery dropCompactStorage() {
return new DefaultAlterTable(
keyspace,
tableName,
true,
allColumnsToAddInOrder,
columnsToAddRegular,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
options);
}
@NonNull
@Override
public AlterTableDropColumnEnd dropColumns(@NonNull CqlIdentifier... columnNames) {
ImmutableSet.Builder builder =
ImmutableSet.builder().addAll(columnsToDrop);
for (CqlIdentifier columnName : columnNames) {
builder = builder.add(columnName);
}
return new DefaultAlterTable(
keyspace,
tableName,
dropCompactStorage,
allColumnsToAddInOrder,
columnsToAddRegular,
columnsToAddStatic,
builder.build(),
columnsToRename,
columnToAlter,
columnToAlterType,
options);
}
@NonNull
@Override
public AlterTableRenameColumnEnd renameColumn(
@NonNull CqlIdentifier from, @NonNull CqlIdentifier to) {
return new DefaultAlterTable(
keyspace,
tableName,
dropCompactStorage,
allColumnsToAddInOrder,
columnsToAddRegular,
columnsToAddStatic,
columnsToDrop,
ImmutableCollections.append(columnsToRename, from, to),
columnToAlter,
columnToAlterType,
options);
}
@NonNull
@Override
public BuildableQuery alterColumn(@NonNull CqlIdentifier columnName, @NonNull DataType dataType) {
return new DefaultAlterTable(
keyspace,
tableName,
dropCompactStorage,
allColumnsToAddInOrder,
columnsToAddRegular,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnName,
dataType,
options);
}
@NonNull
@Override
public AlterTableWithOptionsEnd withOption(@NonNull String name, @NonNull Object value) {
return new DefaultAlterTable(
keyspace,
tableName,
dropCompactStorage,
allColumnsToAddInOrder,
columnsToAddRegular,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
ImmutableCollections.append(options, name, value));
}
@NonNull
@Override
public String asCql() {
StringBuilder builder = new StringBuilder("ALTER TABLE ");
CqlHelper.qualify(keyspace, tableName, builder);
if (columnToAlter != null) {
return builder
.append(" ALTER ")
.append(columnToAlter.asCql(true))
.append(" TYPE ")
.append(columnToAlterType.asCql(true, true))
.toString();
} else if (!allColumnsToAddInOrder.isEmpty()) {
builder.append(" ADD ");
if (allColumnsToAddInOrder.size() > 1) {
builder.append('(');
}
boolean first = true;
for (Map.Entry column : allColumnsToAddInOrder.entrySet()) {
if (first) {
first = false;
} else {
builder.append(',');
}
builder
.append(column.getKey().asCql(true))
.append(' ')
.append(column.getValue().asCql(true, true));
if (columnsToAddStatic.contains(column.getKey())) {
builder.append(" STATIC");
}
}
if (allColumnsToAddInOrder.size() > 1) {
builder.append(')');
}
return builder.toString();
} else if (!columnsToDrop.isEmpty()) {
boolean moreThanOneDrop = columnsToDrop.size() > 1;
CqlHelper.appendIds(
columnsToDrop,
builder,
moreThanOneDrop ? " DROP (" : " DROP ",
",",
moreThanOneDrop ? ")" : "");
return builder.toString();
} else if (!columnsToRename.isEmpty()) {
builder.append(" RENAME ");
boolean first = true;
for (Map.Entry entry : columnsToRename.entrySet()) {
if (first) {
first = false;
} else {
builder.append(" AND ");
}
builder
.append(entry.getKey().asCql(true))
.append(" TO ")
.append(entry.getValue().asCql(true));
}
return builder.toString();
} else if (dropCompactStorage) {
return builder.append(" DROP COMPACT STORAGE").toString();
} else if (!options.isEmpty()) {
return builder.append(OptionsUtils.buildOptions(options, true)).toString();
}
// While this is incomplete, we should return partially build query at this point for toString
// purposes.
return builder.toString();
}
@Override
public String toString() {
return asCql();
}
@NonNull
@Override
public Map getOptions() {
return options;
}
@Nullable
public CqlIdentifier getKeyspace() {
return keyspace;
}
@NonNull
public CqlIdentifier getTable() {
return tableName;
}
@NonNull
public ImmutableMap getAllColumnsToAddInOrder() {
return allColumnsToAddInOrder;
}
@NonNull
public ImmutableSet getColumnsToAddRegular() {
return columnsToAddRegular;
}
@NonNull
public ImmutableSet getColumnsToAddStatic() {
return columnsToAddStatic;
}
@NonNull
public ImmutableSet getColumnsToDrop() {
return columnsToDrop;
}
@NonNull
public ImmutableMap getColumnsToRename() {
return columnsToRename;
}
@Nullable
public CqlIdentifier getColumnToAlter() {
return columnToAlter;
}
@Nullable
public DataType getColumnToAlterType() {
return columnToAlterType;
}
public boolean isDropCompactStorage() {
return dropCompactStorage;
}
}