Package org.apache.hadoop.yarn.server.federation.store.sql

Class FederationQueryRunner

java.lang.Object

org.apache.hadoop.yarn.server.federation.store.sql.FederationQueryRunner

public class FederationQueryRunner
extends java.lang.Object

QueryRunner is used to execute stored procedure SQL and parse the returned results.

Field Summary

Fields

Modifier and Type	Field	Description
static java.lang.String	DELETE_QUEUE_SQL
static java.lang.String	INSERT_SEQUENCE_TABLE_SQL
static org.slf4j.Logger	LOG
static java.lang.String	QUERY_SEQUENCE_TABLE_SQL
static java.lang.String	UPDATE_SEQUENCE_TABLE_SQL
static java.lang.String	YARN_ROUTER_CURRENT_KEY_ID
static java.lang.String	YARN_ROUTER_SEQUENCE_NUM

Constructor Summary

Constructors

Constructor	Description
FederationQueryRunner()

Method Summary

Modifier and Type	Method	Description
void	close(java.sql.Statement stmt)	Close Statement.
void	deletePolicyByQueue(java.sql.Connection connection, java.lang.String queue)
<T> T	execute(java.sql.Connection conn, java.lang.String procedure, ResultSetHandler<T> rsh, java.lang.Object... params)	Execute Stored Procedure SQL.
void	fillStatement(java.sql.CallableStatement stmt, java.lang.Object... params)	Set Statement parameters.
protected java.sql.CallableStatement	getCallableStatement(java.sql.Connection conn, java.lang.String procedure)	Get CallableStatement from Conn.
protected void	rethrow(java.sql.SQLException cause, java.lang.String sql, java.lang.Object... params)	Re-throw SQL exception.
int	selectOrUpdateSequenceTable(java.sql.Connection connection, java.lang.String sequenceName, boolean isUpdate)	We query or update the SequenceTable.
void	truncateTable(java.sql.Connection connection, java.lang.String tableName)
void	updateSequenceTable(java.sql.Connection connection, java.lang.String sequenceName, int sequenceValue)

Methods inherited from class java.lang.Object

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

Field Detail

YARN_ROUTER_SEQUENCE_NUM

public static final java.lang.String YARN_ROUTER_SEQUENCE_NUM

See Also:

Constant Field Values

YARN_ROUTER_CURRENT_KEY_ID

public static final java.lang.String YARN_ROUTER_CURRENT_KEY_ID

See Also:

Constant Field Values

QUERY_SEQUENCE_TABLE_SQL

public static final java.lang.String QUERY_SEQUENCE_TABLE_SQL

See Also:

Constant Field Values

INSERT_SEQUENCE_TABLE_SQL

public static final java.lang.String INSERT_SEQUENCE_TABLE_SQL

See Also:

Constant Field Values

UPDATE_SEQUENCE_TABLE_SQL

public static final java.lang.String UPDATE_SEQUENCE_TABLE_SQL

See Also:

Constant Field Values

DELETE_QUEUE_SQL

public static final java.lang.String DELETE_QUEUE_SQL

See Also:

Constant Field Values

LOG

public static final org.slf4j.Logger LOG

Constructor Detail

FederationQueryRunner

public FederationQueryRunner()

Method Detail

execute

public <T> T execute(java.sql.Connection conn,
                     java.lang.String procedure,
                     ResultSetHandler<T> rsh,
                     java.lang.Object... params)
              throws java.sql.SQLException

Execute Stored Procedure SQL.

Type Parameters:

T - Generic T.

Parameters:

conn - Database Connection.

procedure - Stored Procedure SQL.

rsh - Result Set handler.

params - List of stored procedure parameters.

Returns:

Stored Procedure Result Set.

Throws:

java.sql.SQLException - An exception occurred when calling a stored procedure.

getCallableStatement

@VisibleForTesting
protected java.sql.CallableStatement getCallableStatement(java.sql.Connection conn,
                                                          java.lang.String procedure)
                                                   throws java.sql.SQLException

Get CallableStatement from Conn.

Parameters:

conn - Database Connection.

procedure - Stored Procedure SQL.

Returns:

CallableStatement.

Throws:

java.sql.SQLException - An exception occurred when calling a stored procedure.

fillStatement

public void fillStatement(java.sql.CallableStatement stmt,
                          java.lang.Object... params)
                   throws java.sql.SQLException

Set Statement parameters.

Parameters:

stmt - CallableStatement.

params - Stored procedure parameters.

Throws:

java.sql.SQLException - An exception occurred when calling a stored procedure.

close

public void close(java.sql.Statement stmt)
           throws java.sql.SQLException

Close Statement.

Parameters:

stmt - CallableStatement.

Throws:

java.sql.SQLException - An exception occurred when calling a stored procedure.

rethrow

protected void rethrow(java.sql.SQLException cause,
                       java.lang.String sql,
                       java.lang.Object... params)
                throws java.sql.SQLException

Re-throw SQL exception.

Parameters:

cause - SQLException.

sql - Stored Procedure SQL.

params - Stored procedure parameters.

Throws:

java.sql.SQLException - An exception occurred when calling a stored procedure.

selectOrUpdateSequenceTable

public int selectOrUpdateSequenceTable(java.sql.Connection connection,
                                       java.lang.String sequenceName,
                                       boolean isUpdate)
                                throws java.sql.SQLException

We query or update the SequenceTable.

Parameters:

connection - database conn.

sequenceName - sequenceName, We currently have 2 sequences, YARN_ROUTER_SEQUENCE_NUM and YARN_ROUTER_CURRENT_KEY_ID.

isUpdate - true, means we will update the SequenceTable, false, we query the SequenceTable.

Returns:

SequenceValue.

Throws:

java.sql.SQLException - An exception occurred when calling a stored procedure.

updateSequenceTable

public void updateSequenceTable(java.sql.Connection connection,
                                java.lang.String sequenceName,
                                int sequenceValue)
                         throws java.sql.SQLException

Throws:

java.sql.SQLException

deletePolicyByQueue

public void deletePolicyByQueue(java.sql.Connection connection,
                                java.lang.String queue)
                         throws java.sql.SQLException

Throws:

java.sql.SQLException

truncateTable

public void truncateTable(java.sql.Connection connection,
                          java.lang.String tableName)
                   throws java.sql.SQLException

Throws:

java.sql.SQLException