fr.ms.log4jdbc.messagefactory.StatementHandler 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
There is a newer version: 0.0.5
/*
 * 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.messagefactory;

import java.lang.reflect.Method;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import fr.ms.log4jdbc.context.JdbcContext;
import fr.ms.log4jdbc.invocationhandler.MessageFactory;
import fr.ms.log4jdbc.invocationhandler.MessageInvocationHandler.MessageInvocationContext;
import fr.ms.log4jdbc.invocationhandler.TimeInvocation;
import fr.ms.log4jdbc.message.MessageHandlerImpl;
import fr.ms.log4jdbc.proxy.Handlers;
import fr.ms.log4jdbc.sql.Query;
import fr.ms.log4jdbc.sql.QuerySQLFactory;
import fr.ms.log4jdbc.sql.ResultSetCollectorQuery;
import fr.ms.log4jdbc.sql.impl.EmptyQuery;
import fr.ms.log4jdbc.sql.impl.WrapperQuery;

/**
 * 
 * @see Marco4J
 * 
 * 
 * @author Marco Semiao
 * 
 */
public class StatementHandler implements MessageFactory {

  private final Statement statement;

  protected WrapperQuery query;

  protected final QuerySQLFactory querySQLFactory;

  public StatementHandler(final Statement statement, final QuerySQLFactory querySQLFactory) {
    this.statement = statement;
    this.querySQLFactory = querySQLFactory;
  }

  public MessageHandlerImpl transformMessage(final Object proxy, final Method method, final Object[] args,
      final MessageInvocationContext mic, final MessageHandlerImpl message) {
    final TimeInvocation timeInvocation = mic.getInvokeTime();
    final JdbcContext jdbcContext = mic.getJdbcContext();

    final String nameMethod = method.getName();

    final boolean addBatchMethod = nameMethod.equals("addBatch") && args != null && args.length >= 1;
    if (addBatchMethod) {
      final String sql = (String) args[0];

      final WrapperQuery query = querySQLFactory.newQuerySQL(jdbcContext, sql);
      query.setMethodQuery(Query.METHOD_BATCH);
      query.setTimeInvocation(timeInvocation);

      jdbcContext.addQuery(query, true);

      mic.setQuery(query);
      message.setQuery(query);

      return message;
    }

    final boolean executeBatchMethod = nameMethod.equals("executeBatch") && args == null;
    if (executeBatchMethod) {
      final Object invoke = timeInvocation.getInvoke();
      int[] updateCounts = null;

      final Class returnType = method.getReturnType();
      if (invoke != null) {
        if (int[].class.equals(returnType)) {
          updateCounts = (int[]) invoke;
        }
      }

      jdbcContext.getBatchContext().executeBatch(updateCounts);
      jdbcContext.resetBatch();
      return message;
    }

    final boolean executeMethod = nameMethod.startsWith("execute") && args != null && args.length >= 1;
    if (executeMethod) {
      final String sql = (String) args[0];
      final WrapperQuery query = querySQLFactory.newQuerySQL(jdbcContext, sql);
      query.setMethodQuery(Query.METHOD_EXECUTE);
      query.setTimeInvocation(timeInvocation);
      final Integer updateCount = getUpdateCount(timeInvocation, method);
      query.setUpdateCount(updateCount);

      jdbcContext.addQuery(query, false);

      mic.setQuery(query);
      message.setQuery(query);

      // execute retourne true boolean - GetResultSet
      final Class returnType = method.getReturnType();
      if (Boolean.class.equals(returnType) || Boolean.TYPE.equals(returnType)) {
        final Boolean invokeBoolean = (Boolean) timeInvocation.getInvoke();
        if (invokeBoolean.booleanValue()) {
          this.query = query;
        }
      }

      return message;
    }

    mic.setQuery(query);

    return message;
  }

  public Object wrap(final Object invoke, final Object[] args, final MessageInvocationContext mic) {
    if (invoke != null) {
      if (invoke instanceof ResultSet) {
        final JdbcContext jdbcContext = mic.getJdbcContext();

        final ResultSet resultSet = (ResultSet) invoke;

        ResultSetCollectorQuery rscQuery = mic.getQuery();
        if (rscQuery == null) {
          rscQuery = new EmptyQuery();
        }

        return Handlers.getResultSet(resultSet, jdbcContext, rscQuery);
      }
    }
    return invoke;
  }

  protected Integer getUpdateCount(final TimeInvocation timeInvocation, final Method method) {
    Integer updateCount = null;
    final Object invoke = timeInvocation.getInvoke();
    final Class returnType = method.getReturnType();
    if (Integer.class.equals(returnType) || Integer.TYPE.equals(returnType)) {
      updateCount = (Integer) invoke;
    } else if (Boolean.class.equals(returnType) || Boolean.TYPE.equals(returnType)) {
      final Boolean invokeBoolean = (Boolean) invoke;
      if (timeInvocation.getTargetException() == null && !invokeBoolean.booleanValue()) {
        try {
          updateCount = new Integer(statement.getUpdateCount());
        } catch (final SQLException e) {
          return null;
        }
      }
    }
    return updateCount;
  }
}