org.socketio.netty.storage.SessionStorage Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2012 Ronen Hamias, Anton Kharenko
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.socketio.netty.storage;

import java.net.SocketAddress;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.socketio.netty.TransportType;
import org.socketio.netty.packets.ConnectPacket;
import org.socketio.netty.pipeline.UnsupportedTransportTypeException;
import org.socketio.netty.session.*;
import org.socketio.netty.storage.memoizer.Computable;
import org.socketio.netty.storage.memoizer.MemoizerConcurrentMap;

import io.netty.channel.Channel;

/**
 * 
 * @author Anton Kharenko
 * 
 */
public class SessionStorage {

	private final Logger log = LoggerFactory.getLogger(getClass());
	
	private final MemoizerConcurrentMap sessionsMemoizer = new MemoizerConcurrentMap();
	
	private final int localPort;
	
	public SessionStorage(int localPort) {
		this.localPort = localPort;
	}

	public boolean containSession(final String sessionId) {
		return sessionsMemoizer.containsKey(sessionId);
	}
	
	public void removeSession(final String sessionId) {
		sessionsMemoizer.remove(sessionId);
	}

	public IManagedSession getSession(final ConnectPacket connectPacket,
			final Channel channel,
			final ISessionDisconnectHandler disconnectHandler) throws Exception {
		
		IManagedSession session = createSession(connectPacket, channel, disconnectHandler, null);
		
		// If transport protocol was changed then remove old session and create new one instead
		if (connectPacket.getTransportType() != session.getTransportType()) {
			session.markAsUpdgraded();
			
			String oldSessionId = session.getSessionId();
			TransportType oldTransportType = session.getTransportType();
			
			final String sessionId = connectPacket.getSessionId();
			removeSession(sessionId);
			session = createSession(connectPacket, channel, disconnectHandler, session.getTransportType());
			
			log.debug("{} transport type {} session was upgraded to new transport type {} and session {}", 
					oldTransportType.name(), oldSessionId, session.getTransportType().name(), session.getSessionId());
		}
		
		return session;
	}
	
	private IManagedSession createSession(final ConnectPacket connectPacket,
			final Channel channel,
			final ISessionDisconnectHandler disconnectHandler,
			final TransportType upgradedFromTransportType) throws Exception {
		final TransportType transportType = connectPacket.getTransportType();
		final String sessionId = connectPacket.getSessionId();
		final String origin = connectPacket.getOrigin();
		final String jsonpIndexParam = connectPacket.getJsonpIndexParam();
		final SocketAddress remoteAddress = connectPacket.getRemoteAddress();
		try {
			return sessionsMemoizer.get(sessionId,
					new Computable() {
						@Override
						public IManagedSession compute(String sessionId) throws Exception {
							if (transportType == TransportType.WEBSOCKET) {
								return new WebSocketSession(channel, sessionId,
										origin, disconnectHandler, upgradedFromTransportType, localPort, remoteAddress);
							} else if (transportType == TransportType.FLASHSOCKET) {
								return new FlashSocketSession(channel, sessionId,
										origin, disconnectHandler, upgradedFromTransportType, localPort, remoteAddress);
							} else if (transportType == TransportType.XHR_POLLING) {
								return new XHRPollingSession(channel,sessionId, 
										origin, disconnectHandler, upgradedFromTransportType, localPort, remoteAddress);
							} else if (transportType == TransportType.JSONP_POLLING) {
								return new JsonpPollingSession(channel,sessionId, 
										origin, disconnectHandler, upgradedFromTransportType, localPort, jsonpIndexParam, remoteAddress);
							} else {
								throw new UnsupportedTransportTypeException(transportType);
							}
						}
					});
		} catch (Exception e) {
			throw new Exception(String.format(
					"Failed to create new session: %s",
					connectPacket.toString()), e);
		}
	}

	public IManagedSession getSessionIfExist(final String sessionId) {
		IManagedSession session = null;
		try {
			session = sessionsMemoizer.containsKey(sessionId) ? sessionsMemoizer
					.get(sessionId) : null;
		} catch (Exception ignore) {
		}
		return session;
	}
}