com.corundumstudio.socketio.namespace.Namespace Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2012 Nikita Koksharov
 *
 * 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 com.corundumstudio.socketio.namespace;

import java.util.Collections;
import java.util.Queue;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ConcurrentMap;

import com.corundumstudio.socketio.AckRequest;
import com.corundumstudio.socketio.BroadcastOperations;
import com.corundumstudio.socketio.SocketIOClient;
import com.corundumstudio.socketio.SocketIONamespace;
import com.corundumstudio.socketio.annotation.ScannerEngine;
import com.corundumstudio.socketio.listener.ConnectListener;
import com.corundumstudio.socketio.listener.DataListener;
import com.corundumstudio.socketio.listener.DisconnectListener;
import com.corundumstudio.socketio.misc.ConcurrentHashSet;
import com.corundumstudio.socketio.parser.JsonSupport;
import com.corundumstudio.socketio.transport.NamespaceClient;

/**
 * Hub object for all clients in one namespace.
 * Namespace shares by different namespace-clients.
 *
 * @see com.corundumstudio.socketio.transport.NamespaceClient
 */
public class Namespace implements SocketIONamespace {

    public static final String DEFAULT_NAME = "";

    private final Set allClients = new ConcurrentHashSet();
    private final ConcurrentMap> eventListeners =
                                                            new ConcurrentHashMap>();
    private final ConcurrentMap, Queue>> jsonObjectListeners =
                                                            new ConcurrentHashMap, Queue>>();
    private final Queue> messageListeners = new ConcurrentLinkedQueue>();
    private final Queue connectListeners = new ConcurrentLinkedQueue();
    private final Queue disconnectListeners = new ConcurrentLinkedQueue();

    private final ConcurrentMap> roomClients = new ConcurrentHashMap>();

    private final String name;
    private final JsonSupport jsonSupport;

    public Namespace(String name, JsonSupport jsonSupport) {
        super();
        this.name = name;
        this.jsonSupport = jsonSupport;
    }

    public void addClient(SocketIOClient client) {
        allClients.add(client);
    }

    public String getName() {
        return name;
    }

    @Override
    @SuppressWarnings({"unchecked", "rawtypes"})
    public  void addEventListener(String eventName, Class eventClass, DataListener listener) {
        EventEntry entry = eventListeners.get(eventName);
        if (entry == null) {
            entry = new EventEntry(eventClass);
            EventEntry oldEntry = eventListeners.putIfAbsent(eventName, entry);
            if (oldEntry != null) {
                entry = oldEntry;
            }
        }
        entry.addListener(listener);
        jsonSupport.addEventMapping(eventName, eventClass);
    }

    @Override
    public  void addJsonObjectListener(Class clazz, DataListener listener) {
        Queue> queue = jsonObjectListeners.get(clazz);
        if (queue == null) {
            queue = new ConcurrentLinkedQueue>();
            Queue> oldQueue = jsonObjectListeners.putIfAbsent(clazz, queue);
            if (oldQueue != null) {
                queue = oldQueue;
            }
        }
        queue.add(listener);
        jsonSupport.addJsonClass(clazz);
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    public void onEvent(NamespaceClient client, String eventName, Object data, AckRequest ackRequest) {
        EventEntry entry = eventListeners.get(eventName);
        if (entry == null) {
            return;
        }
        Queue listeners = entry.getListeners();
        for (DataListener dataListener : listeners) {
            dataListener.onData(client, data, ackRequest);
        }
    }

    public void onMessage(NamespaceClient client, String data, AckRequest ackRequest) {
        for (DataListener listener : messageListeners) {
            listener.onData(client, data, ackRequest);
        }
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    public void onJsonObject(NamespaceClient client, Object data, AckRequest ackRequest) {
        Queue> queue = jsonObjectListeners.get(data.getClass());
        if (queue == null) {
            return;
        }
        for (DataListener dataListener : queue) {
            dataListener.onData(client, data, ackRequest);
        }
    }

    @Override
    public void addDisconnectListener(DisconnectListener listener) {
        disconnectListeners.add(listener);
    }

    public void onDisconnect(SocketIOClient client) {
        for (DisconnectListener listener : disconnectListeners) {
            listener.onDisconnect(client);
        }
        allClients.remove(client);
    }

    @Override
    public void addConnectListener(ConnectListener listener) {
        connectListeners.add(listener);
    }

    public void onConnect(SocketIOClient client) {
        for (ConnectListener listener : connectListeners) {
            listener.onConnect(client);
        }
    }

    @Override
    public void addMessageListener(DataListener listener) {
        messageListeners.add(listener);
    }

    public Queue> getMessageListeners() {
        return messageListeners;
    }

    @Override
    public BroadcastOperations getBroadcastOperations() {
        return new BroadcastOperations(allClients);
    }


    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + ((name == null) ? 0 : name.hashCode());
        return result;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        Namespace other = (Namespace) obj;
        if (name == null) {
            if (other.name != null)
                return false;
        } else if (!name.equals(other.name))
            return false;
        return true;
    }

    @Override
    public void addListeners(Object listeners) {
        ScannerEngine engine = new ScannerEngine();
        engine.scan(this, listeners, listeners.getClass());
    }

    @Override
    public void addListeners(Object listeners, Class listenersClass) {
        ScannerEngine engine = new ScannerEngine();
        engine.scan(this, listeners, listenersClass);
    }

    public void joinRoom(Object roomKey, SocketIOClient namespaceClient) {
        Queue clients = roomClients.get(roomKey);
        if (clients == null) {
            clients = new ConcurrentLinkedQueue();
            Queue oldClients = roomClients.putIfAbsent(roomKey, clients);
            if (oldClients != null) {
                clients = oldClients;
            }
        }
        clients.add(namespaceClient);
        if (clients != roomClients.get(roomKey)) {
            // re-join if queue has been replaced
            joinRoom(roomKey, namespaceClient);
        }
    }

    public void leaveRoom(Object roomKey, SocketIOClient namespaceClient) {
        Queue clients = roomClients.get(roomKey);
        if (clients == null) {
            return;
        }
        clients.remove(namespaceClient);
        if (clients.isEmpty()) {
            roomClients.remove(roomKey);
            // join which was added after queue deletion
            for (SocketIOClient socketIOClient : clients) {
                joinRoom(roomKey, socketIOClient);
            }
        }
    }

    public Iterable getRoomClients(Object roomKey) {
        Queue clients = roomClients.get(roomKey);
        if (clients == null) {
            return Collections.emptyList();
        }
        return clients;
    }

}