fr.ms.log4jdbc.proxy.Log4JdbcProxy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of log4jdbc-file Show documentation
The newest version!
/*
 * This file is part of Log4Jdbc.
 *
 * Log4Jdbc is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Log4Jdbc is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Log4Jdbc.  If not, see .
 *
 */
package fr.ms.log4jdbc.proxy;

import java.lang.reflect.InvocationHandler;
import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.Statement;

import fr.ms.lang.reflect.ProxyOperationFactory;
import fr.ms.lang.reflect.ProxyUtils;
import fr.ms.lang.reflect.TraceTimeInvocationHandler;
import fr.ms.log4jdbc.SqlOperationLogger;
import fr.ms.log4jdbc.context.Log4JdbcContext;
import fr.ms.log4jdbc.context.jdbc.ConnectionContextJDBC;
import fr.ms.log4jdbc.proxy.handler.Log4JdbcInvocationHandler;
import fr.ms.log4jdbc.proxy.handler.TraceTimeInvocationOperationFactory;
import fr.ms.log4jdbc.proxy.jdbc.operation.factory.ConnectionOperationFactory;
import fr.ms.log4jdbc.proxy.jdbc.operation.factory.PreparedStatementOperationFactory;
import fr.ms.log4jdbc.proxy.jdbc.operation.factory.ResultSetOperationFactory;
import fr.ms.log4jdbc.proxy.jdbc.operation.factory.StatementOperationFactory;
import fr.ms.log4jdbc.sql.QueryImpl;
import fr.ms.log4jdbc.sql.internal.QueryFactory;
import fr.ms.log4jdbc.sql.internal.QueryNamedFactory;
import fr.ms.log4jdbc.sql.internal.QuerySQLFactory;
import fr.ms.log4jdbc.utils.ServicesJDBC;

/**
 *
 * @see Marco4J
 *
 *
 * @author Marco Semiao
 *
 */
public final class Log4JdbcProxy {

	public static Connection proxyConnection(final Connection connection, final Log4JdbcContext log4JdbcContext,
			final Driver driver, final String url) {
		final ConnectionContextJDBC connectionContext = log4JdbcContext.newConnectionContext(connection, driver, url);
		final Connection wrap = proxyConnection(connection, connectionContext);

		return wrap;
	}

	public static Connection proxyConnection(final Connection connection, final Log4JdbcContext log4JdbcContext,
			final Class clazz) {
		final ConnectionContextJDBC connectionContext = log4JdbcContext.newConnectionContext(connection, clazz);
		final Connection wrap = proxyConnection(connection, connectionContext);

		return wrap;
	}

	private static Connection proxyConnection(final Connection connection,
			final ConnectionContextJDBC connectionContext) {
		final SqlOperationLogger[] logs = ServicesJDBC.getMessageLogger(SqlOperationLogger.CONNECTION);

		final ProxyOperationFactory factory = new ConnectionOperationFactory(connectionContext, connection);

		final InvocationHandler handler = createHandler(connection, logs, factory);

		final Connection instance = (Connection) ProxyUtils.newProxyInstance(connection, handler);

		return instance;
	}

	public static Statement proxyStatement(final Statement statement, final ConnectionContextJDBC connectionContext) {
		final SqlOperationLogger[] logs = ServicesJDBC.getMessageLogger(SqlOperationLogger.STATEMENT);

		final QueryFactory queryFactory = QuerySQLFactory.getInstance();
		final ProxyOperationFactory factory = new StatementOperationFactory(connectionContext, statement, queryFactory);

		final InvocationHandler handler = createHandler(statement, logs, factory);

		final Statement instance = (Statement) ProxyUtils.newProxyInstance(statement, handler);

		return instance;
	}

	public static PreparedStatement proxyPreparedStatement(final PreparedStatement statement,
			final ConnectionContextJDBC connectionContext, final QueryImpl query) {
		final SqlOperationLogger[] logs = ServicesJDBC.getMessageLogger(SqlOperationLogger.PREPARED_STATEMENT);

		final QueryFactory queryFactory = QuerySQLFactory.getInstance();
		final ProxyOperationFactory factory = new PreparedStatementOperationFactory(connectionContext, statement,
				queryFactory, query);

		final InvocationHandler handler = createHandler(statement, logs, factory);

		final PreparedStatement instance = (PreparedStatement) ProxyUtils.newProxyInstance(statement, handler);

		return instance;
	}

	public static CallableStatement proxyCallableStatement(final CallableStatement statement,
			final ConnectionContextJDBC connectionContext, final QueryImpl query) {
		final SqlOperationLogger[] logs = ServicesJDBC.getMessageLogger(SqlOperationLogger.CALLABLE_STATEMENT);

		final QueryFactory queryFactory = QueryNamedFactory.getInstance();
		final ProxyOperationFactory factory = new PreparedStatementOperationFactory(connectionContext, statement,
				queryFactory, query);

		final InvocationHandler handler = createHandler(statement, logs, factory);

		final CallableStatement instance = (CallableStatement) ProxyUtils.newProxyInstance(statement, handler);

		return instance;
	}

	public static ResultSet proxyResultSet(final ResultSet resultSet, final ConnectionContextJDBC connectionContext,
			final QueryImpl query) {
		final SqlOperationLogger[] logs = ServicesJDBC.getMessageLogger(SqlOperationLogger.RESULT_SET);

		final ProxyOperationFactory factory = new ResultSetOperationFactory(connectionContext, resultSet, query);

		final InvocationHandler handler = createHandler(resultSet, logs, factory);

		final ResultSet instance = (ResultSet) ProxyUtils.newProxyInstance(resultSet, handler);

		return instance;
	}

	static final InvocationHandler createHandler(final Object implementation, final SqlOperationLogger[] logs,
			ProxyOperationFactory factory) {
		if (TraceTimeInvocationHandler.LOG.isDebugEnabled()) {
			factory = new TraceTimeInvocationOperationFactory(factory);
			final InvocationHandler handler = new Log4JdbcInvocationHandler(implementation, logs, factory);
			final InvocationHandler decorator = new TraceTimeInvocationHandler(handler);
			return decorator;
		} else {
			final InvocationHandler handler = new Log4JdbcInvocationHandler(implementation, logs, factory);
			return handler;
		}
	}
}