| /* |
| * 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.support; |
| |
| import org.codehaus.groovy.grails.web.servlet.GrailsApplicationAttributes; |
| import org.hibernate.FlushMode; |
| import org.hibernate.Session; |
| import org.hibernate.SessionFactory; |
| import org.springframework.context.ApplicationContext; |
| import org.springframework.dao.DataAccessResourceFailureException; |
| import org.springframework.orm.hibernate3.SessionFactoryUtils; |
| import org.springframework.orm.hibernate3.support.OpenSessionInViewFilter; |
| import org.springframework.web.context.WebApplicationContext; |
| import org.springframework.web.context.support.WebApplicationContextUtils; |
| |
| import javax.servlet.FilterChain; |
| import javax.servlet.ServletException; |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| import java.io.IOException; |
| |
| /** |
| * <p>Uses the GrailsApplication sesssionFactory to apply the filter |
| * |
| * @author Graeme Rocher |
| * @since Oct 20, 2005 |
| */ |
| public class GrailsOpenSessionInViewFilter extends OpenSessionInViewFilter { |
| |
| protected SessionFactory lookupSessionFactory() { |
| WebApplicationContext parent = |
| WebApplicationContextUtils.getRequiredWebApplicationContext(getServletContext()); |
| |
| ApplicationContext context = (ApplicationContext)getServletContext().getAttribute(GrailsApplicationAttributes.APPLICATION_CONTEXT); |
| |
| if(context != null) { |
| return (SessionFactory) context.getBean(getSessionFactoryBeanName(), SessionFactory.class); |
| } |
| return null; |
| } |
| |
| protected Session getSession(SessionFactory sessionFactory) throws DataAccessResourceFailureException { |
| Session session = SessionFactoryUtils.getSession(sessionFactory, true); |
| session.setFlushMode(FlushMode.AUTO); |
| return session; |
| } |
| |
| protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException { |
| SessionFactory sf = lookupSessionFactory(); |
| if(sf != null) { |
| super.doFilterInternal(request,response,filterChain); |
| } |
| else { |
| filterChain.doFilter(request,response); |
| } |
| } |
| |
| protected void closeSession(Session session, SessionFactory sessionFactory) { |
| if(!FlushMode.NEVER.equals(session.getFlushMode())) { |
| session.flush(); |
| } |
| super.closeSession(session, sessionFactory); |
| } |
| } |