package io.ebean.datasource.pool;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

Implements the Statement methods for ExtendedPreparedStatement.

PreparedStatements should always be used and the intention is that there
should be no use of Statement at all. The implementation here is generally
for the case where someone uses the Statement api on an ExtendedPreparedStatement.
/**
 * Implements the Statement methods for ExtendedPreparedStatement.
 * <p>
 * PreparedStatements should always be used and the intention is that there
 * should be no use of Statement at all. The implementation here is generally
 * for the case where someone uses the Statement api on an ExtendedPreparedStatement.
 */
abstract class ExtendedStatement extends PreparedStatementDelegator {

  
The pooled connection this Statement belongs to.
/**
   * The pooled connection this Statement belongs to.
   */
  final PooledConnection pooledConnection;

  
Create the ExtendedStatement for a given pooledConnection.
/**
   * Create the ExtendedStatement for a given pooledConnection.
   */
  ExtendedStatement(PooledConnection pooledConnection, PreparedStatement delegate) {
    super(delegate);
    this.pooledConnection = pooledConnection;
  }

  
Put the statement back into the statement cache.
/**
   * Put the statement back into the statement cache.
   */
  @Override
  public abstract void close() throws SQLException;

  
Return the underlying connection.
/**
   * Return the underlying connection.
   */
  @Override
  public Connection getConnection() throws SQLException {
    try {
      return delegate.getConnection();
    } catch (SQLException ex) {
      pooledConnection.markWithError(ex);
      throw ex;
    }
  }

  
Add the sql for batch execution.
/**
   * Add the sql for batch execution.
   */
  @Override
  public void addBatch(String sql) throws SQLException {
    try {
      pooledConnection.setLastStatement(sql);
      delegate.addBatch(sql);
    } catch (SQLException ex) {
      pooledConnection.markWithError(ex);
      throw ex;
    }
  }

  
Execute the sql.
/**
   * Execute the sql.
   */
  @Override
  public boolean execute(String sql) throws SQLException {
    try {
      pooledConnection.setLastStatement(sql);
      return delegate.execute(sql);
    } catch (SQLException ex) {
      pooledConnection.markWithError(ex);
      throw ex;
    }
  }

  
Execute the query.
/**
   * Execute the query.
   */
  @Override
  public ResultSet executeQuery(String sql) throws SQLException {
    try {
      pooledConnection.setLastStatement(sql);
      return delegate.executeQuery(sql);
    } catch (SQLException ex) {
      pooledConnection.markWithError(ex);
      throw ex;
    }
  }

  
Execute the dml sql.
/**
   * Execute the dml sql.
   */
  @Override
  public int executeUpdate(String sql) throws SQLException {
    try {
      pooledConnection.setLastStatement(sql);
      return delegate.executeUpdate(sql);
    } catch (SQLException ex) {
      pooledConnection.markWithError(ex);
      throw ex;
    }
  }

}