Override default vraptor HibernateTransactionInterceptor

HibernateTransactionInterceptor

import br.com.caelum.vraptor.Intercepts;
import br.com.caelum.vraptor.Validator;
import br.com.caelum.vraptor.core.InterceptorStack;
import br.com.caelum.vraptor.http.MutableResponse;
import br.com.caelum.vraptor.interceptor.Interceptor;
import br.com.caelum.vraptor.resource.ResourceMethod;
import org.hibernate.Session;
import org.hibernate.Transaction;

@Intercepts
public class HibernateTransactionInterceptor implements Interceptor {
	
	private final Session session;
	private final Validator validator;
	private final MutableResponse response;

	public HibernateTransactionInterceptor(Session session, Validator validator, MutableResponse response) {
		this.session = session;
		this.validator = validator;
		this.response = response;
	}
	
	@Override
	public void intercept(InterceptorStack stack, ResourceMethod method, Object instance) {
		addRedirectListener();

		Transaction transaction = null;

		try {
			transaction = session.beginTransaction();
			stack.next(method, instance);
			if (!validator.hasErrors() && !transaction.wasCommitted()) {
				transaction.commit();
			}
		} finally {
			if (transaction != null && transaction.isActive()) {
				transaction.rollback();
			}
		}
	}

	private void addRedirectListener() {
		response.addRedirectListener(new MutableResponse.RedirectListener() {
			@Override
			public void beforeRedirect() {
				if (!validator.hasErrors() && session.getTransaction().isActive()) {
					session.getTransaction().commit();
				}
			}
		});
	}

	public boolean accepts(ResourceMethod method) {
		return true;
	}
}