All Downloads are FREE. Search and download functionalities are using the official Maven repository.
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.
net.snowflake.client.jdbc.SnowflakeStatementV1 Maven / Gradle / Ivy
/*
* Copyright (c) 2012-2017 Snowflake Computing Inc. All rights reserved.
*/
package net.snowflake.client.jdbc;
import net.snowflake.client.core.ParameterBindingDTO;
import net.snowflake.client.core.ResultUtil;
import net.snowflake.client.core.SFBaseResultSet;
import net.snowflake.client.core.SFException;
import net.snowflake.client.core.SFStatement;
import java.sql.BatchUpdateException;
import java.util.ArrayList;
import java.util.List;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.sql.SQLWarning;
import java.sql.Statement;
import java.util.Map;
import net.snowflake.client.core.StmtUtil;
import net.snowflake.client.log.SFLogger;
import net.snowflake.client.log.SFLoggerFactory;
/**
* Snowflake statement
*
* @author jhuang
*/
class SnowflakeStatementV1 implements Statement
{
static final SFLogger logger = SFLoggerFactory.getLogger(SnowflakeStatementV1.class);
protected SnowflakeConnectionV1 connection;
/*
* The maximum number of rows this statement ( should return (0 => all rows).
*/
private int maxRows = 0;
private ResultSet resultSet = null;
private ResultSet currentResultSet = null;
private int fetchSize = 50;
private Boolean isClosed = false;
private int updateCount = -1;
// TODO: escape processing for sql statement
private boolean escapeProcessing = false;
// timeout in seconds
private int queryTimeout = 0;
// max field size limited to 16MB
private final int maxFieldSize = 16777216;
protected SFStatement sfStatement;
/** batch of sql strings added by addBatch */
protected final List batch = new ArrayList<>();
protected SQLWarning sqlWarnings;
SnowflakeStatementV1(SnowflakeConnectionV1 conn)
{
logger.debug(
" public SnowflakeStatement(SnowflakeConnectionV1 conn)");
connection = conn;
sfStatement = new SFStatement(conn.getSfSession());
}
/**
* Execute SQL query
*
* @param sql sql statement
* @return ResultSet
* @throws java.sql.SQLException if @link{#executeQueryInternal(String, Map)} throws an exception
*/
@Override
public ResultSet executeQuery(String sql) throws SQLException
{
return executeQueryInternal(sql, null);
}
/**
* Execute an update statement
*
* @param sql sql statement
* @return number of rows updated
* @throws java.sql.SQLException if @link{#executeUpdateInternal(String, Map)} throws exception
*/
@Override
public int executeUpdate(String sql) throws SQLException
{
return executeUpdateInternal(sql, null);
}
int executeUpdateInternal(String sql,
Map parameterBindings)
throws SQLException
{
if (StmtUtil.checkStageManageCommand(sql) != null)
{
throw new SnowflakeSQLException(ErrorCode.
UNSUPPORTED_STATEMENT_TYPE_IN_EXECUTION_API,
sql.length() > 20 ? sql.substring(0, 20) + "..." : sql);
}
SFBaseResultSet sfResultSet = null;
try
{
sfResultSet = sfStatement.execute(sql, parameterBindings);
sfResultSet.setSession(this.connection.getSfSession());
updateCount = ResultUtil.calculateUpdateCount(sfResultSet);
}
catch (SFException ex)
{
throw new SnowflakeSQLException(ex.getCause(),
ex.getSqlState(), ex.getVendorCode(), ex.getParams());
}
if (updateCount == -1)
{
throw new SnowflakeSQLException(ErrorCode.
UNSUPPORTED_STATEMENT_TYPE_IN_EXECUTION_API,
sql.length() > 20 ? sql.substring(0, 20) + "..." : sql);
}
return updateCount;
}
/**
* Internal method for executing a query with bindings accepted.
*
* @param sql sql statement
* @param parameterBindings parameters bindings
* @return query result set
* @throws SQLException if @link{SFStatement.execute(String)} throws exception
*/
ResultSet executeQueryInternal(
String sql,
Map parameterBindings)
throws SQLException
{
SFBaseResultSet sfResultSet = null;
try
{
sfResultSet = sfStatement.execute(sql, parameterBindings);
sfResultSet.setSession(this.connection.getSfSession());
}
catch (SFException ex)
{
throw new SnowflakeSQLException(ex.getCause(),
ex.getSqlState(), ex.getVendorCode(), ex.getParams());
}
currentResultSet = null;
resultSet = new SnowflakeResultSetV1(sfResultSet, this);
// Fix a bug with getMoreResults returning true after a client
// calling executeQuery which has returned the result set already
return getResultSet();
}
/**
* Execute sql
*
* @param sql sql statement
* @return whether there is result set or not
* @throws java.sql.SQLException if @link{#executeQuery(String)} throws exception
*/
boolean executeInternal(String sql,
Map parameterBindings)
throws SQLException
{
connection.injectedDelay();
logger.debug("execute: " + sql);
String trimmedSql = sql.trim();
if (trimmedSql.length() >= 20
&& trimmedSql.toLowerCase().startsWith(
"set-sf-property"))
{
executeSetProperty(sql);
return false;
}
else
{
SFBaseResultSet sfResultSet = null;
try
{
sfResultSet = sfStatement.execute(sql, parameterBindings);
sfResultSet.setSession(this.connection.getSfSession());
currentResultSet = null;
resultSet = new SnowflakeResultSetV1(sfResultSet, this);
if (connection.getSfSession().isExecuteReturnCountForDML() &&
!sfResultSet.getStatementType().isGenerateResultSet())
{
updateCount = ResultUtil.calculateUpdateCount(sfResultSet);
resultSet = null;
currentResultSet = null;
return false;
}
updateCount = -1;
return true;
}
catch (SFException ex)
{
throw new SnowflakeSQLException(ex.getCause(),
ex.getSqlState(), ex.getVendorCode(), ex.getParams());
}
}
}
/**
* Execute sql
*
* @param sql sql statement
* @return whether there is result set or not
* @throws java.sql.SQLException if @link{#executeQuery(String)} throws exception
*/
@Override
public boolean execute(String sql) throws SQLException
{
return executeInternal(sql, null);
}
@Override
public boolean execute(String sql, int autoGeneratedKeys)
throws SQLException
{
logger.debug(
"public int execute(String sql, int autoGeneratedKeys)");
if (autoGeneratedKeys == Statement.NO_GENERATED_KEYS)
{
return execute(sql);
}
else
{
throw new SQLFeatureNotSupportedException();
}
}
@Override
public boolean execute(String sql, int[] columnIndexes) throws SQLException
{
logger.debug(
"public boolean execute(String sql, int[] columnIndexes)");
throw new SQLFeatureNotSupportedException();
}
@Override
public boolean execute(String sql, String[] columnNames) throws SQLException
{
logger.debug(
"public boolean execute(String sql, String[] columnNames)");
throw new SQLFeatureNotSupportedException();
}
/**
* Batch Execute. If one of the commands in the batch failed, JDBC will
* continuing processing and throw BatchUpdateException after all commands
* are processed.
* @return
* @throws SQLException
*/
@Override
public int[] executeBatch() throws SQLException
{
logger.debug("public int[] executeBatch()");
return executeBatchInternal();
}
/**
* This method will iterate through batch and provide sql and bindings to
* underlying SFStatement to get result set.
*
* Note, array binds use a different code path since only one network
* roundtrip in the array bind execution case.
*
* @return
* @throws SQLException
*/
int[] executeBatchInternal() throws SQLException
{
if (isClosed)
{
throw new SnowflakeSQLException(ErrorCode.STATEMENT_CLOSED);
}
SQLException exceptionReturned = null;
int updateCounts[] = new int[batch.size()];
for (int i=0; i iface) throws SQLException
{
logger.debug("public boolean isWrapperFor(Class iface)");
return iface.isInstance(this);
}
@SuppressWarnings("unchecked")
@Override
public T unwrap(Class iface) throws SQLException
{
logger.debug("public T unwrap(Class iface)");
if (!iface.isInstance(this))
{
throw new SQLException(
this.getClass().getName() + " not unwrappable from " + iface
.getName());
}
return (T) this;
}
@Override
public void closeOnCompletion() throws SQLException
{
logger.debug("public void closeOnCompletion()");
throw new SQLFeatureNotSupportedException();
}
@Override
public boolean isCloseOnCompletion() throws SQLException
{
logger.debug("public boolean isCloseOnCompletion()");
throw new SQLFeatureNotSupportedException();
}
@Override
public void close() throws SQLException
{
logger.debug("public void close()");
currentResultSet = null;
resultSet = null;
isClosed = true;
batch.clear();
sfStatement.close();
}
@Override
public void cancel() throws SQLException
{
logger.debug("public void cancel()");
try
{
sfStatement.cancel();
}
catch (SFException ex)
{
throw new SnowflakeSQLException(ex, ex.getSqlState(),
ex.getVendorCode(), ex.getParams());
}
}
@Override
public void clearWarnings() throws SQLException
{
logger.debug("public void clearWarnings()");
sqlWarnings = null;
}
@Override
public void addBatch(String sql) throws SQLException
{
logger.debug("public void addBatch(String sql)");
if (isClosed)
{
throw new SnowflakeSQLException(ErrorCode.STATEMENT_CLOSED);
}
batch.add(new BatchEntry(sql, null));
}
@Override
public void clearBatch() throws SQLException
{
logger.debug("public void clearBatch()");
if (isClosed)
{
throw new SnowflakeSQLException(ErrorCode.STATEMENT_CLOSED);
}
batch.clear();
}
private void executeSetProperty(final String sql)
{
logger.debug("setting property");
// tokenize the sql
String[] tokens = sql.split("\\s+");
if (tokens == null || tokens.length < 2)
{
return;
}
if ("tracing".equalsIgnoreCase(tokens[1]))
{
if (tokens.length >= 3)
{
/*connection.tracingLevel = Level.parse(tokens[2].toUpperCase());
if (connection.tracingLevel != null)
{
Logger snowflakeLogger = Logger.getLogger("net.snowflake");
snowflakeLogger.setLevel(connection.tracingLevel);
}*/
}
}
else
{
this.sfStatement.executeSetProperty(sql);
}
}
public SFStatement getSfStatement()
{
return sfStatement;
}
protected void appendWarning(SQLWarning w)
{
if (sqlWarnings == null)
{
sqlWarnings = w;
}
else
{
sqlWarnings.setNextWarning(w);
}
}
final class BatchEntry
{
private final String sql;
private final Map parameterBindings;
BatchEntry(String sql,
Map parameterBindings)
{
this.sql = sql;
this.parameterBindings = parameterBindings;
}
public String getSql()
{
return sql;
}
public Map getParameterBindings()
{
return parameterBindings;
}
}
}
