fr.ms.log4jdbc.context.internal.BatchContext 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.context.internal;

import java.util.ArrayList;
import java.util.List;

import fr.ms.lang.delegate.DefaultSyncLongFactory;
import fr.ms.lang.delegate.SyncLong;
import fr.ms.lang.delegate.SyncLongFactory;
import fr.ms.lang.ref.ReferenceFactory;
import fr.ms.lang.ref.ReferenceObject;
import fr.ms.log4jdbc.sql.Query;
import fr.ms.log4jdbc.sql.QueryImpl;

/**
 *
 * @see Marco4J
 *
 *
 * @author Marco Semiao
 *
 */
public class BatchContext implements Cloneable {

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

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

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

    private String state;

    private boolean batchInit = false;

    private long batchNumber;

    private final TransactionContext transactionContext;

    private final static String REF_MESSAGE_FULL = "LOG4JDBC : Memory Full, clean Queries Batch";
    private ReferenceObject refQueriesBatch = ReferenceFactory.newReference(REF_MESSAGE_FULL, new ArrayList());

    public BatchContext(final TransactionContext transactionContext) {
	this.transactionContext = transactionContext;
    }

    public void addQuery(final QueryImpl query) {
	transactionContext.addQuery(query, true);
	query.setState(Query.STATE_NOT_EXECUTE);

	final List queriesBatch = (List) refQueriesBatch.get();
	if (queriesBatch != null) {
	    queriesBatch.add(query);
	}

	query.setBatchContext(this);

	if (!batchInit) {
	    batchInit = true;
	    batchNumber = totalBatchNumber.incrementAndGet();
	    openBatch.incrementAndGet();
	}

	state = Query.STATE_NOT_EXECUTE;
    }

    public void executeBatch(final int[] updateCounts) {
	final List queriesBatch = (List) refQueriesBatch.get();
	if (queriesBatch == null) {
	    return;
	}

	final int sizeQueries = queriesBatch.size();

	if (sizeQueries > 0) {
	    boolean countsOk = true;
	    if (updateCounts == null || sizeQueries != updateCounts.length) {
		countsOk = false;
	    } else {
		for (int i = 0; i < updateCounts.length; i++) {
		    if (updateCounts[i] < 0) {
			countsOk = false;
			break;
		    }
		}
	    }
	    for (int i = 0; i < sizeQueries; i++) {
		final QueryImpl q = (QueryImpl) queriesBatch.get(i);
		q.setState(Query.STATE_EXECUTE);
		if (countsOk) {
		    q.setUpdateCount(new Integer(updateCounts[i]));
		}
	    }
	}

	state = Query.STATE_EXECUTE;
    }

    public Query[] getQueriesBatch() {
	final List queriesBatch = (List) refQueriesBatch.get();
	if (queriesBatch == null) {
	    return null;
	}
	return (Query[]) queriesBatch.toArray(new Query[queriesBatch.size()]);
    }

    public long getOpenBatch() {
	return openBatch.get();
    }

    public void decrement() {
	if (batchInit) {
	    openBatch.decrementAndGet();
	}
    }

    public long getBatchNumber() {
	return batchNumber;
    }

    public String getState() {
	return state;
    }

    public Object clone() throws CloneNotSupportedException {
	final BatchContext b = (BatchContext) super.clone();
	final List queriesBatch = (List) refQueriesBatch.get();
	if (queriesBatch == null) {
	    b.refQueriesBatch = ReferenceFactory.newReference(REF_MESSAGE_FULL, new ArrayList());
	} else {
	    b.refQueriesBatch = ReferenceFactory.newReference(REF_MESSAGE_FULL, new ArrayList(queriesBatch));
	}
	return b;
    }
}