Package com.databricks.jdbc.client.impl.thrift

Class DatabricksThriftServiceClient

java.lang.Object

com.databricks.jdbc.client.impl.thrift.DatabricksThriftServiceClient

All Implemented Interfaces:

DatabricksClient, DatabricksMetadataClient

public class DatabricksThriftServiceClient
extends Object
implements DatabricksClient, DatabricksMetadataClient

Constructor Summary

Constructors

Constructor	Description
DatabricksThriftServiceClient(IDatabricksConnectionContext connectionContext)

Method Summary

Modifier and Type	Method	Description
void	cancelStatement(String statementId)	Cancels a statement in Databricks server
void	closeStatement(String statementId)	Closes a statement in Databricks server
ImmutableSessionInfo	createSession(ComputeResource cluster, String catalog, String schema, Map<String,String> sessionConf)	Creates a new session for given warehouse-Id, catalog and session.
void	deleteSession(IDatabricksSession session, ComputeResource cluster)	Deletes a session for given session-Id
DatabricksResultSet	executeStatement(String sql, ComputeResource computeResource, Map<Integer,ImmutableSqlParameter> parameters, StatementType statementType, IDatabricksSession session, IDatabricksStatement parentStatement)	Executes a statement in Databricks server
Collection<ExternalLink>	getResultChunks(String statementId, long chunkIndex)	Fetches the chunk details for given chunk index and statement-Id.
DatabricksResultSet	listCatalogs(IDatabricksSession session)	Returns the list of catalogs
DatabricksResultSet	listColumns(IDatabricksSession session, String catalog, String schemaNamePattern, String tableNamePattern, String columnNamePattern)	Returns the list of columns
DatabricksResultSet	listFunctions(IDatabricksSession session, String catalog, String schemaNamePattern, String functionNamePattern)	Returns the list of functions
DatabricksResultSet	listPrimaryKeys(IDatabricksSession session, String catalog, String schema, String table)	Returns the list of primary keys
DatabricksResultSet	listSchemas(IDatabricksSession session, String catalog, String schemaNamePattern)	Returns the list of schemas
DatabricksResultSet	listTables(IDatabricksSession session, String catalog, String schemaNamePattern, String tableNamePattern, String[] tableTypes)	Returns the list of tables
DatabricksResultSet	listTableTypes(IDatabricksSession session)	Returns list of table types
DatabricksResultSet	listTypeInfo(IDatabricksSession session)	Returns information about types supported by Databricks server

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DatabricksThriftServiceClient

public DatabricksThriftServiceClient(IDatabricksConnectionContext connectionContext)
                              throws DatabricksParsingException

Throws:

DatabricksParsingException

Method Detail

createSession

public ImmutableSessionInfo createSession(ComputeResource cluster,
                                          String catalog,
                                          String schema,
                                          Map<String,String> sessionConf)
                                   throws DatabricksSQLException

Description copied from interface: DatabricksClient

Creates a new session for given warehouse-Id, catalog and session.

Specified by:

createSession in interface DatabricksClient

Parameters:

cluster - underlying SQL-warehouse or all-purpose cluster

catalog - for the session

schema - for the session

sessionConf - session configuration

Returns:

created session

Throws:

DatabricksSQLException

deleteSession

public void deleteSession(IDatabricksSession session,
                          ComputeResource cluster)
                   throws DatabricksSQLException

Description copied from interface: DatabricksClient

Deletes a session for given session-Id

Specified by:

deleteSession in interface DatabricksClient

Parameters:

session - for which the session should be deleted

cluster - underlying SQL-warehouse or all-purpose cluster

Throws:

DatabricksSQLException

executeStatement

public DatabricksResultSet executeStatement(String sql,
                                            ComputeResource computeResource,
                                            Map<Integer,ImmutableSqlParameter> parameters,
                                            StatementType statementType,
                                            IDatabricksSession session,
                                            IDatabricksStatement parentStatement)
                                     throws SQLException

Description copied from interface: DatabricksClient

Executes a statement in Databricks server

Specified by:

executeStatement in interface DatabricksClient

Parameters:

sql - SQL statement that needs to be executed

computeResource - underlying SQL-warehouse or all-purpose cluster

parameters - SQL parameters for the statement

statementType - type of statement (metadata, update or generic SQL)

session - underlying session

parentStatement - statement instance if called from a statement

Returns:

response for statement execution

Throws:

SQLException

closeStatement

public void closeStatement(String statementId)
                    throws DatabricksSQLException

Description copied from interface: DatabricksClient

Closes a statement in Databricks server

Specified by:

closeStatement in interface DatabricksClient

Parameters:

statementId - statement which should be closed

Throws:

DatabricksSQLException

cancelStatement

public void cancelStatement(String statementId)
                     throws DatabricksSQLException

Description copied from interface: DatabricksClient

Cancels a statement in Databricks server

Specified by:

cancelStatement in interface DatabricksClient

Parameters:

statementId - statement which should be aborted

Throws:

DatabricksSQLException

getResultChunks

public Collection<ExternalLink> getResultChunks(String statementId,
                                                long chunkIndex)
                                         throws DatabricksSQLException

Description copied from interface: DatabricksClient

Fetches the chunk details for given chunk index and statement-Id.

Specified by:

getResultChunks in interface DatabricksClient

Parameters:

statementId - statement-Id for which chunk should be fetched

chunkIndex - chunkIndex for which chunk should be fetched

Throws:

DatabricksSQLException

listTypeInfo

public DatabricksResultSet listTypeInfo(IDatabricksSession session)
                                 throws DatabricksSQLException

Description copied from interface: DatabricksMetadataClient

Returns information about types supported by Databricks server

Specified by:

listTypeInfo in interface DatabricksMetadataClient

Throws:

DatabricksSQLException

listCatalogs

public DatabricksResultSet listCatalogs(IDatabricksSession session)
                                 throws SQLException

Description copied from interface: DatabricksMetadataClient

Returns the list of catalogs

Specified by:

listCatalogs in interface DatabricksMetadataClient

Throws:

SQLException

listSchemas

public DatabricksResultSet listSchemas(IDatabricksSession session,
                                       String catalog,
                                       String schemaNamePattern)
                                throws SQLException

Description copied from interface: DatabricksMetadataClient

Returns the list of schemas

Specified by:

listSchemas in interface DatabricksMetadataClient

Parameters:

session - underlying session

catalog - catalogName which must match to catalog in database

schemaNamePattern - must match to schema name in database (can be a regex pattern or absolute name)

Returns:

a DatabricksResultSet representing list of schemas

Throws:

SQLException

listTables

public DatabricksResultSet listTables(IDatabricksSession session,
                                      String catalog,
                                      String schemaNamePattern,
                                      String tableNamePattern,
                                      String[] tableTypes)
                               throws SQLException

Description copied from interface: DatabricksMetadataClient

Returns the list of tables

Specified by:

listTables in interface DatabricksMetadataClient

Parameters:

session - underlying session

catalog - catalogName which must match to catalog in database

schemaNamePattern - must match to schema name in database (can be a regex pattern or absolute name)

tableNamePattern - must match to table name in database (can be a regex pattern or absolute name)

Returns:

a DatabricksResultSet representing list of tables

Throws:

SQLException

listTableTypes

public DatabricksResultSet listTableTypes(IDatabricksSession session)

Description copied from interface: DatabricksMetadataClient

Returns list of table types

Specified by:

listTableTypes in interface DatabricksMetadataClient

listColumns

public DatabricksResultSet listColumns(IDatabricksSession session,
                                       String catalog,
                                       String schemaNamePattern,
                                       String tableNamePattern,
                                       String columnNamePattern)
                                throws DatabricksSQLException

Description copied from interface: DatabricksMetadataClient

Returns the list of columns

Specified by:

listColumns in interface DatabricksMetadataClient

Parameters:

session - underlying session

catalog - catalogName which must match to catalog in database

schemaNamePattern - must match to schema name in database (can be a regex pattern or absolute name)

tableNamePattern - must match to table name in database (can be a regex pattern or absolute name)

columnNamePattern - must match to column name in database (can be a regex pattern or absolute name)

Returns:

a DatabricksResultSet representing list of columns

Throws:

DatabricksSQLException

listFunctions

public DatabricksResultSet listFunctions(IDatabricksSession session,
                                         String catalog,
                                         String schemaNamePattern,
                                         String functionNamePattern)
                                  throws DatabricksSQLException

Description copied from interface: DatabricksMetadataClient

Returns the list of functions

Specified by:

listFunctions in interface DatabricksMetadataClient

Parameters:

session - underlying session

catalog - catalogName which must match to catalog in database

schemaNamePattern - must match to schema name in database (can be a regex pattern or absolute name)

functionNamePattern - must match to function name in database (can be a regex pattern or absolute name)

Returns:

a DatabricksResultSet representing list of functions

Throws:

DatabricksSQLException

listPrimaryKeys

public DatabricksResultSet listPrimaryKeys(IDatabricksSession session,
                                           String catalog,
                                           String schema,
                                           String table)
                                    throws SQLException

Description copied from interface: DatabricksMetadataClient

Returns the list of primary keys

Specified by:

listPrimaryKeys in interface DatabricksMetadataClient

Parameters:

session - underlying session

catalog - catalogName which must match to catalog in database

schema - must match to a schema in database

table - must match to a table in database

Returns:

a DatabricksResultSet representing list of functions

Throws:

SQLException