com.datastax.dse.driver.internal.querybuilder.schema.DefaultAlterDseTable 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.dse.driver.internal.querybuilder.schema;
import static com.datastax.oss.driver.internal.querybuilder.schema.Utils.appendSet;
import com.datastax.dse.driver.api.querybuilder.schema.AlterDseTableAddColumnEnd;
import com.datastax.dse.driver.api.querybuilder.schema.AlterDseTableDropColumnEnd;
import com.datastax.dse.driver.api.querybuilder.schema.AlterDseTableRenameColumnEnd;
import com.datastax.dse.driver.api.querybuilder.schema.AlterDseTableStart;
import com.datastax.dse.driver.api.querybuilder.schema.AlterDseTableWithOptionsEnd;
import com.datastax.dse.driver.api.querybuilder.schema.DseGraphEdgeSide;
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.internal.querybuilder.CqlHelper;
import com.datastax.oss.driver.internal.querybuilder.ImmutableCollections;
import com.datastax.oss.driver.internal.querybuilder.schema.OptionsUtils;
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 DefaultAlterDseTable
implements AlterDseTableStart,
AlterDseTableAddColumnEnd,
AlterDseTableDropColumnEnd,
AlterDseTableRenameColumnEnd,
AlterDseTableWithOptionsEnd,
BuildableQuery {
private final CqlIdentifier keyspace;
private final CqlIdentifier tableName;
private final ImmutableMap columnsToAddInOrder;
private final ImmutableSet columnsToAdd;
private final ImmutableSet columnsToAddStatic;
private final ImmutableSet columnsToDrop;
private final ImmutableMap columnsToRename;
private final CqlIdentifier columnToAlter;
private final DataType columnToAlterType;
private final DseTableVertexOperation vertexOperation;
private final DseTableEdgeOperation edgeOperation;
private final ImmutableMap options;
private final boolean dropCompactStorage;
public DefaultAlterDseTable(@NonNull CqlIdentifier tableName) {
this(null, tableName);
}
public DefaultAlterDseTable(@Nullable CqlIdentifier keyspace, @NonNull CqlIdentifier tableName) {
this(
keyspace,
tableName,
false,
ImmutableMap.of(),
ImmutableSet.of(),
ImmutableSet.of(),
ImmutableSet.of(),
ImmutableMap.of(),
null,
null,
null,
null,
ImmutableMap.of());
}
public DefaultAlterDseTable(
@Nullable CqlIdentifier keyspace,
@NonNull CqlIdentifier tableName,
boolean dropCompactStorage,
@NonNull ImmutableMap columnsToAddInOrder,
@NonNull ImmutableSet columnsToAdd,
@NonNull ImmutableSet columnsToAddStatic,
@NonNull ImmutableSet columnsToDrop,
@NonNull ImmutableMap columnsToRename,
@Nullable CqlIdentifier columnToAlter,
@Nullable DataType columnToAlterType,
@Nullable DseTableVertexOperation vertexOperation,
@Nullable DseTableEdgeOperation edgeOperation,
@NonNull ImmutableMap options) {
this.keyspace = keyspace;
this.tableName = tableName;
this.dropCompactStorage = dropCompactStorage;
this.columnsToAddInOrder = columnsToAddInOrder;
this.columnsToAdd = columnsToAdd;
this.columnsToAddStatic = columnsToAddStatic;
this.columnsToDrop = columnsToDrop;
this.columnsToRename = columnsToRename;
this.columnToAlter = columnToAlter;
this.columnToAlterType = columnToAlterType;
this.vertexOperation = vertexOperation;
this.edgeOperation = edgeOperation;
this.options = options;
}
@NonNull
@Override
public AlterDseTableAddColumnEnd addColumn(
@NonNull CqlIdentifier columnName, @NonNull DataType dataType) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
ImmutableCollections.append(columnsToAddInOrder, columnName, dataType),
appendSet(columnsToAdd, columnName),
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
vertexOperation,
edgeOperation,
options);
}
@NonNull
@Override
public AlterDseTableAddColumnEnd addStaticColumn(
@NonNull CqlIdentifier columnName, @NonNull DataType dataType) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
ImmutableCollections.append(columnsToAddInOrder, columnName, dataType),
columnsToAdd,
appendSet(columnsToAddStatic, columnName),
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
vertexOperation,
edgeOperation,
options);
}
@NonNull
@Override
public BuildableQuery dropCompactStorage() {
return new DefaultAlterDseTable(
keyspace,
tableName,
true,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
vertexOperation,
edgeOperation,
options);
}
@NonNull
@Override
public AlterDseTableDropColumnEnd dropColumns(@NonNull CqlIdentifier... columnNames) {
ImmutableSet.Builder builder =
ImmutableSet.builder().addAll(columnsToDrop);
for (CqlIdentifier columnName : columnNames) {
builder = builder.add(columnName);
}
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
builder.build(),
columnsToRename,
columnToAlter,
columnToAlterType,
vertexOperation,
edgeOperation,
options);
}
@NonNull
@Override
public AlterDseTableRenameColumnEnd renameColumn(
@NonNull CqlIdentifier from, @NonNull CqlIdentifier to) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
ImmutableCollections.append(columnsToRename, from, to),
columnToAlter,
columnToAlterType,
vertexOperation,
edgeOperation,
options);
}
@NonNull
@Override
public BuildableQuery alterColumn(@NonNull CqlIdentifier columnName, @NonNull DataType dataType) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnName,
dataType,
vertexOperation,
edgeOperation,
options);
}
@NonNull
@Override
public BuildableQuery withVertexLabel(@Nullable CqlIdentifier vertexLabelId) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
new DseTableVertexOperation(DseTableGraphOperationType.WITH, vertexLabelId),
edgeOperation,
options);
}
@NonNull
@Override
public BuildableQuery withoutVertexLabel(@Nullable CqlIdentifier vertexLabelId) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
new DseTableVertexOperation(DseTableGraphOperationType.WITHOUT, vertexLabelId),
edgeOperation,
options);
}
@NonNull
@Override
public BuildableQuery withEdgeLabel(
@Nullable CqlIdentifier edgeLabelId,
@NonNull DseGraphEdgeSide from,
@NonNull DseGraphEdgeSide to) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
vertexOperation,
new DseTableEdgeOperation(DseTableGraphOperationType.WITH, edgeLabelId, from, to),
options);
}
@NonNull
@Override
public BuildableQuery withoutEdgeLabel(@Nullable CqlIdentifier edgeLabelId) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
vertexOperation,
new DseTableEdgeOperation(DseTableGraphOperationType.WITHOUT, edgeLabelId, null, null),
options);
}
@NonNull
@Override
public AlterDseTableWithOptionsEnd withOption(@NonNull String name, @NonNull Object value) {
return new DefaultAlterDseTable(
keyspace,
tableName,
dropCompactStorage,
columnsToAddInOrder,
columnsToAdd,
columnsToAddStatic,
columnsToDrop,
columnsToRename,
columnToAlter,
columnToAlterType,
vertexOperation,
edgeOperation,
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 (!columnsToAdd.isEmpty()) {
builder.append(" ADD ");
if (columnsToAdd.size() > 1) {
builder.append('(');
}
boolean first = true;
for (Map.Entry column : columnsToAddInOrder.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 (columnsToAdd.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 (vertexOperation != null) {
builder.append(' ').append(vertexOperation.getType()).append(' ');
vertexOperation.append(builder);
} else if (edgeOperation != null) {
builder.append(' ').append(edgeOperation.getType()).append(' ');
edgeOperation.append(builder);
} 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 getColumnsToAddInOrder() {
return columnsToAddInOrder;
}
@NonNull
public ImmutableSet getColumnsToAddRegular() {
return columnsToAdd;
}
@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;
}
@Nullable
public DseTableVertexOperation getVertexOperation() {
return vertexOperation;
}
@Nullable
public DseTableEdgeOperation getEdgeOperation() {
return edgeOperation;
}
public boolean isDropCompactStorage() {
return dropCompactStorage;
}
}