com.avaje.ebeaninternal.server.persist.BatchedBeanControl Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) 2006  Robin Bygrave
 * 
 * This file is part of Ebean.
 * 
 * Ebean is free software; you can redistribute it and/or modify it 
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *  
 * Ebean 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 Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with Ebean; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA  
 */
package com.avaje.ebeaninternal.server.persist;

import java.util.ArrayList;
import java.util.HashMap;

import com.avaje.ebeaninternal.api.SpiTransaction;
import com.avaje.ebeaninternal.server.core.PersistRequest;
import com.avaje.ebeaninternal.server.core.PersistRequestBean;
import com.avaje.ebeaninternal.server.deploy.BeanDescriptor;

/**
 * Holds all the batched beans.
 * 
 * The beans are held here which delays the binding to a PreparedStatement. This
 * 'delayed' binding is required as the beans need to be bound and executed in
 * the correct order (according to the depth).
 * 
 */
public class BatchedBeanControl {

	/**
	 * Map of the BatchedBeanHolder objects. They each have a depth and are later
	 * sorted by their depth to get the execution order.
	 */
	private final HashMap beanHoldMap = new HashMap();
	
	private final SpiTransaction transaction;

	private final BatchControl batchControl;

	private int topOrder;

	public BatchedBeanControl(SpiTransaction t, BatchControl batchControl) {
		this.transaction = t;
		this.batchControl = batchControl;
	}

	public ArrayList getPersistList(PersistRequestBean request) {
        BatchedBeanHolder beanHolder = getBeanHolder(request);
        return beanHolder.getList(request);
    }
	
	/**
	 * Return an entry for the given type description. The type description is
	 * typically the bean class name (or table name for MapBeans).
	 */
	private BatchedBeanHolder getBeanHolder(PersistRequestBean request) {
		
		BeanDescriptor beanDescriptor = request.getBeanDescriptor();
		BatchedBeanHolder batchBeanHolder = beanHoldMap.get(beanDescriptor.getFullName());
		if (batchBeanHolder == null) {
			int relativeDepth = transaction.depth(0);
			
			if (relativeDepth == 0){
				topOrder++;
			}
			int stmtOrder = topOrder*100 + relativeDepth;
			
			batchBeanHolder = new BatchedBeanHolder(batchControl, beanDescriptor, stmtOrder);
			beanHoldMap.put(beanDescriptor.getFullName(), batchBeanHolder);
		}
		return batchBeanHolder;
	}

	/**
	 * Return true if this holds no persist requests.
	 */
	public boolean isEmpty() {
		return beanHoldMap.isEmpty();
	}

	/**
	 * Return the held beans ready for sorting and executing.
	 * 
	 * This also has the effect of clearing the cache of beans held.
	 * 
	 */
	public BatchedBeanHolder[] getArray() {
		BatchedBeanHolder[] bsArray = new BatchedBeanHolder[beanHoldMap.size()];
		beanHoldMap.values().toArray(bsArray);
		beanHoldMap.clear();
		topOrder = 0;
		return bsArray;
	}

}