001/**
002 *   GRANITE DATA SERVICES
003 *   Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004 *
005 *   This file is part of the Granite Data Services Platform.
006 *
007 *   Granite Data Services is free software; you can redistribute it and/or
008 *   modify it under the terms of the GNU Lesser General Public
009 *   License as published by the Free Software Foundation; either
010 *   version 2.1 of the License, or (at your option) any later version.
011 *
012 *   Granite Data Services is distributed in the hope that it will be useful,
013 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
014 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
015 *   General Public License for more details.
016 *
017 *   You should have received a copy of the GNU Lesser General Public
018 *   License along with this library; if not, write to the Free Software
019 *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
020 *   USA, or see <http://www.gnu.org/licenses/>.
021 */
022package org.granite.gravity.glassfish;
023
024import javax.servlet.ServletConfig;
025import javax.servlet.ServletException;
026import javax.servlet.ServletRegistration;
027import javax.servlet.http.HttpServlet;
028
029import org.granite.gravity.Gravity;
030import org.granite.gravity.GravityManager;
031import org.granite.gravity.GravityServletUtil;
032import org.granite.logging.Logger;
033
034import com.sun.grizzly.websockets.WebSocketApplication;
035import com.sun.grizzly.websockets.WebSocketEngine;
036
037
038public class GlassFishWebSocketServlet extends HttpServlet {
039        
040        @SuppressWarnings("unused")
041        private static final Logger log = Logger.getLogger(GlassFishWebSocketServlet.class);
042
043        private static final long serialVersionUID = 1L;
044        
045        private WebSocketApplication app;
046        
047        @SuppressWarnings("deprecation")
048        @Override
049        public void init(ServletConfig config) throws ServletException {
050                super.init(config);
051                
052                GravityServletUtil.init(config);
053                Gravity gravity = GravityManager.getGravity(getServletContext());
054                String mapping = null;
055                for (ServletRegistration sr : getServletContext().getServletRegistrations().values()) {
056                        if (!sr.getClassName().equals(getClass().getName()))
057                                continue;
058                        mapping = sr.getMappings().iterator().next(); 
059                }
060                app = new GlassFishWebSocketApplication(getServletContext(), gravity, mapping);
061                try {
062                        WebSocketEngine.getEngine().register(getServletContext().getContextPath(), mapping, app);
063                }
064                catch (NoSuchMethodError e) {
065                        // Deprecated since Grizzly 1.5.?
066                        WebSocketEngine.getEngine().register(app);
067                }
068        }
069
070    @Override
071    public void destroy() {
072        WebSocketEngine.getEngine().unregister(app);
073        app = null;
074    }
075
076}