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.
com.databricks.jdbc.client.impl.helper.MetadataResultSetBuilder Maven / Gradle / Ivy
package com.databricks.jdbc.client.impl.helper;
import static com.databricks.jdbc.client.impl.helper.CommandConstants.*;
import static com.databricks.jdbc.client.impl.helper.MetadataResultConstants.*;
import com.databricks.jdbc.client.StatementType;
import com.databricks.jdbc.core.DatabricksResultSet;
import com.databricks.jdbc.core.DatabricksSQLException;
import com.databricks.sdk.service.sql.StatementState;
import com.databricks.sdk.service.sql.StatementStatus;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
public class MetadataResultSetBuilder {
public static DatabricksResultSet getFunctionsResult(ResultSet resultSet, String catalog)
throws SQLException {
List> rows = getRowsForFunctions(resultSet, FUNCTION_COLUMNS, catalog);
return buildResultSet(FUNCTION_COLUMNS, rows, GET_FUNCTIONS_STATEMENT_ID);
}
public static DatabricksResultSet getColumnsResult(ResultSet resultSet) throws SQLException {
List> rows = getRows(resultSet, COLUMN_COLUMNS);
return buildResultSet(COLUMN_COLUMNS, rows, METADATA_STATEMENT_ID);
}
public static DatabricksResultSet getCatalogsResult(ResultSet resultSet) throws SQLException {
List> rows = getRows(resultSet, CATALOG_COLUMNS);
return buildResultSet(CATALOG_COLUMNS, rows, GET_CATALOGS_STATEMENT_ID);
}
public static DatabricksResultSet getSchemasResult(ResultSet resultSet, String catalog)
throws SQLException {
List> rows = getRowsForSchemas(resultSet, SCHEMA_COLUMNS, catalog);
return buildResultSet(SCHEMA_COLUMNS, rows, METADATA_STATEMENT_ID);
}
public static DatabricksResultSet getTablesResult(ResultSet resultSet, String[] tableTypes)
throws SQLException {
List allowedTableTypes = List.of(tableTypes);
List> rows =
getRows(resultSet, TABLE_COLUMNS).stream()
.filter(row -> allowedTableTypes.contains(row.get(3))) // Filtering based on table type
.collect(Collectors.toList());
return buildResultSet(TABLE_COLUMNS, rows, GET_TABLES_STATEMENT_ID);
}
public static DatabricksResultSet getTableTypesResult() {
return buildResultSet(TABLE_TYPE_COLUMNS, TABLE_TYPES_ROWS, GET_TABLE_TYPE_STATEMENT_ID);
}
public static DatabricksResultSet getTableTypesResult(List> rows) {
return buildResultSet(TABLE_TYPE_COLUMNS, rows, GET_TABLE_TYPE_STATEMENT_ID);
}
public static DatabricksResultSet getTypeInfoResult(List> rows) {
return buildResultSet(TYPE_INFO_COLUMNS, rows, GET_TYPE_INFO_STATEMENT_ID);
}
public static DatabricksResultSet getPrimaryKeysResult(ResultSet resultSet) throws SQLException {
List> rows = getRows(resultSet, PRIMARY_KEYS_COLUMNS);
return buildResultSet(PRIMARY_KEYS_COLUMNS, rows, METADATA_STATEMENT_ID);
}
private static List> getRows(ResultSet resultSet, List columns)
throws SQLException {
List> rows = new ArrayList<>();
while (resultSet.next()) {
List row = new ArrayList<>();
for (ResultColumn column : columns) {
Object object = null;
switch (column.getColumnName()) {
case "NUM_PREC_RADIX":
object = 10;
row.add(object);
continue;
case "NULLABLE":
object = 2;
row.add(object);
continue;
case "SQL_DATA_TYPE":
case "SQL_DATETIME_SUB":
object = 0;
row.add(object);
continue;
case "IS_NULLABLE":
object = "YES";
row.add(object);
continue;
case "IS_AUTOINCREMENT":
case "IS_GENERATEDCOLUMN":
object = "";
row.add(object);
continue;
}
try {
object = resultSet.getObject(column.getResultSetColumnName());
} catch (DatabricksSQLException e) {
if (column.getColumnName().equals("DATA_TYPE")) {
String typeVal = resultSet.getString("columnType");
if (typeVal.contains("(")) typeVal = typeVal.substring(0, typeVal.indexOf('('));
object = getCode(typeVal);
} else if (column.getColumnName().equals("CHAR_OCTET_LENGTH")) {
String typeVal = resultSet.getString("columnType");
String octetLength =
typeVal.contains("(") ? typeVal.substring(typeVal.indexOf('(') + 1) : "";
if (octetLength.contains(",")) {
octetLength = octetLength.substring(0, octetLength.indexOf(","));
}
object = octetLength.isEmpty() ? 0 : Integer.parseInt(octetLength);
} else {
// Remove non-relevant columns from the obtained result set
object = null;
}
}
if (column.getColumnName().equals("COLUMN_SIZE") && object == null) object = 0;
row.add(object);
}
rows.add(row);
}
return rows;
}
static int getCode(String s) {
switch (s) {
case "STRING":
return 12;
case "INT":
return 4;
case "DOUBLE":
return 8;
case "FLOAT":
return 6;
case "BOOLEAN":
return 16;
case "DATE":
return 91;
case "TIMESTAMP":
return 93;
case "DECIMAL":
return 3;
case "BINARY":
return -2;
case "ARRAY":
return 2003;
case "MAP":
return 2002;
case "STRUCT":
return 2002;
case "UNIONTYPE":
return 2002;
case "BYTE":
return -6;
case "SHORT":
return 5;
case "LONG":
return -5;
case "NULL":
return 0;
case "VOID":
return 0;
case "CHAR":
return 1;
case "VARCHAR":
return 12;
case "CHARACTER":
return 1;
case "BIGINT":
return -5;
case "TINYINT":
return -6;
case "SMALLINT":
return 5;
case "INTEGER":
return 4;
}
return 0;
}
private static List> getRowsForFunctions(
ResultSet resultSet, List columns, String catalog) throws SQLException {
List> rows = new ArrayList<>();
while (resultSet.next()) {
List row = new ArrayList<>();
for (ResultColumn column : columns) {
if (column.getColumnName().equals("FUNCTION_CAT")) {
row.add(catalog);
continue;
}
Object object;
try {
object = resultSet.getObject(column.getResultSetColumnName());
if (object == null) {
object = NULL_STRING;
}
} catch (DatabricksSQLException e) {
// Remove non-relevant columns from the obtained result set
object = NULL_STRING;
}
row.add(object);
}
rows.add(row);
}
return rows;
}
private static List> getRowsForSchemas(
ResultSet resultSet, List columns, String catalog) throws SQLException {
// TODO(PECO-1677): Remove this method once the server side ResultSet metadata contains catalogs
List> rows = new ArrayList<>();
while (resultSet.next()) {
List row = new ArrayList<>();
for (ResultColumn column : columns) {
if (column.getColumnName().equals("TABLE_CATALOG")) {
row.add(catalog);
continue;
}
Object object;
try {
object = resultSet.getObject(column.getResultSetColumnName());
if (object == null) {
object = NULL_STRING;
}
} catch (DatabricksSQLException e) {
// Remove non-relevant columns from the obtained result set
object = NULL_STRING;
}
row.add(object);
}
rows.add(row);
}
return rows;
}
private static DatabricksResultSet buildResultSet(
List columns, List> rows, String statementId) {
return new DatabricksResultSet(
new StatementStatus().setState(StatementState.SUCCEEDED),
statementId,
columns.stream().map(ResultColumn::getColumnName).collect(Collectors.toList()),
columns.stream().map(ResultColumn::getColumnTypeString).collect(Collectors.toList()),
columns.stream().map(ResultColumn::getColumnTypeInt).collect(Collectors.toList()),
columns.stream().map(ResultColumn::getColumnPrecision).collect(Collectors.toList()),
rows,
StatementType.METADATA);
}
public static DatabricksResultSet getCatalogsResult(List> rows) {
return buildResultSet(CATALOG_COLUMNS, rows, GET_CATALOGS_STATEMENT_ID);
}
public static DatabricksResultSet getSchemasResult(List> rows) {
return buildResultSet(SCHEMA_COLUMNS, rows, METADATA_STATEMENT_ID);
}
public static DatabricksResultSet getTablesResult(List> rows) {
return buildResultSet(TABLE_COLUMNS_ALL_PURPOSE, rows, GET_TABLES_STATEMENT_ID);
}
public static DatabricksResultSet getColumnsResult(List> rows) {
return buildResultSet(COLUMN_COLUMNS_ALL_PURPOSE, rows, METADATA_STATEMENT_ID);
}
public static DatabricksResultSet getPrimaryKeysResult(List> rows) {
return buildResultSet(PRIMARY_KEYS_COLUMNS_ALL_PURPOSE, rows, METADATA_STATEMENT_ID);
}
public static DatabricksResultSet getFunctionsResult(List> rows) {
return buildResultSet(FUNCTION_COLUMNS_ALL_PURPOSE, rows, GET_FUNCTIONS_STATEMENT_ID);
}
}
