/*
* Licensed 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.facebook.presto.cassandra.util;
import com.datastax.driver.core.querybuilder.QueryBuilder;
import com.datastax.driver.core.querybuilder.Select;
import com.datastax.driver.core.querybuilder.Select.Selection;
import com.facebook.presto.cassandra.CassandraColumnHandle;
import com.facebook.presto.cassandra.CassandraTableHandle;
import com.facebook.presto.spi.ConnectorColumnHandle;
import com.fasterxml.jackson.core.io.JsonStringEncoder;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
public final class CassandraCqlUtils
{
private CassandraCqlUtils()
{
}
private static final String[] KEYWORDS = {"ADD", "ALL", "ALLOW", "ALTER", "AND", "APPLY",
"ASC", "ASCII", "AUTHORIZE", "BATCH", "BEGIN", "BIGINT", "BLOB", "BOOLEAN", "BY",
"CLUSTERING", "COLUMNFAMILY", "COMPACT", "COUNT", "COUNTER", "CREATE", "DECIMAL",
"DELETE", "DESC", "DOUBLE", "DROP", "FILTERING", "FLOAT", "FROM", "GRANT", "IN",
"INDEX", "INET", "INSERT", "INT", "INTO", "KEY", "KEYSPACE", "KEYSPACES", "LIMIT",
"LIST", "MAP", "MODIFY", "NORECURSIVE", "NOSUPERUSER", "OF", "ON", "ORDER", "PASSWORD",
"PERMISSION", "PERMISSIONS", "PRIMARY", "RENAME", "REVOKE", "SCHEMA", "SELECT", "SET",
"STORAGE", "SUPERUSER", "TABLE", "TEXT", "TIMESTAMP", "TIMEUUID", "TO", "TOKEN",
"TRUNCATE", "TTL", "TYPE", "UNLOGGED", "UPDATE", "USE", "USER", "USERS", "USING",
"UUID", "VALUES", "VARCHAR", "VARINT", "WHERE", "WITH", "WRITETIME"};
private static final Set<String> keywords = new HashSet<>(Arrays.asList(KEYWORDS));
public static final String EMPTY_COLUMN_NAME = "__empty__";
public static String validSchemaName(String identifier)
{
return validIdentifier(identifier);
}
public static String validTableName(String identifier)
{
return validIdentifier(identifier);
}
public static String validColumnName(String identifier)
{
if (identifier.isEmpty() || identifier.equals(EMPTY_COLUMN_NAME)) {
return "\"\"";
}
return validIdentifier(identifier);
}
private static String validIdentifier(String identifier)
{
if (!identifier.equals(identifier.toLowerCase())) {
return quoteIdentifier(identifier);
}
if (keywords.contains(identifier.toUpperCase())) {
return quoteIdentifier(identifier);
}
return identifier;
}
private static String quoteIdentifier(String identifier)
{
return '"' + identifier + '"';
}
public static String quoteStringLiteral(String string)
{
return "'" + string.replace("'", "''") + "'";
}
public static String quoteStringLiteralForJson(String string)
{
return '"' + new String(JsonStringEncoder.getInstance().quoteAsUTF8(string)) + '"';
}
public static void appendSelectColumns(StringBuilder stringBuilder, List<? extends ConnectorColumnHandle> columns)
{
appendSelectColumns(stringBuilder, columns, true);
}
private static void appendSelectColumns(StringBuilder stringBuilder, List<? extends ConnectorColumnHandle> columns, boolean first)
{
for (ConnectorColumnHandle column : columns) {
if (first) {
first = false;
}
else {
stringBuilder.append(",");
}
stringBuilder.append(validColumnName(((CassandraColumnHandle) column).getName()));
}
}
public static String cqlNameToSqlName(String name)
{
if (name.isEmpty()) {
return EMPTY_COLUMN_NAME;
}
return name;
}
public static String sqlNameToCqlName(String name)
{
if (name.equals(EMPTY_COLUMN_NAME)) {
return "";
}
return name;
}
public static Selection select(List<CassandraColumnHandle> columns)
{
Selection selection = QueryBuilder.select();
for (CassandraColumnHandle column : columns) {
selection.column(validColumnName(column.getName()));
}
return selection;
}
public static Select selectFrom(CassandraTableHandle tableHandle, List<CassandraColumnHandle> columns)
{
return from(select(columns), tableHandle);
}
public static Select from(Selection selection, CassandraTableHandle tableHandle)
{
String schema = validSchemaName(tableHandle.getSchemaName());
String table = validTableName(tableHandle.getTableName());
return selection.from(schema, table);
}
public static Select selectDistinctFrom(CassandraTableHandle tableHandle, List<CassandraColumnHandle> columns)
{
return from(select(columns).distinct(), tableHandle);
}
public static Select selectCountAllFrom(CassandraTableHandle tableHandle)
{
String schema = validSchemaName(tableHandle.getSchemaName());
String table = validTableName(tableHandle.getTableName());
return QueryBuilder.select().countAll().from(schema, table);
}
}