ChatRoom.java

/**
 * Copyright (c) 2013-2016, The SeedStack authors <http://seedstack.org>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.seedstack.seed.web.fixtures.websocket;

import org.seedstack.seed.it.ITBind;

import javax.inject.Inject;
import javax.inject.Singleton;
import javax.websocket.Session;
import java.util.Collections;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

@ITBind
@Singleton
class ChatRoom {
    private final Set<Session> sessions = Collections.newSetFromMap(new ConcurrentHashMap<>());
    @Inject
    EchoService echoService;

    void addClient(Session session) {
        sessions.add(session);
    }

    void removeClient(Session session) {
        sessions.remove(session);
    }

    void sendMessage(ChatMessage chatMessage) {
        for (Session session : sessions) {
            session.getAsyncRemote().sendText(echoService.echo(chatMessage.getMessage()));
        }
    }
}