Package com.databricks.jdbc.dbclient.impl.thrift

Class DatabricksThriftServiceClient

java.lang.Object

com.databricks.jdbc.dbclient.impl.thrift.DatabricksThriftServiceClient

All Implemented Interfaces:

IDatabricksClient, IDatabricksMetadataClient

public class DatabricksThriftServiceClient
extends Object
implements IDatabricksClient, IDatabricksMetadataClient

Field Summary

Fields

Modifier and Type	Field	Description
static JdbcLogger	LOGGER

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(IDatabricksComputeResource cluster, String catalog, String schema, Map<String,String> sessionConf)	Creates a new session for given warehouse-Id, catalog and session.
void	deleteSession(IDatabricksSession session, IDatabricksComputeResource cluster)	Deletes a session for given session-Id
DatabricksResultSet	executeStatement(String sql, IDatabricksComputeResource computeResource, Map<Integer,ImmutableSqlParameter> parameters, StatementType statementType, IDatabricksSession session, IDatabricksStatement parentStatement)	Executes a statement in Databricks server
IDatabricksConnectionContext	getConnectionContext()
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
void	resetAccessToken(String newAccessToken)	Update the access token based on new value provided by the customer

Methods inherited from class java.lang.Object

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

Field Detail

LOGGER

public static final JdbcLogger LOGGER

Constructor Detail

DatabricksThriftServiceClient

public DatabricksThriftServiceClient(IDatabricksConnectionContext connectionContext)
                              throws DatabricksParsingException

Throws:

DatabricksParsingException

Method Detail

getConnectionContext

public IDatabricksConnectionContext getConnectionContext()

Specified by:

getConnectionContext in interface IDatabricksClient

resetAccessToken

public void resetAccessToken(String newAccessToken)

Description copied from interface: IDatabricksClient

Update the access token based on new value provided by the customer

Specified by:

resetAccessToken in interface IDatabricksClient

Parameters:

newAccessToken - new access token value

createSession

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

Description copied from interface: IDatabricksClient

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

Specified by:

createSession in interface IDatabricksClient

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,
                          IDatabricksComputeResource cluster)
                   throws DatabricksSQLException

Description copied from interface: IDatabricksClient

Deletes a session for given session-Id

Specified by:

deleteSession in interface IDatabricksClient

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,
                                            IDatabricksComputeResource computeResource,
                                            Map<Integer,ImmutableSqlParameter> parameters,
                                            StatementType statementType,
                                            IDatabricksSession session,
                                            IDatabricksStatement parentStatement)
                                     throws SQLException

Description copied from interface: IDatabricksClient

Executes a statement in Databricks server

Specified by:

executeStatement in interface IDatabricksClient

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: IDatabricksClient

Closes a statement in Databricks server

Specified by:

closeStatement in interface IDatabricksClient

Parameters:

statementId - statement which should be closed

Throws:

DatabricksSQLException

cancelStatement

public void cancelStatement(String statementId)
                     throws DatabricksSQLException

Description copied from interface: IDatabricksClient

Cancels a statement in Databricks server

Specified by:

cancelStatement in interface IDatabricksClient

Parameters:

statementId - statement which should be aborted

Throws:

DatabricksSQLException

getResultChunks

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

Description copied from interface: IDatabricksClient

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

Specified by:

getResultChunks in interface IDatabricksClient

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: IDatabricksMetadataClient

Returns information about types supported by Databricks server

Specified by:

listTypeInfo in interface IDatabricksMetadataClient

Throws:

DatabricksSQLException

listCatalogs

public DatabricksResultSet listCatalogs(IDatabricksSession session)
                                 throws SQLException

Description copied from interface: IDatabricksMetadataClient

Returns the list of catalogs

Specified by:

listCatalogs in interface IDatabricksMetadataClient

Throws:

SQLException

listSchemas

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

Description copied from interface: IDatabricksMetadataClient

Returns the list of schemas

Specified by:

listSchemas in interface IDatabricksMetadataClient

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: IDatabricksMetadataClient

Returns the list of tables

Specified by:

listTables in interface IDatabricksMetadataClient

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: IDatabricksMetadataClient

Returns list of table types

Specified by:

listTableTypes in interface IDatabricksMetadataClient

listColumns

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

Description copied from interface: IDatabricksMetadataClient

Returns the list of columns

Specified by:

listColumns in interface IDatabricksMetadataClient

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: IDatabricksMetadataClient

Returns the list of functions

Specified by:

listFunctions in interface IDatabricksMetadataClient

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: IDatabricksMetadataClient

Returns the list of primary keys

Specified by:

listPrimaryKeys in interface IDatabricksMetadataClient

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