groovy/modules/grails/src/persistence/org/codehaus/groovy/grails/orm/hibernate/metaclass/AbstractStaticPersistentMethod.java

/*
 * Copyright 2004-2005 the original author or authors.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */ 
package org.codehaus.groovy.grails.orm.hibernate.metaclass;

import java.util.Map;
import java.util.regex.Pattern;

import ognl.DefaultTypeConverter;
import ognl.Ognl;

import org.codehaus.groovy.grails.commons.metaclass.AbstractStaticMethodInvocation;
import org.hibernate.Criteria;
import org.hibernate.SessionFactory;
import org.hibernate.criterion.Order;
import org.springframework.orm.hibernate3.HibernateTemplate;
import org.springframework.util.Assert;

/**
 * 
 * 
 * @author Steven Devijver
 * @since Aug 8, 2005
 */
public abstract class AbstractStaticPersistentMethod extends
		AbstractStaticMethodInvocation {

	private SessionFactory sessionFactory = null;
	private ClassLoader classLoader = null;
	private DefaultTypeConverter converter = new DefaultTypeConverter();
	private Map context = Ognl.createDefaultContext(this);
	
	protected static final String ARGUMENT_MAX = "max";
	protected static final String ARGUMENT_OFFSET = "offset";
	protected static final String ARGUMENT_ORDER = "order";
	protected static final String ARGUMENT_SORT = "sort";
	protected static final String ORDER_DESC = "desc";
	protected static final String ORDER_ASC = "asc";
		
	public AbstractStaticPersistentMethod(SessionFactory sessionFactory, ClassLoader classLoader, Pattern pattern) {
		super();
		setPattern(pattern);
		this.sessionFactory = sessionFactory;
		this.classLoader = classLoader;
	}

	protected HibernateTemplate getHibernateTemplate() {
		Assert.notNull(sessionFactory, "Session factory is required!");
		return new HibernateTemplate(this.sessionFactory);
	}
	
	public Object invoke(Class clazz, String methodName, Object[] arguments) {
		ClassLoader originalClassLoader = Thread.currentThread().getContextClassLoader();
		Thread.currentThread().setContextClassLoader(this.classLoader);
		Object returnValue = doInvokeInternal(clazz, methodName, arguments);
		Thread.currentThread().setContextClassLoader(originalClassLoader);
		return returnValue;
	}

	protected abstract Object doInvokeInternal(Class clazz, String methodName, Object[] arguments);
	
	protected void populateArgumentsForCriteria(Criteria c, Map argMap) {
		
		Integer maxParam = null;
		Integer offsetParam = null;
		if(argMap.containsKey(ARGUMENT_MAX)) {			
			maxParam = (Integer)converter.convertValue(context,argMap.get(ARGUMENT_MAX),Integer.class);
		}
		if(argMap.containsKey(ARGUMENT_OFFSET)) {
			offsetParam = (Integer)converter.convertValue(context,argMap.get(ARGUMENT_OFFSET),Integer.class);
		}
		String orderParam = (String)argMap.get(ARGUMENT_ORDER);
		
		final String sort = (String)argMap.get(ARGUMENT_SORT);
		final String order = ORDER_DESC.equalsIgnoreCase(orderParam) ? ORDER_DESC : ORDER_ASC;
		final int max = maxParam == null ? -1 : maxParam.intValue();
		final int offset = offsetParam == null ? -1 : offsetParam.intValue();
		if(max > -1)
			c.setMaxResults(max);
		if(offset > -1)
			c.setFirstResult(offset);
		if(sort != null) {
			if(ORDER_DESC.equals(order)) {
				c.addOrder( Order.desc(sort) );
			}
			else {
				c.addOrder( Order.asc(sort) );
			}
		}		
	}	
}