org.bitcoinrt.server.ConnectionBroker.java Source code

Introduction

Here is the source code for org.bitcoinrt.server.ConnectionBroker.java

Source

/*
 * Copyright 2002-2013 the original author or authors.
 *
 * 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.bitcoinrt.server;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationListener;
import org.springframework.integration.Message;
import org.springframework.integration.ip.IpHeaders;
import org.springframework.integration.ip.tcp.connection.TcpConnectionEvent;
import org.springframework.integration.ip.tcp.connection.TcpConnectionEvent.EventType;
import org.springframework.integration.ip.tcp.connection.TcpConnectionEvent.TcpConnectionEventType;
import org.springframework.integration.support.MessageBuilder;
import org.springframework.integration.x.ip.websocket.WebSocketEvent.WebSocketEventType;

/**
 * Serves as a registry for connected clients.
 *
 * @author Gunnar Hillert
 * @since 1.0
 *
 */
public class ConnectionBroker implements ApplicationListener<TcpConnectionEvent> {

    private static final Logger logger = LoggerFactory.getLogger(ConnectionBroker.class);

    private final Map<String, AtomicInteger> clients = new ConcurrentHashMap<String, AtomicInteger>();

    public ConnectionBroker() {
        super();
    }

    public List<Message<?>> createBroadCastMessages(Message<?> message) {

        final List<Message<?>> messages = new ArrayList<Message<?>>();

        for (String connectionId : clients.keySet()) {
            messages.add(
                    MessageBuilder.fromMessage(message).setHeader(IpHeaders.CONNECTION_ID, connectionId).build());
        }

        return messages;

    }

    /**
     * Returns the number of currently connected clients.
     */
    public int connectedClients() {
        return clients.size();
    }

    @Override
    public void onApplicationEvent(TcpConnectionEvent event) {

        final EventType eventType = event.getType();

        if (TcpConnectionEventType.OPEN.equals(eventType)) {

        } else if (TcpConnectionEventType.CLOSE.equals(eventType)) {
            logger.info("TCP Event: CLOSED - Removing client: {}", event.getConnectionId());
            clients.remove(event.getConnectionId());
        } else if (TcpConnectionEventType.EXCEPTION.equals(eventType)) {
            logger.warn("EXCEPTION Event - Removing client: {}", event.getConnectionId());
            clients.remove(event.getConnectionId());
        } else if (WebSocketEventType.HANDSHAKE_COMPLETE.equals(eventType)) {
            logger.info("HANDSHAKE_COMPLETE - Adding client: {}", event.getConnectionId());
            clients.put(event.getConnectionId(), new AtomicInteger());
        } else if (WebSocketEventType.WEBSOCKET_CLOSED.equals(eventType)) {
            logger.info("WEBSOCKET_CLOSED - Removing client: {}", event.getConnectionId());
            clients.remove(event.getConnectionId());
        } else {
            throw new IllegalArgumentException(String.format("EventType '%s' is not supported.", eventType));
        }
    }
}