com.codeminders.socketio.server.transport.AbstractTransportProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of socket-io Show documentation
Socket.IO server classes
There is a newer version: 1.0.10
package com.codeminders.socketio.server.transport;

import com.codeminders.socketio.protocol.EngineIOProtocol;
import com.codeminders.socketio.server.*;

import javax.servlet.ServletConfig;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import java.util.*;
import java.util.logging.Logger;

/**
 * @author Alexander Sova ([email protected])
 */
public abstract class AbstractTransportProvider implements TransportProvider {

    private static final Logger LOGGER = Logger.getLogger(AbstractTransportProvider.class.getName());

    protected Map transports = new EnumMap<>(TransportType.class);

    /**
     *   Creates and initializes all available transports
     */
    @Override
    public void init(ServletConfig config, ServletContext context)
            throws ServletException
    {
        addIfNotNull(TransportType.XHR_POLLING,   createXHRPollingTransport());
        addIfNotNull(TransportType.JSONP_POLLING, createJSONPPollingTransport());
        addIfNotNull(TransportType.WEB_SOCKET,    createWebSocketTransport());

        for(Transport t : transports.values())
            t.init(config, context);
    }

    @Override
    public void destroy()
    {
        for (Transport t : getTransports())
            t.destroy();
    }

    @Override
    public Transport getTransport(ServletRequest request)
            throws UnsupportedTransportException, SocketIOProtocolException
    {
        String transportName = request.getParameter(EngineIOProtocol.TRANSPORT);
        if(transportName == null)
            throw new SocketIOProtocolException("Missing transport parameter");

        TransportType type = TransportType.UNKNOWN;

        if("websocket".equals(transportName))
            type = TransportType.from(transportName);

        if("polling".equals(transportName)) {
            if(request.getParameter(EngineIOProtocol.JSONP_INDEX) != null)
                type = TransportType.JSONP_POLLING;
            else
                type = TransportType.XHR_POLLING;
        }

        Transport t = transports.get(type);
        if(t == null)
            throw new UnsupportedTransportException(transportName);

        return t;
    }

    @Override
    public Transport getTransport(TransportType type)
    {
        return transports.get(type);
    }

    @Override
    public Collection getTransports()
    {
        return transports.values();
    }

    protected Transport createXHRPollingTransport()
    {
        return new XHRPollingTransport();
    }

    protected Transport createJSONPPollingTransport()
    {
        return null;
    }

    protected Transport createWebSocketTransport()
    {
        return null;
    }

    private void addIfNotNull(TransportType type, Transport transport)
    {
        if(transport != null)
            transports.put(type, transport);
    }
}