001    package org.granite.gravity.jetty8;
002    
003    import javax.servlet.ServletConfig;
004    import javax.servlet.ServletException;
005    import javax.servlet.http.HttpServletRequest;
006    
007    import org.eclipse.jetty.websocket.WebSocket;
008    import org.eclipse.jetty.websocket.WebSocketHandler;
009    import org.eclipse.jetty.websocket.WebSocketServlet;
010    import org.granite.gravity.GravityServletUtil;
011    
012    
013    public class JettyWebSocketServlet extends WebSocketServlet {
014            
015            private static final long serialVersionUID = 1L;
016            
017            private WebSocketHandler webSocketHandler;
018            
019            @Override
020            public void init(ServletConfig config) throws ServletException {
021                    super.init(config);
022                    
023                    GravityServletUtil.init(config);
024                    
025                    webSocketHandler = new JettyWebSocketHandler(getServletContext());
026            }
027    
028            
029            public WebSocket doWebSocketConnect(HttpServletRequest request, String protocol) {
030                    return webSocketHandler.doWebSocketConnect(request, protocol);
031            }
032    }