/* 
This file is part of the PolePosition database benchmark
http://www.polepos.org

This program 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 2
of the License, or (at your option) any later version.

This program 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 this program; if not, write to the Free
Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
MA  02111-1307, USA. */

package org.zoodb.test.jdo.pole;

import java.util.Collection;
import java.util.Iterator;

import javax.jdo.Extent;
import javax.jdo.JDOFatalInternalException;
import javax.jdo.PersistenceManager;
import javax.jdo.Query;
import javax.jdo.Transaction;

import org.zoodb.test.testutil.TestTools;


public abstract class JdoDriver extends DriverBase {
    
	private transient PersistenceManager pm;
    
	public void prepare(PersistenceManager pm) {
		this.pm = pm;
	}
	
	public void closeDatabase() {
        Transaction tx = db().currentTransaction();
        if (tx.isActive()) {
            tx.rollback();
        }
		pm.close();
		pm = null;
		TestTools.closePM();
	}
	
	protected PersistenceManager db() {
		return pm;
	}
	
    public void begin(){
		Transaction currentTransaction = db().currentTransaction();
		if(! currentTransaction.isActive()) {
			currentTransaction.begin();
		}
    }
    
    public void commit() {
        db().currentTransaction().commit();
    }
    
    public void store(Object obj) {
        db().makePersistent(obj);
    }
    
    protected void doQuery( Query q, Object param){
        Collection<?> result = (Collection<?>)q.execute(param);
        Iterator<?> it = result.iterator();
        while (it.hasNext()){
            Object o = it.next();
            if (o instanceof CheckSummable) {
            	try {
            		addToCheckSum(((CheckSummable)o).checkSum());
            	} catch (JDOFatalInternalException e){
            		Throwable[] nestedExceptions = e.getNestedExceptions();
            		if (nestedExceptions != null) {
            			for (int i = 0; i < nestedExceptions.length; i++) {
            				nestedExceptions[i].printStackTrace();
						}
            		}
            		
            	}
            }
        }
    }
    
    protected void readExtent(Class<?> clazz) {
        Extent<?> extent = db().getExtent( clazz, false );
        int count = 0;
        Iterator<?> itr = extent.iterator();
        while (itr.hasNext()){
            Object o = itr.next();
            count++;
            if (o instanceof CheckSummable) {
                addToCheckSum(((CheckSummable)o).checkSum());  
            }
        }
        extent.closeAll();
		
		//ensure that n is not optimized away
		if (count == 0) {
			throw new IllegalStateException();
		}
    }
    
	protected void delete(Object obj) {
		db().deletePersistent(obj);
	}
    
}