001 /*
002 GRANITE DATA SERVICES
003 Copyright (C) 2011 GRANITE DATA SERVICES S.A.S.
004
005 This file is part of Granite Data Services.
006
007 Granite Data Services is free software; you can redistribute it and/or modify
008 it under the terms of the GNU Library General Public License as published by
009 the Free Software Foundation; either version 2 of the License, or (at your
010 option) any later version.
011
012 Granite Data Services is distributed in the hope that it will be useful, but
013 WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
014 FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public License
015 for more details.
016
017 You should have received a copy of the GNU Library General Public License
018 along with this library; if not, see <http://www.gnu.org/licenses/>.
019 */
020
021 package org.granite.config.servlet3;
022
023 import java.util.HashSet;
024 import java.util.Set;
025
026 import javax.servlet.FilterRegistration;
027 import javax.servlet.Servlet;
028 import javax.servlet.ServletContainerInitializer;
029 import javax.servlet.ServletContext;
030 import javax.servlet.ServletException;
031 import javax.servlet.ServletRegistration;
032 import javax.servlet.annotation.HandlesTypes;
033
034 import org.granite.config.GraniteConfigListener;
035 import org.granite.messaging.webapp.AMFMessageFilter;
036 import org.granite.messaging.webapp.AMFMessageServlet;
037 import org.granite.util.ClassUtil;
038
039 /**
040 * @author William DRAI
041 */
042 @HandlesTypes({FlexFilter.class})
043 public class GraniteServlet3Initializer implements ServletContainerInitializer {
044
045 public void onStartup(Set<Class<?>> scannedClasses, ServletContext servletContext) throws ServletException {
046 Set<Class<?>> classes = new HashSet<Class<?>>();
047 if (scannedClasses != null) {
048 classes.addAll(scannedClasses);
049 classes.remove(FlexFilter.class); // JBossWeb adds the annotation ???
050 }
051 if (classes.size() > 1)
052 throw new ServletException("Application must have only one FlexFilter configuration class");
053
054 if (!classes.isEmpty()) {
055 // Configure GraniteDS only if we find a config class annotated with @FlexFilter
056 Class<?> clazz = classes.iterator().next();
057 FlexFilter flexFilter = clazz.getAnnotation(FlexFilter.class);
058
059 servletContext.setAttribute(GraniteConfigListener.GRANITE_CONFIG_ATTRIBUTE, clazz);
060
061 servletContext.addListener(new GraniteConfigListener());
062
063 if (servletContext.getFilterRegistration("AMFMessageFilter") == null) {
064 FilterRegistration.Dynamic graniteFilter = servletContext.addFilter("AMFMessageFilter", AMFMessageFilter.class);
065 graniteFilter.addMappingForUrlPatterns(null, true, flexFilter.graniteUrlMapping());
066 }
067 if (servletContext.getServletRegistration("AMFMessageServlet") == null) {
068 ServletRegistration.Dynamic graniteServlet = servletContext.addServlet("AMFMessageServlet", AMFMessageServlet.class);
069 graniteServlet.setLoadOnStartup(1);
070 graniteServlet.addMapping(flexFilter.graniteUrlMapping());
071 }
072
073 try {
074 if (servletContext.getServletRegistration("GravityServlet") == null) {
075 Class<? extends Servlet> gravityAsyncServletClass = ClassUtil.forName("org.granite.gravity.servlet3.GravityAsyncServlet", Servlet.class);
076 ServletRegistration.Dynamic gravityServlet = servletContext.addServlet("GravityServlet", gravityAsyncServletClass);
077 gravityServlet.setLoadOnStartup(1);
078 gravityServlet.setAsyncSupported(true);
079 gravityServlet.addMapping(flexFilter.gravityUrlMapping());
080 }
081 }
082 catch (ClassNotFoundException e) {
083 servletContext.log("Could not setup GravityAsyncServlet", e);
084 }
085 }
086 }
087 }