Package com.databricks.jdbc.client.impl.sdk

Class DatabricksSdkClient

java.lang.Object

com.databricks.jdbc.client.impl.sdk.DatabricksSdkClient

All Implemented Interfaces:

DatabricksClient

public class DatabricksSdkClient
extends Object
implements DatabricksClient

Implementation of DatabricksClient interface using Databricks Java SDK.

Constructor Summary

Constructors

Constructor	Description
DatabricksSdkClient(IDatabricksConnectionContext connectionContext)
DatabricksSdkClient(IDatabricksConnectionContext connectionContext, com.databricks.sdk.service.sql.StatementExecutionService statementExecutionService, com.databricks.sdk.core.ApiClient apiClient)

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 warehouse, String catalog, String schema, Map<String,String> sessionConf)	Creates a new session for given warehouse-Id, catalog and session.
void	deleteSession(IDatabricksSession session, ComputeResource warehouse)	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.

Methods inherited from class java.lang.Object

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

Constructor Detail

DatabricksSdkClient

public DatabricksSdkClient(IDatabricksConnectionContext connectionContext)
                    throws DatabricksParsingException

Throws:

DatabricksParsingException

DatabricksSdkClient

public DatabricksSdkClient(IDatabricksConnectionContext connectionContext,
                           com.databricks.sdk.service.sql.StatementExecutionService statementExecutionService,
                           com.databricks.sdk.core.ApiClient apiClient)
                    throws DatabricksParsingException

Throws:

DatabricksParsingException

Method Detail

createSession

public ImmutableSessionInfo createSession(ComputeResource warehouse,
                                          String catalog,
                                          String schema,
                                          Map<String,String> sessionConf)

Description copied from interface: DatabricksClient

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

Specified by:

createSession in interface DatabricksClient

Parameters:

warehouse - underlying SQL-warehouse or all-purpose cluster

catalog - for the session

schema - for the session

sessionConf - session configuration

Returns:

created session

deleteSession

public void deleteSession(IDatabricksSession session,
                          ComputeResource warehouse)
                   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

warehouse - 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)

Description copied from interface: DatabricksClient

Closes a statement in Databricks server

Specified by:

closeStatement in interface DatabricksClient

Parameters:

statementId - statement which should be closed

cancelStatement

public void cancelStatement(String statementId)

Description copied from interface: DatabricksClient

Cancels a statement in Databricks server

Specified by:

cancelStatement in interface DatabricksClient

Parameters:

statementId - statement which should be aborted

getResultChunks

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

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