fr.ms.log4jdbc.sql.QueryImpl Maven / Gradle / Ivy

Go to download
/*
 * 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.sql;

import java.sql.ResultSet;
import java.util.Date;
import java.util.Map;

import fr.ms.lang.delegate.DefaultSyncLongFactory;
import fr.ms.lang.delegate.SyncLong;
import fr.ms.lang.delegate.SyncLongFactory;
import fr.ms.lang.reflect.TimeInvocation;
import fr.ms.log4jdbc.context.Batch;
import fr.ms.log4jdbc.context.BatchImpl;
import fr.ms.log4jdbc.context.Transaction;
import fr.ms.log4jdbc.context.TransactionImpl;
import fr.ms.log4jdbc.context.internal.BatchContext;
import fr.ms.log4jdbc.context.internal.ConnectionContext;
import fr.ms.log4jdbc.context.internal.TransactionContext;
import fr.ms.log4jdbc.resultset.ResultSetCollector;
import fr.ms.log4jdbc.resultset.ResultSetCollectorImpl;

/**
 *
 * @see Marco4J
 *
 *
 * @author Marco Semiao
 *
 */
public class QueryImpl implements Query {

    private final static SyncLongFactory syncLongFactory = DefaultSyncLongFactory.getInstance();

    private final static SyncLong nbQueryTotal = syncLongFactory.newLong();

    private long queryNumber;
    private TimeInvocation timeInvocation;

    private String methodQuery;

    private final QuerySQL query;

    private Integer updateCount;

    private ResultSetCollectorImpl resultSetCollector;

    private String state = Query.STATE_COMMIT;

    private Batch batch;

    private Transaction transaction;

    private Object savePoint;

    QueryImpl(final QuerySQL query) {
	this.query = query;
    }

    public void execute() {
	this.queryNumber = nbQueryTotal.incrementAndGet();
    }

    public Date getDate() {
	if (timeInvocation == null) {
	    return null;
	}
	return timeInvocation.getStartDate();
    }

    public long getExecTime() {
	if (timeInvocation == null) {
	    return -1;
	}
	return timeInvocation.getExecTime();
    }

    public long getQueryNumber() {
	return queryNumber;
    }

    public String getMethodQuery() {
	return methodQuery;
    }

    public String getJDBCQuery() {
	if (query == null) {
	    return null;
	}
	return query.getJDBCQuery();
    }

    public Map getJDBCParameters() {
	if (query == null) {
	    return null;
	}
	return query.getJDBCParameters();
    }

    public String getTypeQuery() {
	if (query == null) {
	    return null;
	}
	return query.getTypeQuery();
    }

    public String getSQLQuery() {
	if (query == null) {
	    return null;
	}
	return query.getSQLQuery();
    }

    public Integer getUpdateCount() {
	return updateCount;
    }

    public ResultSetCollector getResultSetCollector() {
	if (resultSetCollector == null || resultSetCollector.isMetaDataError()) {
	    return null;
	}
	return resultSetCollector;
    }

    public String getState() {
	return state;
    }

    public Transaction getTransaction() {
	return transaction;
    }

    public Batch getBatch() {
	return batch;
    }

    public Object putParams(final Object key, final Object value) {
	if (query == null) {
	    return null;
	}
	return query.putParams(key, value);
    }

    public void setTimeInvocation(final TimeInvocation timeInvocation) {
	this.timeInvocation = timeInvocation;
    }

    public void setUpdateCount(final Integer updateCount) {
	if (updateCount != null && updateCount.intValue() >= 0) {
	    this.updateCount = updateCount;
	}
    }

    public void initResultSetCollector(final ConnectionContext connectionContext) {
	if (this.resultSetCollector == null) {
	    this.resultSetCollector = new ResultSetCollectorImpl(connectionContext);
	}
    }

    public void initResultSetCollector(final ConnectionContext connectionContext, final ResultSet rs) {
	initResultSetCollector(connectionContext);
	this.resultSetCollector.setRs(rs);
    }

    public void setMethodQuery(final String methodQuery) {
	this.methodQuery = methodQuery;
    }

    public void setState(final String state) {
	this.state = state;
    }

    public void setBatchContext(final BatchContext batchContext) {
	this.batch = new BatchImpl(batchContext);
    }

    public void setTransactionContext(final TransactionContext transactionContext) {
	this.transaction = new TransactionImpl(transactionContext);
    }

    public void setSavePoint(final Object savePoint) {
	this.savePoint = savePoint;
    }

    public Object getSavePoint() {
	return savePoint;
    }

    public String toString() {
	return "QueryImpl [queryNumber=" + queryNumber + ", state=" + state + ", query=" + query + "]";
    }
}