/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.arrow.driver.jdbc.utils;

import static com.google.common.base.Preconditions.checkNotNull;

import java.sql.Array;
import java.sql.Blob;
import java.sql.CallableStatement;
import java.sql.Clob;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.NClob;
import java.sql.PreparedStatement;
import java.sql.SQLClientInfoException;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.sql.SQLXML;
import java.sql.Savepoint;
import java.sql.Statement;
import java.sql.Struct;
import java.util.Map;
import java.util.Properties;
import java.util.concurrent.Executor;

import org.apache.arrow.driver.jdbc.ArrowFlightJdbcPooledConnection;

/**
 * Auxiliary wrapper class for {@link Connection}, used on {@link ArrowFlightJdbcPooledConnection}.
 */
public class ConnectionWrapper implements Connection {
  private final Connection realConnection;

  public ConnectionWrapper(final Connection connection) {
    realConnection = checkNotNull(connection);
  }

  @Override
  public <T> T unwrap(final Class<T> type) {
    return type.cast(realConnection);
  }

  @Override
  public boolean isWrapperFor(final Class<?> type) {
    return realConnection.getClass().isAssignableFrom(type);
  }

  @Override
  public Statement createStatement() throws SQLException {
    return realConnection.createStatement();
  }

  @Override
  public PreparedStatement prepareStatement(final String sqlQuery) throws SQLException {
    return realConnection.prepareStatement(sqlQuery);
  }

  @Override
  public CallableStatement prepareCall(final String sqlQuery) throws SQLException {
    return realConnection.prepareCall(sqlQuery);
  }

  @Override
  public String nativeSQL(final String sqlStatement) throws SQLException {
    return realConnection.nativeSQL(sqlStatement);
  }

  @Override
  public void setAutoCommit(boolean autoCommit) throws SQLException {
    realConnection.setAutoCommit(autoCommit);
  }

  @Override
  public boolean getAutoCommit() throws SQLException {
    return realConnection.getAutoCommit();
  }

  @Override
  public void commit() throws SQLException {
    realConnection.commit();
  }

  @Override
  public void rollback() throws SQLException {
    realConnection.rollback();
  }

  @Override
  public void close() throws SQLException {
    realConnection.close();
  }

  @Override
  public boolean isClosed() throws SQLException {
    return realConnection.isClosed();
  }

  @Override
  public DatabaseMetaData getMetaData() throws SQLException {
    return realConnection.getMetaData();
  }

  @Override
  public void setReadOnly(final boolean readOnly) throws SQLException {
    realConnection.setReadOnly(readOnly);
  }

  @Override
  public boolean isReadOnly() throws SQLException {
    return realConnection.isReadOnly();
  }

  @Override
  public void setCatalog(final String catalogName) throws SQLException {
    realConnection.setCatalog(catalogName);
  }

  @Override
  public String getCatalog() throws SQLException {
    return realConnection.getCatalog();
  }

  @Override
  public void setTransactionIsolation(final int transactionIsolationId) throws SQLException {
    realConnection.setTransactionIsolation(transactionIsolationId);
  }

  @Override
  public int getTransactionIsolation() throws SQLException {
    return realConnection.getTransactionIsolation();
  }

  @Override
  public SQLWarning getWarnings() throws SQLException {
    return realConnection.getWarnings();
  }

  @Override
  public void clearWarnings() throws SQLException {
    realConnection.clearWarnings();
  }

  @Override
  public Statement createStatement(final int resultSetTypeId, final int resultSetConcurrencyId)
      throws SQLException {
    return realConnection.createStatement(resultSetTypeId, resultSetConcurrencyId);
  }

  @Override
  public PreparedStatement prepareStatement(final String sqlQuery, final int resultSetTypeId,
                                            final int resultSetConcurrencyId)
      throws SQLException {
    return realConnection.prepareStatement(sqlQuery, resultSetTypeId, resultSetConcurrencyId);
  }

  @Override
  public CallableStatement prepareCall(final String query, final int resultSetTypeId,
                                       final int resultSetConcurrencyId)
      throws SQLException {
    return realConnection.prepareCall(query, resultSetTypeId, resultSetConcurrencyId);
  }

  @Override
  public Map<String, Class<?>> getTypeMap() throws SQLException {
    return realConnection.getTypeMap();
  }

  @Override
  public void setTypeMap(final Map<String, Class<?>> typeNameToClass) throws SQLException {
    realConnection.setTypeMap(typeNameToClass);
  }

  @Override
  public void setHoldability(final int holdabilityId) throws SQLException {
    realConnection.setHoldability(holdabilityId);
  }

  @Override
  public int getHoldability() throws SQLException {
    return realConnection.getHoldability();
  }

  @Override
  public Savepoint setSavepoint() throws SQLException {
    return realConnection.setSavepoint();
  }

  @Override
  public Savepoint setSavepoint(final String savepointName) throws SQLException {
    return realConnection.setSavepoint(savepointName);
  }

  @Override
  public void rollback(final Savepoint savepoint) throws SQLException {
    realConnection.rollback(savepoint);
  }

  @Override
  public void releaseSavepoint(final Savepoint savepoint) throws SQLException {
    realConnection.releaseSavepoint(savepoint);
  }

  @Override
  public Statement createStatement(final int resultSetType,
                                   final int resultSetConcurrency,
                                   final int resultSetHoldability) throws SQLException {
    return realConnection.createStatement(resultSetType, resultSetConcurrency,
        resultSetHoldability);
  }

  @Override
  public PreparedStatement prepareStatement(final String sqlQuery,
                                            final int resultSetType,
                                            final int resultSetConcurrency,
                                            final int resultSetHoldability) throws SQLException {
    return realConnection.prepareStatement(sqlQuery, resultSetType, resultSetConcurrency,
        resultSetHoldability);
  }

  @Override
  public CallableStatement prepareCall(final String sqlQuery,
                                       final int resultSetType,
                                       final int resultSetConcurrency,
                                       final int resultSetHoldability) throws SQLException {
    return realConnection.prepareCall(sqlQuery, resultSetType, resultSetConcurrency,
        resultSetHoldability);
  }

  @Override
  public PreparedStatement prepareStatement(final String sqlQuery, final int autoGeneratedKeysId)
      throws SQLException {
    return realConnection.prepareStatement(sqlQuery, autoGeneratedKeysId);
  }

  @Override
  public PreparedStatement prepareStatement(final String sqlQuery, final int[] columnIndices)
      throws SQLException {
    return realConnection.prepareStatement(sqlQuery, columnIndices);
  }

  @Override
  public PreparedStatement prepareStatement(final String sqlQuery, final String[] columnNames)
      throws SQLException {
    return realConnection.prepareStatement(sqlQuery, columnNames);
  }

  @Override
  public Clob createClob() throws SQLException {
    return realConnection.createClob();
  }

  @Override
  public Blob createBlob() throws SQLException {
    return realConnection.createBlob();
  }

  @Override
  public NClob createNClob() throws SQLException {
    return realConnection.createNClob();
  }

  @Override
  public SQLXML createSQLXML() throws SQLException {
    return realConnection.createSQLXML();
  }

  @Override
  public boolean isValid(final int timeout) throws SQLException {
    return realConnection.isValid(timeout);
  }

  @Override
  public void setClientInfo(final String propertyName, final String propertyValue)
      throws SQLClientInfoException {
    realConnection.setClientInfo(propertyName, propertyValue);
  }

  @Override
  public void setClientInfo(final Properties properties) throws SQLClientInfoException {
    realConnection.setClientInfo(properties);
  }

  @Override
  public String getClientInfo(final String propertyName) throws SQLException {
    return realConnection.getClientInfo(propertyName);
  }

  @Override
  public Properties getClientInfo() throws SQLException {
    return realConnection.getClientInfo();
  }

  @Override
  public Array createArrayOf(final String typeName, final Object[] elements) throws SQLException {
    return realConnection.createArrayOf(typeName, elements);
  }

  @Override
  public Struct createStruct(final String typeName, final Object[] attributes)
      throws SQLException {
    return realConnection.createStruct(typeName, attributes);
  }

  @Override
  public void setSchema(final String schemaName) throws SQLException {
    realConnection.setSchema(schemaName);
  }

  @Override
  public String getSchema() throws SQLException {
    return realConnection.getSchema();
  }

  @Override
  public void abort(final Executor executor) throws SQLException {
    realConnection.abort(executor);
  }

  @Override
  public void setNetworkTimeout(final Executor executor, final int timeoutInMillis)
      throws SQLException {
    realConnection.setNetworkTimeout(executor, timeoutInMillis);
  }

  @Override
  public int getNetworkTimeout() throws SQLException {
    return realConnection.getNetworkTimeout();
  }
}
