pl.bristleback.server.bristle.engine.tomcat.servlet.TomcatServletWebsocketEngine.java Source code

Introduction

Here is the source code for pl.bristleback.server.bristle.engine.tomcat.servlet.TomcatServletWebsocketEngine.java

Source

/*
 * Bristleback Websocket Framework - Copyright (c) 2010-2013 http://bristleback.pl
 * ---------------------------------------------------------------------------
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by the
 * Free Software Foundation; either version 3 of the License, or (at your
 * option) any later version.
 * This library is distributed in the hope that it will be useful,
 * but without any warranty; without even the implied warranty of merchantability
 * or fitness for a particular purpose.
 * You should have received a copy of the GNU Lesser General Public License along
 * with this program; if not, see <http://www.gnu.org/licenses/lgpl.html>.
 * ---------------------------------------------------------------------------
 */

package pl.bristleback.server.bristle.engine.tomcat.servlet;

import org.apache.catalina.connector.RequestFacade;
import org.apache.log4j.Logger;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Component;
import pl.bristleback.server.bristle.api.FrontController;
import pl.bristleback.server.bristle.api.ServerEngine;
import pl.bristleback.server.bristle.api.ServletServerEngine;
import pl.bristleback.server.bristle.api.WebsocketConnector;
import pl.bristleback.server.bristle.conf.BristleInitializationException;
import pl.bristleback.server.bristle.engine.base.AbstractServerEngine;
import pl.bristleback.server.bristle.engine.tomcat.TomcatConnector;

import javax.inject.Inject;
import javax.inject.Named;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;

/**
 * Tomcat 7.0.32+ WebSocket ServerEngine
 * <p/>
 * Created on: 2012-06-17 10:14:44 <br/>
 *
 * @author Andrea Nanni
 */
@Component(TomcatServletWebsocketEngine.ENGINE_NAME)
@Lazy
public class TomcatServletWebsocketEngine extends AbstractServerEngine
        implements ServerEngine, ServletServerEngine {
    private static Logger log = Logger.getLogger(TomcatServletWebsocketEngine.class.getName());

    public static final String ENGINE_NAME = "system.engine.tomcat.servlet";

    @Inject
    @Named("defaultFrontController")
    private FrontController frontController;

    private TomcatWebsocketServletHolder servletHolder;

    public void handleRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        HttpServletRequest hsr = request;
        while (hsr instanceof HttpServletRequestWrapper) {
            hsr = (HttpServletRequest) ((HttpServletRequestWrapper) hsr).getRequest();
        }
        RequestFacade facade = (RequestFacade) hsr;

        servletHolder.service(facade, response);
    }

    @Override
    public void startServer() {
        log.info("Bristleback Tomcat servlet engine started.");
        servletHolder = new TomcatWebsocketServletHolder(this);
        try {
            servletHolder.init();
        } catch (ServletException e) {
            throw new BristleInitializationException("Exception while initializing Tomcat Servlet Engine", e);
        }

    }

    @Override
    public void stopServer() {
        log.info("Bristleback Tomcat servlet engine stopped.");
    }

    @Override
    public void sendMessage(WebsocketConnector connector, String contentAsString) throws Exception {
        CharBuffer buffer = CharBuffer.wrap(contentAsString);
        ((TomcatConnector) connector).getConnection().writeTextMessage(buffer);
    }

    @Override
    public void sendMessage(WebsocketConnector connector, byte[] contentAsBytes) throws Exception {
        ByteBuffer buffer = ByteBuffer.wrap(contentAsBytes);
        ((TomcatConnector) connector).getConnection().writeBinaryMessage(buffer);
    }

    public FrontController getFrontController() {
        return frontController;
    }
}