package org.javasimon.jdbc4;

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.javasimon.Counter;
import org.javasimon.SimonManager;
import org.javasimon.Split;

/**
 * Class implements Simon JDBC proxy connection.
 * <p>
 * Every method of this connection is implemented as call of real connection method.
 * Several methods have added work with Simons (starting, stopping, etc.) for monitoring
 * purposes.
 * </p>
 * <p>
 * From all statement-return-methods (<code>createStatement(*)</code>,
 * <code>prepareStatement(*)</code>, <code>prepareCall(*)</code>) connection returns own
 * implementation of statement classes. Those classes are also proxies and provides
 * additional Simons for monitoring features of JDBC driver.
 * </p>
 * Monitoring connection ensure following Simons:
 * <ul>
 * <li>lifespan (<code>org.javasimon.jdbc4.conn</code>, stopwatch) - measure connection life and count</li>
 * <li>commits (<code>org.javasimon.jdbc4.conn.commits</code>, counter) - measure executed commits of all connections</li>
 * <li>rollbacks (<code>org.javasimon.jdbc4.conn.rollbacks</code>, counter) - measure executed rollbacks of all connections</li>
 * </ul>
 *
 * @author Radovan Sninsky
 * @see java.sql.Connection
 * @since 2.4
 */
public final class SimonConnection implements Connection {
        private final Connection conn;
        private final WrapperSupport<Connection> wrapperSupport;
        private final SqlNormalizerFactory sqlNormalizerFactory;
        private String prefix;

        private Split life;
        private Counter commits;
        private Counter rollbacks;

        /**
         * Class constructor, initializes Simons (lifespan, active, commits
         * and rollbacks) related to the DB connection.
         *
         * @param conn real DB connection
         * @param prefix hierarchy prefix for connection Simons
         * @param sqlNormalizerFactory factory to map queries to Simon keys
         */
        public SimonConnection(Connection conn, String prefix, SqlNormalizerFactory sqlNormalizerFactory) {
                this.conn = conn;
                this.sqlNormalizerFactory = sqlNormalizerFactory;
                this.wrapperSupport = new WrapperSupport<>(this.conn, Connection.class);
                this.prefix = prefix;

                commits = SimonManager.getCounter(prefix + ".conn.commits");
                rollbacks = SimonManager.getCounter(prefix + ".conn.rollbacks");
                life = SimonManager.getStopwatch(prefix + ".conn").start();
        }

        public SimonConnection(Connection conn, String prefix) {
                this(conn, prefix, new RegexBasedNormalizerFactory());
        }

        /**
         * Closes the real connection, stops lifespan Simon and decreases the active Simon.
         *
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public void close() throws SQLException {
                conn.close();

                life.stop();
        }

        /**
         * Commits the real connection and increases the commits Simon.
         *
         * @throws java.sql.SQLException if real commit fails
         */
        @Override
        public void commit() throws SQLException {
                conn.commit();

                commits.increase();
        }

        /**
         * Rollback the real connection and increase the rollbacks Simon.
         *
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public void rollback() throws SQLException {
                conn.rollback();

                rollbacks.increase();
        }

        /**
         * Rollbacks the real connection and increases the rollbacks Simon.
         *
         * @param savepoint the <code>Savepoint</code> object to roll back to
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public void rollback(Savepoint savepoint) throws SQLException {
                conn.rollback(savepoint);

                rollbacks.increase();
        }

        /**
         * Calls the real createStatement and wraps the returned statement by Simon's statement.
         *
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public Statement createStatement() throws SQLException {
                return new SimonStatement(this, conn.createStatement(), prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real createStatement and wraps returned statement by Simon's statement.
         *
         * @param rsType result set type
         * @param rsConcurrency result set concurrency
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public Statement createStatement(int rsType, int rsConcurrency) throws SQLException {
                return new SimonStatement(this, conn.createStatement(rsType, rsConcurrency), prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real createStatement and wraps returned statement by Simon's statement.
         *
         * @param rsType result set type
         * @param rsConcurrency result set concurrency
         * @param rsHoldability result set holdability
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public Statement createStatement(int rsType, int rsConcurrency, int rsHoldability) throws SQLException {
                return new SimonStatement(this, conn.createStatement(rsType, rsConcurrency, rsHoldability), prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareStatement and wraps returned statement by Simon's statement.
         *
         * @param sql SQL statement
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public PreparedStatement prepareStatement(String sql) throws SQLException {
                return new SimonPreparedStatement(this, conn.prepareStatement(sql), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareStatement and wraps returned statement by Simon's statement.
         *
         * @param sql SQL statement
         * @param autoGeneratedKeys auto generated keys
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public PreparedStatement prepareStatement(String sql, int autoGeneratedKeys) throws SQLException {
                return new SimonPreparedStatement(this, conn.prepareStatement(sql, autoGeneratedKeys), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareStatement and wraps returned statement by Simon's statement.
         *
         * @param sql SQL statement
         * @param rsType result set type
         * @param rsConcurrency result set concurrency
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public PreparedStatement prepareStatement(String sql, int rsType, int rsConcurrency) throws SQLException {
                return new SimonPreparedStatement(this, conn.prepareStatement(sql, rsType, rsConcurrency), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareStatement and wraps returned statement by Simon's statement.
         *
         * @param sql SQL statement
         * @param rsType result set type
         * @param rsConcurrency result set concurrency
         * @param rsHoldability result set holdability
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public PreparedStatement prepareStatement(String sql, int rsType, int rsConcurrency, int rsHoldability) throws SQLException {
                return new SimonPreparedStatement(this, conn.prepareStatement(sql, rsType, rsConcurrency, rsHoldability), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareStatement and wraps returned statement by Simon's statement.
         *
         * @param sql SQL statement
         * @param columnIndexes an array of column indexes indicating the columns
         * that should be returned from the inserted row or rows
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public PreparedStatement prepareStatement(String sql, int[] columnIndexes) throws SQLException {
                return new SimonPreparedStatement(this, conn.prepareStatement(sql, columnIndexes), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareStatement and wraps returned statement by Simon's statement.
         *
         * @param sql SQL statement
         * @param columnNames an array of column names indicating the columns
         * that should be returned from the inserted row or rows
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public PreparedStatement prepareStatement(String sql, String[] columnNames) throws SQLException {
                return new SimonPreparedStatement(this, conn.prepareStatement(sql, columnNames), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareCall and wraps returned statement by Simon's statement.
         *
         * @param sql an SQL statement, typically a JDBC function call escape string
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public CallableStatement prepareCall(String sql) throws SQLException {
                return new SimonCallableStatement(conn, conn.prepareCall(sql), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareCall and wraps returned statement by Simon's statement.
         *
         * @param sql an SQL statement, typically a JDBC function call escape string
         * @param rsType result set type
         * @param rsConcurrency result set concurrency
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public CallableStatement prepareCall(String sql, int rsType, int rsConcurrency) throws SQLException {
                return new SimonCallableStatement(conn, conn.prepareCall(sql, rsType, rsConcurrency), sql, prefix, sqlNormalizerFactory);
        }

        /**
         * Calls real prepareCall and wraps returned statement by Simon's statement.
         *
         * @param sql an SQL statement, typically a JDBC function call escape string
         * @param rsType result set type
         * @param rsConcurrency result set concurrency
         * @param rsHoldability result set holdability
         * @return Simon's statement with wrapped real statement
         * @throws java.sql.SQLException if real operation fails
         */
        @Override
        public CallableStatement prepareCall(String sql, int rsType, int rsConcurrency, int rsHoldability) throws SQLException {
                return new SimonCallableStatement(conn, conn.prepareCall(sql, rsType, rsConcurrency, rsHoldability), sql, prefix, sqlNormalizerFactory);
        }

        //// NOT MONITORED

        @Override
        public String nativeSQL(String s) throws SQLException {
                return conn.nativeSQL(s);
        }

        @Override
        public void setAutoCommit(boolean b) throws SQLException {
                conn.setAutoCommit(b);
        }

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

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

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

        @Override
        public void setReadOnly(boolean b) throws SQLException {
                conn.setReadOnly(b);
        }

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

        @Override
        public void setCatalog(String s) throws SQLException {
                conn.setCatalog(s);
        }

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

        @Override
        public void setTransactionIsolation(int i) throws SQLException {
                conn.setTransactionIsolation(i);
        }

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

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

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

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

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

        @Override
        public void setHoldability(int i) throws SQLException {
                conn.setHoldability(i);
        }

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

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

        @Override
        public Savepoint setSavepoint(String s) throws SQLException {
                return conn.setSavepoint(s);
        }

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

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

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

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

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

        @Override
        public boolean isValid(int i) throws SQLException {
                return conn.isValid(i);
        }

        @Override
        public void setClientInfo(String s, String s1) throws SQLClientInfoException {
                conn.setClientInfo(s, s1);
        }

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

        @Override
        public String getClientInfo(String s) throws SQLException {
                return conn.getClientInfo(s);
        }

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

        @Override
        public Array createArrayOf(String s, Object[] objects) throws SQLException {
                return conn.createArrayOf(s, objects);
        }

        @Override
        public Struct createStruct(String s, Object[] objects) throws SQLException {
                return conn.createStruct(s, objects);
        }

    @Override
    public void setSchema(String schema) throws SQLException {
        conn.setSchema(schema);
    }

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

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

    @Override
    public void setNetworkTimeout(Executor executor, int milliseconds) throws SQLException {
        conn.setNetworkTimeout(executor, milliseconds);
    }

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

    @Override
        public <T> T unwrap(Class<T> iface) throws SQLException {
                return wrapperSupport.unwrap(iface);
        }

        @Override
        public boolean isWrapperFor(Class<?> iface) throws SQLException {
                return wrapperSupport.isWrapperFor(iface);
        }
}