fr.ms.log4jdbc.proxy.jdbc.operation.ResultSetOperation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of log4jdbc-driver 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.jdbc.operation;

import java.lang.reflect.Method;

import fr.ms.lang.reflect.TimeInvocation;
import fr.ms.log4jdbc.SqlOperation;
import fr.ms.log4jdbc.SqlOperationContext;
import fr.ms.log4jdbc.context.jdbc.ConnectionContextJDBC;
import fr.ms.log4jdbc.proxy.handler.Log4JdbcOperation;
import fr.ms.log4jdbc.proxy.jdbc.operation.factory.ResultSetOperationFactory;
import fr.ms.log4jdbc.sql.QueryImpl;

/**
 *
 * @see Marco4J
 *
 *
 * @author Marco Semiao
 *
 */
public class ResultSetOperation implements Log4JdbcOperation {

	private final static String ERROR = "LOG4JDBC-ERROR";

	private final TimeInvocation timeInvocation;
	private final Method method;
	private final Object[] args;

	private final ResultSetOperationFactory context;

	private final ConnectionContextJDBC connectionContext;

	private QueryImpl query;

	public ResultSetOperation(final ResultSetOperationFactory context, final ConnectionContextJDBC connectionContext,
			final TimeInvocation timeInvocation, final Method method, final Object[] args) {
		this.timeInvocation = timeInvocation;
		this.method = method;
		this.args = args;

		this.context = context;
		this.connectionContext = connectionContext;
	}

	public SqlOperation getOperation() {

		final Object invoke = timeInvocation.getInvoke();
		boolean valid = timeInvocation.getTargetException() == null;
		final String nameMethod = method.getName();

		if (nameMethod.equals("next") && invoke != null) {
			valid = valid && ((Boolean) invoke).booleanValue();
			query = context.next(valid);
		} else if (nameMethod.equals("previous") && invoke != null) {
			valid = valid && ((Boolean) invoke).booleanValue();
			query = context.previous(valid);
		} else if (nameMethod.equals("first") && invoke != null) {
			valid = valid && ((Boolean) invoke).booleanValue();
			query = context.first(valid);
		} else if (nameMethod.equals("last") && invoke != null) {
			valid = valid && ((Boolean) invoke).booleanValue();
			query = context.last(valid);
		} else if (nameMethod.equals("beforeFirst")) {
			context.beforeFirst();
		} else if (nameMethod.equals("afterLast")) {
			context.afterLast();
		} else if (nameMethod.equals("wasNull") && context.lastCell != null && invoke != null
				&& ((Boolean) invoke).booleanValue()) {
			context.wasNull();
		} else if (nameMethod.startsWith("getMetaData") && invoke != null) {
			context.getMetaData(invoke);
		} else if (nameMethod.startsWith("close")) {
			query = context.close();
		} else if (nameMethod.startsWith("get") && args != null && args.length > 0) {
			get(valid, invoke);
		}

		final SqlOperationContext sqlOperationContext = new SqlOperationContext(timeInvocation, connectionContext,
				query);
		return sqlOperationContext;
	}

	private void get(final boolean valid, Object invoke) {
		final Class clazz = method.getParameterTypes()[0];

		if (!valid) {
			invoke = ERROR;
			query = context.close();
		}
		context.addValueColumn(clazz, args, invoke);
	}

	public Object getInvoke() {
		final Object invoke = timeInvocation.getInvoke();
		return invoke;
	}
}