Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
*
* 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 org.apache.cassandra.cql.jdbc;
import static org.apache.cassandra.utils.ByteBufferUtil.bytes;
import static org.apache.cassandra.utils.ByteBufferUtil.string;
import java.nio.ByteBuffer;
import java.nio.charset.CharacterCodingException;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLTransientException;
import java.sql.Types;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import org.apache.cassandra.thrift.Column;
import org.apache.cassandra.thrift.CqlMetadata;
import org.apache.cassandra.thrift.CqlResult;
import org.apache.cassandra.thrift.CqlResultType;
import org.apache.cassandra.thrift.CqlRow;
import org.apache.cassandra.utils.ByteBufferUtil;
public class MetadataResultSets
{
static final String TABLE_CONSTANT = "TABLE";
public static final MetadataResultSets instance = new MetadataResultSets();
// Private Constructor
private MetadataResultSets() {}
/**
* Make a {@code Column} from a column name and a value.
*
* @param name
* the name of the column
* @param value
* the value of the column as a {@code ByteBufffer}
*
* @return {@code Column}
*/
private static final Column makeColumn(String name, ByteBuffer value)
{
return new Column(bytes(name)).setValue(value).setTimestamp(System.currentTimeMillis());
}
private static final CqlRow makeRow(String key, List columnList)
{
return new CqlRow(bytes(key), columnList);
}
private static CqlMetadata makeMetadataAllString(List colNameList)
{
Map namesMap = new HashMap();
Map valuesMap = new HashMap();
for (String name : colNameList)
{
namesMap.put(bytes(name), Entry.ASCII_TYPE);
valuesMap.put(bytes(name), Entry.UTF8_TYPE);
}
return new CqlMetadata(namesMap,valuesMap,Entry.ASCII_TYPE,Entry.UTF8_TYPE);
}
private static CqlMetadata makeMetadata(List entries)
{
Map namesMap = new HashMap();
Map valuesMap = new HashMap();
for (Entry entry : entries)
{
namesMap.put(bytes(entry.name), Entry.ASCII_TYPE);
valuesMap.put(bytes(entry.name), entry.type);
}
return new CqlMetadata(namesMap,valuesMap,Entry.ASCII_TYPE,Entry.UTF8_TYPE);
}
private CqlResult makeCqlResult(Entry[][] rowsOfcolsOfKvps, int position)
{
CqlResult result = new CqlResult(CqlResultType.ROWS);
CqlMetadata meta = null;
CqlRow row = null;
Column column = null;
List columnlist = new LinkedList();
List rowlist = new LinkedList();
List colNamesList = new ArrayList();
for (int rowcnt = 0; rowcnt < rowsOfcolsOfKvps.length; rowcnt++ )
{
colNamesList = new ArrayList();
columnlist = new LinkedList();
for (int colcnt = 0; colcnt < rowsOfcolsOfKvps[0].length; colcnt++ )
{
column = makeColumn(rowsOfcolsOfKvps[rowcnt][colcnt].name,rowsOfcolsOfKvps[rowcnt][colcnt].value);
columnlist.add(column);
colNamesList.add(rowsOfcolsOfKvps[rowcnt][colcnt].name);
}
row = makeRow(rowsOfcolsOfKvps[rowcnt][position-1].name,columnlist);
rowlist.add(row);
}
meta = makeMetadataAllString(colNamesList);
result.setSchema(meta).setRows(rowlist);
return result;
}
private CqlResult makeCqlResult(List> rows, int position) throws CharacterCodingException
{
CqlResult result = new CqlResult(CqlResultType.ROWS);
CqlMetadata meta = null;
CqlRow row = null;
Column column = null;
List columnlist = new LinkedList();
List rowlist = new LinkedList();
assert(!rows.isEmpty());
for (List aRow : rows )
{
columnlist = new LinkedList();
assert (!aRow.isEmpty());
// only need to do it once
if (meta == null) meta = makeMetadata(aRow);
for (Entry entry : aRow )
{
column = makeColumn(entry.name,entry.value);
columnlist.add(column);
}
row = makeRow(string(columnlist.get(position-1).name),columnlist);
rowlist.add(row);
}
result.setSchema(meta).setRows(rowlist);
return result;
}
public CassandraResultSet makeTableTypes(CassandraStatement statement) throws SQLException
{
final Entry[][] tableTypes = { { new Entry("TABLE_TYPE",bytes(TABLE_CONSTANT),Entry.ASCII_TYPE)} };
// use tableTypes with the key in column number 1 (one based)
CqlResult cqlresult = makeCqlResult(tableTypes, 1);
CassandraResultSet result = new CassandraResultSet(statement,cqlresult);
return result;
}
public CassandraResultSet makeCatalogs(CassandraStatement statement) throws SQLException
{
final Entry[][] catalogs = { { new Entry("TABLE_CAT",bytes(statement.connection.getCatalog()),Entry.ASCII_TYPE)} };
// use catalogs with the key in column number 1 (one based)
CqlResult cqlresult = makeCqlResult(catalogs, 1);
CassandraResultSet result = new CassandraResultSet(statement,cqlresult);
return result;
}
public CassandraResultSet makeSchemas(CassandraStatement statement, String schemaPattern) throws SQLException
{
if ("%".equals(schemaPattern)) schemaPattern = null;
// TABLE_SCHEM String => schema name
// TABLE_CATALOG String => catalog name (may be null)
String query = "SELECT keyspace_name FROM system.schema_keyspaces";
if (schemaPattern!=null) query = query + " where keyspace_name = '" + schemaPattern + "'";
String catalog = statement.connection.getCatalog();
Entry entryCatalog = new Entry("TABLE_CATALOG",bytes(catalog),Entry.ASCII_TYPE);
CassandraResultSet result;
List col;
List> rows = new ArrayList>();
// determine the schemas
result = (CassandraResultSet)statement.executeQuery(query);
while (result.next())
{
Entry entrySchema = new Entry("TABLE_SCHEM",bytes(result.getString(1)),Entry.ASCII_TYPE);
col = new ArrayList();
col.add(entrySchema);
col.add(entryCatalog);
rows.add(col);
}
// just return the empty result if there were no rows
if (rows.isEmpty() )return result;
// use schemas with the key in column number 2 (one based)
CqlResult cqlresult;
try
{
cqlresult = makeCqlResult(rows, 1);
}
catch (CharacterCodingException e)
{
throw new SQLTransientException(e);
}
result = new CassandraResultSet(statement,cqlresult);
return result;
}
public CassandraResultSet makeTables(CassandraStatement statement, String schemaPattern, String tableNamePattern) throws SQLException
{
// 1. TABLE_CAT String => table catalog (may be null)
// 2. TABLE_SCHEM String => table schema (may be null)
// 3. TABLE_NAME String => table name
// 4. TABLE_TYPE String => table type. Typical types are "TABLE", "VIEW", "SYSTEM TABLE", "GLOBAL TEMPORARY", "LOCAL TEMPORARY", "ALIAS", "SYNONYM".
// 5. REMARKS String => explanatory comment on the table
// 6. TYPE_CAT String => the types catalog (may be null)
// 7. TYPE_SCHEM String => the types schema (may be null)
// 8. TYPE_NAME String => type name (may be null)
// 9. SELF_REFERENCING_COL_NAME String => name of the designated "identifier" column of a typed table (may be null)
// 10. REF_GENERATION String => specifies how values in SELF_REFERENCING_COL_NAME are created. Values are "SYSTEM", "USER", "DERIVED". (may be null)
if ("%".equals(schemaPattern)) schemaPattern = null;
if ("%".equals(tableNamePattern)) tableNamePattern = null;
// example query to retrieve tables
// SELECT keyspace_name,columnfamily_name,comment from schema_columnfamilies WHERE columnfamily_name = 'Test2';
StringBuilder query = new StringBuilder("SELECT keyspace_name,columnfamily_name,comment FROM system.schema_columnfamilies");
int filterCount = 0;
if (schemaPattern != null) filterCount++;
if (tableNamePattern != null) filterCount++;
// check to see if it is qualified
if (filterCount > 0)
{
String expr = "%s = '%s'";
query.append(" WHERE ");
if (schemaPattern != null)
{
query.append(String.format(expr, "keyspace_name", schemaPattern));
filterCount--;
if (filterCount > 0) query.append(" AND ");
}
if (tableNamePattern != null) query.append(String.format(expr, "columnfamily_name", tableNamePattern));
query.append(" ALLOW FILTERING");
}
// System.out.println(query.toString());
String catalog = statement.connection.getCatalog();
Entry entryCatalog = new Entry("TABLE_CAT", bytes(catalog), Entry.ASCII_TYPE);
Entry entryTableType = new Entry("TABLE_TYPE", bytes(TABLE_CONSTANT), Entry.ASCII_TYPE);
Entry entryTypeCatalog = new Entry("TYPE_CAT", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entryTypeSchema = new Entry("TYPE_SCHEM", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entryTypeName = new Entry("TYPE_NAME", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entrySRCN = new Entry("SELF_REFERENCING_COL_NAME", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entryRefGeneration = new Entry("REF_GENERATION", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
CassandraResultSet result;
List col;
List> rows = new ArrayList>();
// determine the schemas
result = (CassandraResultSet)statement.executeQuery(query.toString());
while (result.next())
{
Entry entrySchema = new Entry("TABLE_SCHEM", bytes(result.getString(1)), Entry.ASCII_TYPE);
Entry entryTableName = new Entry("TABLE_NAME",
(result.getString(2) == null) ? ByteBufferUtil.EMPTY_BYTE_BUFFER : bytes(result.getString(2)),
Entry.ASCII_TYPE);
Entry entryRemarks = new Entry("REMARKS",
(result.getString(3) == null) ? ByteBufferUtil.EMPTY_BYTE_BUFFER : bytes(result.getString(3)),
Entry.ASCII_TYPE);
col = new ArrayList();
col.add(entryCatalog);
col.add(entrySchema);
col.add(entryTableName);
col.add(entryTableType);
col.add(entryRemarks);
col.add(entryTypeCatalog);
col.add(entryTypeSchema);
col.add(entryTypeName);
col.add(entrySRCN);
col.add(entryRefGeneration);
rows.add(col);
}
// just return the empty result if there were no rows
if (rows.isEmpty()) return result;
// use schemas with the key in column number 2 (one based)
CqlResult cqlresult;
try
{
cqlresult = makeCqlResult(rows, 1);
}
catch (CharacterCodingException e)
{
throw new SQLTransientException(e);
}
result = new CassandraResultSet(statement, cqlresult);
return result;
}
public CassandraResultSet makeColumns(CassandraStatement statement, String schemaPattern, String tableNamePattern, String columnNamePattern) throws SQLException
{
// 1.TABLE_CAT String => table catalog (may be null)
// 2.TABLE_SCHEM String => table schema (may be null)
// 3.TABLE_NAME String => table name
// 4.COLUMN_NAME String => column name
// 5.DATA_TYPE int => SQL type from java.sql.Types
// 6.TYPE_NAME String => Data source dependent type name, for a UDT the type name is fully qualified
// 7.COLUMN_SIZE int => column size.
// 8.BUFFER_LENGTH is not used.
// 9.DECIMAL_DIGITS int => the number of fractional digits. Null is returned for data types where DECIMAL_DIGITS is not applicable.
// 10.NUM_PREC_RADIX int => Radix (typically either 10 or 2)
// 11.NULLABLE int => is NULL allowed. - columnNoNulls - might not allow NULL values
// - columnNullable - definitely allows NULL values
// - columnNullableUnknown - nullability unknown
//
// 12.REMARKS String => comment describing column (may be null)
// 13.COLUMN_DEF String => default value for the column, which should be interpreted as a string when the value is enclosed in
// single quotes (may be null)
// 14.SQL_DATA_TYPE int => unused
// 15.SQL_DATETIME_SUB int => unused
// 16.CHAR_OCTET_LENGTH int => for char types the maximum number of bytes in the column
// 17.ORDINAL_POSITION int => index of column in table (starting at 1)
// 18.IS_NULLABLE String => ISO rules are used to determine the nullability for a column. - YES --- if the parameter can include
// NULLs
// - NO --- if the parameter cannot include NULLs
// - empty string --- if the nullability for the parameter is unknown
//
// 19.SCOPE_CATLOG String => catalog of table that is the scope of a reference attribute (null if DATA_TYPE isn't REF)
// 20.SCOPE_SCHEMA String => schema of table that is the scope of a reference attribute (null if the DATA_TYPE isn't REF)
// 21.SCOPE_TABLE String => table name that this the scope of a reference attribure (null if the DATA_TYPE isn't REF)
// 22.SOURCE_DATA_TYPE short => source type of a distinct type or user-generated Ref type, SQL type from java.sql.Types (null if
// DATA_TYPE isn't DISTINCT or user-generated REF)
// 23.IS_AUTOINCREMENT String => Indicates whether this column is auto incremented - YES --- if the column is auto incremented
// - NO --- if the column is not auto incremented
// - empty string --- if it cannot be determined whether the column is auto incremented parameter is unknown
// 24. IS_GENERATEDCOLUMN String => Indicates whether this is a generated column ï YES --- if this a generated column
// - NO --- if this not a generated column
// - empty string --- if it cannot be determined whether this is a generated column
if ("%".equals(schemaPattern)) schemaPattern = null;
if ("%".equals(tableNamePattern)) tableNamePattern = null;
if ("%".equals(columnNamePattern)) columnNamePattern = null;
StringBuilder query = new StringBuilder("SELECT keyspace_name, columnfamily_name, column_name, validator FROM system.schema_columns");
int filterCount = 0;
if (schemaPattern != null) filterCount++;
if (tableNamePattern != null) filterCount++;
if (columnNamePattern != null) filterCount++;
// check to see if it is qualified
if (filterCount > 0)
{
String expr = "%s = '%s'";
query.append(" WHERE ");
if (schemaPattern != null)
{
query.append(String.format(expr, "keyspace_name", schemaPattern));
filterCount--;
if (filterCount > 0) query.append(" AND ");
}
if (tableNamePattern != null)
{
query.append(String.format(expr, "columnfamily_name", tableNamePattern));
filterCount--;
if (filterCount > 0) query.append(" AND ");
}
if (columnNamePattern != null) query.append(String.format(expr, "column_name", columnNamePattern));
query.append(" ALLOW FILTERING");
}
// System.out.println(query.toString());
String catalog = statement.connection.getCatalog();
Entry entryCatalog = new Entry("TABLE_CAT", bytes(catalog), Entry.ASCII_TYPE);
Entry entryBufferLength = new Entry("BUFFER_LENGTH", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.INT32_TYPE);
Entry entryRemarks = new Entry("REMARKS", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entryColumnDef = new Entry("COLUMN_DEF", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entrySQLDataType = new Entry("SQL_DATA_TYPE", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.INT32_TYPE);
Entry entrySQLDateTimeSub = new Entry("SQL_DATETIME_SUB", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.INT32_TYPE);
Entry entryScopeCatalog = new Entry("SCOPE_CATLOG", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entryScopeSchema = new Entry("SCOPE_SCHEMA", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entryScopeTable = new Entry("SCOPE_TABLE", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entrySOURCEDT = new Entry("SOURCE_DATA_TYPE", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.INT32_TYPE);
Entry entryISAutoIncrement = new Entry("IS_AUTOINCREMENT", bytes("NO"), Entry.ASCII_TYPE);
Entry entryISGeneratedColumn = new Entry("IS_GENERATEDCOLUMN", bytes("NO"), Entry.ASCII_TYPE);
List col;
List> rows = new ArrayList>();
int ordinalPosition = 0;
// define the PK columns
List pks = getPrimaryKeys(statement, schemaPattern, tableNamePattern);
Iterator it = pks.iterator();
while (it.hasNext())
{
PKInfo info = it.next();
Entry entrySchema = new Entry("TABLE_SCHEM", bytes(info.schema), Entry.ASCII_TYPE);
Entry entryTableName = new Entry("TABLE_NAME", bytes(info.table),Entry.ASCII_TYPE);
if (columnNamePattern != null && !info.name.contains(columnNamePattern))
{
continue;
}
Entry entryColumnName = new Entry("COLUMN_NAME", bytes(info.name), Entry.ASCII_TYPE);
Entry entryDataType = new Entry("DATA_TYPE", bytes(info.type), Entry.INT32_TYPE);
Entry entryTypeName = new Entry("TYPE_NAME", bytes(info.typeName), Entry.ASCII_TYPE);
int lenght = -1;
AbstractJdbcType jtype = TypesMap.getTypeForComparator(info.typeName);
if (jtype instanceof JdbcBytes) lenght = Integer.MAX_VALUE / 2;
if (jtype instanceof JdbcAscii || jtype instanceof JdbcUTF8) lenght = Integer.MAX_VALUE;
if (jtype instanceof JdbcUUID) lenght = 36;
if (jtype instanceof JdbcInt32) lenght = 4;
if (jtype instanceof JdbcLong) lenght = 8;
Entry entryColumnSize = new Entry("COLUMN_SIZE", bytes(lenght), Entry.INT32_TYPE);
ByteBuffer dd = ByteBufferUtil.EMPTY_BYTE_BUFFER;
// if (jtype instanceof JdbcDouble) dd = bytes(17);
// if (jtype instanceof JdbcFloat) dd = bytes(11);
Entry entryDecimalDigits = new Entry("DECIMAL_DIGITS", dd, Entry.INT32_TYPE);
int npr = 2;
if (jtype != null && (jtype.getJdbcType() == Types.DECIMAL || jtype.getJdbcType() == Types.NUMERIC)) npr = 10;
Entry entryNPR = new Entry("NUM_PREC_RADIX", bytes(npr), Entry.INT32_TYPE);
ByteBuffer charol = ByteBufferUtil.EMPTY_BYTE_BUFFER;
if (jtype instanceof JdbcAscii || jtype instanceof JdbcUTF8) charol = bytes(Integer.MAX_VALUE);
Entry entryCharOctetLength = new Entry("CHAR_OCTET_LENGTH", charol, Entry.INT32_TYPE);
ordinalPosition++;
Entry entryOrdinalPosition = new Entry("ORDINAL_POSITION", bytes(ordinalPosition), Entry.INT32_TYPE);
Entry entryNullable = new Entry("NULLABLE", bytes(DatabaseMetaData.columnNoNulls), Entry.INT32_TYPE);
Entry entryISNullable = new Entry("IS_NULLABLE", bytes("NO"), Entry.ASCII_TYPE);
col = new ArrayList();
col.add(entryCatalog);
col.add(entrySchema);
col.add(entryTableName);
col.add(entryColumnName);
col.add(entryDataType);
col.add(entryTypeName);
col.add(entryColumnSize);
col.add(entryBufferLength);
col.add(entryDecimalDigits);
col.add(entryNPR);
col.add(entryNullable);
col.add(entryRemarks);
col.add(entryColumnDef);
col.add(entrySQLDataType);
col.add(entrySQLDateTimeSub);
col.add(entryCharOctetLength);
col.add(entryOrdinalPosition);
col.add(entryISNullable);
col.add(entryScopeCatalog);
col.add(entryScopeSchema);
col.add(entryScopeTable);
col.add(entrySOURCEDT);
col.add(entryISAutoIncrement);
col.add(entryISGeneratedColumn);
rows.add(col);
}
// define the columns
CassandraResultSet result = (CassandraResultSet) statement.executeQuery(query.toString());
while (result.next())
{
Entry entrySchema = new Entry("TABLE_SCHEM", bytes(result.getString(1)), Entry.ASCII_TYPE);
Entry entryTableName = new Entry("TABLE_NAME",
(result.getString(2) == null) ? ByteBufferUtil.EMPTY_BYTE_BUFFER : bytes(result.getString(2)),
Entry.ASCII_TYPE);
Entry entryColumnName = new Entry("COLUMN_NAME",
(result.getString(3) == null) ? ByteBufferUtil.EMPTY_BYTE_BUFFER : bytes(result.getString(3)),
Entry.ASCII_TYPE);
String validator = result.getString(4);
AbstractJdbcType jtype = TypesMap.getTypeForComparator(validator);
Entry entryDataType = new Entry("DATA_TYPE", bytes(jtype == null ? Types.OTHER : jtype.getJdbcType()), Entry.INT32_TYPE);
int idx = validator.lastIndexOf('.');
Entry entryTypeName = new Entry("TYPE_NAME", bytes(validator.substring(idx + 1)), Entry.ASCII_TYPE);
int lenght = -1;
if (jtype instanceof JdbcBytes) lenght = Integer.MAX_VALUE / 2;
if (jtype instanceof JdbcAscii || jtype instanceof JdbcUTF8) lenght = Integer.MAX_VALUE;
if (jtype instanceof JdbcUUID) lenght = 36;
if (jtype instanceof JdbcInt32) lenght = 4;
if (jtype instanceof JdbcLong) lenght = 8;
Entry entryColumnSize = new Entry("COLUMN_SIZE", bytes(lenght), Entry.INT32_TYPE);
ByteBuffer dd = ByteBufferUtil.EMPTY_BYTE_BUFFER;
// if (jtype instanceof JdbcDouble) dd = bytes(17);
// if (jtype instanceof JdbcFloat) dd = bytes(11);
Entry entryDecimalDigits = new Entry("DECIMAL_DIGITS", dd, Entry.INT32_TYPE);
int npr = 2;
if (jtype != null && (jtype.getJdbcType() == Types.DECIMAL || jtype.getJdbcType() == Types.NUMERIC)) npr = 10;
Entry entryNPR = new Entry("NUM_PREC_RADIX", bytes(npr), Entry.INT32_TYPE);
ByteBuffer charol = ByteBufferUtil.EMPTY_BYTE_BUFFER;
if (jtype instanceof JdbcAscii || jtype instanceof JdbcUTF8) charol = bytes(Integer.MAX_VALUE);
Entry entryCharOctetLength = new Entry("CHAR_OCTET_LENGTH", charol, Entry.INT32_TYPE);
ordinalPosition++;
Entry entryOrdinalPosition = new Entry("ORDINAL_POSITION", bytes(ordinalPosition), Entry.INT32_TYPE);
Entry entryNullable = new Entry("NULLABLE", bytes(DatabaseMetaData.columnNullable), Entry.INT32_TYPE);
Entry entryISNullable = new Entry("IS_NULLABLE", bytes("YES"), Entry.ASCII_TYPE);
col = new ArrayList();
col.add(entryCatalog);
col.add(entrySchema);
col.add(entryTableName);
col.add(entryColumnName);
col.add(entryDataType);
col.add(entryTypeName);
col.add(entryColumnSize);
col.add(entryBufferLength);
col.add(entryDecimalDigits);
col.add(entryNPR);
col.add(entryNullable);
col.add(entryRemarks);
col.add(entryColumnDef);
col.add(entrySQLDataType);
col.add(entrySQLDateTimeSub);
col.add(entryCharOctetLength);
col.add(entryOrdinalPosition);
col.add(entryISNullable);
col.add(entryScopeCatalog);
col.add(entryScopeSchema);
col.add(entryScopeTable);
col.add(entrySOURCEDT);
col.add(entryISAutoIncrement);
col.add(entryISGeneratedColumn);
rows.add(col);
}
// just return the empty result if there were no rows
if (rows.isEmpty()) return result;
// use schemas with the key in column number 2 (one based)
CqlResult cqlresult;
try
{
cqlresult = makeCqlResult(rows, 1);
}
catch (CharacterCodingException e)
{
throw new SQLTransientException(e);
}
result = new CassandraResultSet(statement, cqlresult);
return result;
}
public CassandraResultSet makeIndexes(CassandraStatement statement, String schema, String table, boolean unique, boolean approximate) throws SQLException
{
//1.TABLE_CAT String => table catalog (may be null)
//2.TABLE_SCHEM String => table schema (may be null)
//3.TABLE_NAME String => table name
//4.NON_UNIQUE boolean => Can index values be non-unique. false when TYPE is tableIndexStatistic
//5.INDEX_QUALIFIER String => index catalog (may be null); null when TYPE is tableIndexStatistic
//6.INDEX_NAME String => index name; null when TYPE is tableIndexStatistic
//7.TYPE short => index type: - tableIndexStatistic - this identifies table statistics that are returned in conjuction with a table's index descriptions
//- tableIndexClustered - this is a clustered index
//- tableIndexHashed - this is a hashed index
//- tableIndexOther - this is some other style of index
//
//8.ORDINAL_POSITION short => column sequence number within index; zero when TYPE is tableIndexStatistic
//9.COLUMN_NAME String => column name; null when TYPE is tableIndexStatistic
//10.ASC_OR_DESC String => column sort sequence, "A" => ascending, "D" => descending, may be null if sort sequence is not supported; null when TYPE is tableIndexStatistic
//11.CARDINALITY int => When TYPE is tableIndexStatistic, then this is the number of rows in the table; otherwise, it is the number of unique values in the index.
//12.PAGES int => When TYPE is tableIndexStatisic then this is the number of pages used for the table, otherwise it is the number of pages used for the current index.
//13.FILTER_CONDITION String => Filter condition, if any. (may be null)
StringBuilder query = new StringBuilder("SELECT keyspace_name, columnfamily_name, column_name, component_index, index_name, index_options, index_type FROM system.schema_columns");
int filterCount = 0;
if (schema != null) filterCount++;
if (table != null) filterCount++;
// check to see if it is qualified
if (filterCount > 0)
{
String expr = "%s = '%s'";
query.append(" WHERE ");
if (schema != null)
{
query.append(String.format(expr, "keyspace_name", schema));
filterCount--;
if (filterCount > 0) query.append(" AND ");
}
if (table != null) query.append(String.format(expr, "columnfamily_name", table));
query.append(" ALLOW FILTERING");
}
// System.out.println(query.toString());
String catalog = statement.connection.getCatalog();
Entry entryCatalog = new Entry("TABLE_CAT", bytes(catalog), Entry.ASCII_TYPE);
CassandraResultSet result;
List col;
List> rows = new ArrayList>();
int ordinalPosition = 0;
// define the columns
result = (CassandraResultSet) statement.executeQuery(query.toString());
while (result.next())
{
if (result.getString(7) == null) continue; //if there is no index_type its not an index
ordinalPosition++;
Entry entrySchema = new Entry("TABLE_SCHEM", bytes(result.getString(1)), Entry.ASCII_TYPE);
Entry entryTableName = new Entry("TABLE_NAME",
(result.getString(2) == null) ? ByteBufferUtil.EMPTY_BYTE_BUFFER : bytes(result.getString(2)),
Entry.ASCII_TYPE);
Entry entryNonUnique = new Entry("NON_UNIQUE", bytes("true"),Entry.BOOLEAN_TYPE);
Entry entryIndexQualifier = new Entry("INDEX_QUALIFIER", bytes(catalog), Entry.ASCII_TYPE);
Entry entryIndexName = new Entry("INDEX_NAME", (result.getString(5) == null ? ByteBufferUtil.EMPTY_BYTE_BUFFER : bytes(result.getString(5))), Entry.ASCII_TYPE);
Entry entryType = new Entry("TYPE", bytes(DatabaseMetaData.tableIndexHashed), Entry.INT32_TYPE);
Entry entryOrdinalPosition = new Entry("ORDINAL_POSITION", bytes(ordinalPosition), Entry.INT32_TYPE);
Entry entryColumnName = new Entry("COLUMN_NAME",
(result.getString(3) == null) ? ByteBufferUtil.EMPTY_BYTE_BUFFER : bytes(result.getString(3)),
Entry.ASCII_TYPE);
Entry entryAoD = new Entry("ASC_OR_DESC",ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
Entry entryCardinality = new Entry("CARDINALITY", bytes(-1), Entry.INT32_TYPE);
Entry entryPages = new Entry("PAGES", bytes(-1), Entry.INT32_TYPE);
Entry entryFilter = new Entry("FILTER_CONDITION",ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
col = new ArrayList();
col.add(entryCatalog);
col.add(entrySchema);
col.add(entryTableName);
col.add(entryNonUnique);
col.add(entryIndexQualifier);
col.add(entryIndexName);
col.add(entryType);
col.add(entryOrdinalPosition);
col.add(entryColumnName);
col.add(entryAoD);
col.add(entryCardinality);
col.add(entryPages);
col.add(entryFilter);
rows.add(col);
}
// just return the empty result if there were no rows
if (rows.isEmpty()) return result;
// use schemas with the key in column number 2 (one based)
CqlResult cqlresult;
try
{
cqlresult = makeCqlResult(rows, 1);
}
catch (CharacterCodingException e)
{
throw new SQLTransientException(e);
}
result = new CassandraResultSet(statement, cqlresult);
return result;
}
public List getPrimaryKeys(CassandraStatement statement, String schema, String table) throws SQLException
{
StringBuilder query = new StringBuilder("SELECT keyspace_name, columnfamily_name, key_aliases, key_validator, column_aliases, comparator FROM system.schema_columnfamilies");
int filterCount = 0;
if (schema != null) filterCount++;
if (table != null) filterCount++;
// check to see if it is qualified
if (filterCount > 0)
{
String expr = "%s = '%s'";
query.append(" WHERE ");
if (schema != null)
{
query.append(String.format(expr, "keyspace_name", schema));
filterCount--;
if (filterCount > 0) query.append(" AND ");
}
if (table != null) query.append(String.format(expr, "columnfamily_name", table));
query.append(" ALLOW FILTERING");
}
// System.out.println(query.toString());
List retval = new ArrayList();
CassandraResultSet result = (CassandraResultSet) statement.executeQuery(query.toString());
if (result.next()) // all is reported back in one json row
{
String rschema = result.getString(1);
String rtable = result.getString(2);
String validator = result.getString(4);
String key_aliases = result.getString(3);
buildPKInfo(retval,rschema,rtable,key_aliases,validator);
String comparator = result.getString(6);
String column_aliases = result.getString(5);
buildPKInfo(retval,rschema,rtable,column_aliases,comparator);
}
return retval;
}
private void buildPKInfo(List retval,String schema, String table,String aliases,String validator)
{
String[] typeNames = new String[0];
int[] types = new int[0];
if (validator != null)
{
String[] validatorArray = new String[]{validator};
String check = "CompositeType";
int idx = validator.indexOf(check);
if (idx > 0)
{
validator = validator.substring(idx+check.length()+1,validator.length()-1);
validatorArray = validator.split(",");
}
types = new int[validatorArray.length];
typeNames = new String[validatorArray.length];
for (int i = 0; i < validatorArray.length; i++)
{
AbstractJdbcType jtype = TypesMap.getTypeForComparator(validatorArray[i]);
types[i] = (jtype != null ? jtype.getJdbcType() : Types.OTHER);
int dotidx = validatorArray[i].lastIndexOf('.');
typeNames[i] = validatorArray[i].substring(dotidx + 1);
}
}
if (aliases != null)
{
aliases = aliases.replace("[","");
aliases = aliases.replace("]","");
aliases = aliases.replace("\"","");
if (aliases.trim().length() != 0)
{
String[] kaArray = aliases.split(",");
for (int i = 0; i < kaArray.length; i++)
{
PKInfo pki = new PKInfo();
pki.name = kaArray[i];
pki.schema = schema;
pki.table = table;
pki.type = (i < types.length ? types[i] : Types.OTHER);
pki.typeName = (i < typeNames.length ? typeNames[i] : "unknown");
retval.add(pki);
}
}
}
}
private class PKInfo
{
public String typeName;
public String schema;
public String table;
public String name;
public int type;
}
public CassandraResultSet makePrimaryKeys(CassandraStatement statement, String schema, String table) throws SQLException
{
//1.TABLE_CAT String => table catalog (may be null)
//2.TABLE_SCHEM String => table schema (may be null)
//3.TABLE_NAME String => table name
//4.COLUMN_NAME String => column name
//5.KEY_SEQ short => sequence number within primary key( a value of 1 represents the first column of the primary key, a value of 2 would represent the second column within the primary key).
//6.PK_NAME String => primary key name (may be null)
List pks = getPrimaryKeys(statement, schema, table);
Iterator it = pks.iterator();
String catalog = statement.connection.getCatalog();
Entry entryCatalog = new Entry("TABLE_CAT", bytes(catalog), Entry.ASCII_TYPE);
List col;
List> rows = new ArrayList>();
int seq = 0;
// define the columns
while (it.hasNext())
{
PKInfo info = it.next();
Entry entrySchema = new Entry("TABLE_SCHEM", bytes(info.schema), Entry.ASCII_TYPE);
Entry entryTableName = new Entry("TABLE_NAME", bytes(info.table),Entry.ASCII_TYPE);
Entry entryColumnName = new Entry("COLUMN_NAME", bytes(info.name), Entry.ASCII_TYPE);
seq++;
Entry entryKeySeq = new Entry("KEY_SEQ", bytes(seq), Entry.INT32_TYPE);
Entry entryPKName = new Entry("PK_NAME", ByteBufferUtil.EMPTY_BYTE_BUFFER, Entry.ASCII_TYPE);
col = new ArrayList();
col.add(entryCatalog);
col.add(entrySchema);
col.add(entryTableName);
col.add(entryColumnName);
col.add(entryKeySeq);
col.add(entryPKName);
rows.add(col);
}
// just return the empty result if there were no rows
if (rows.isEmpty()) return new CassandraResultSet();
// use schemas with the key in column number 2 (one based)
CqlResult cqlresult;
try
{
cqlresult = makeCqlResult(rows, 1);
}
catch (CharacterCodingException e)
{
throw new SQLTransientException(e);
}
return new CassandraResultSet(statement, cqlresult);
}
private class Entry
{
static final String UTF8_TYPE = "UTF8Type";
static final String ASCII_TYPE = "AsciiType";
static final String INT32_TYPE = "Int32Type";
static final String BOOLEAN_TYPE = "BooleanType";
String name = null;
ByteBuffer value = null;
String type = null;
private Entry(String name,ByteBuffer value,String type)
{
this.name = name;
this.value = value;
this.type = type;
}
}
}
