com.music.web.websocket.GameHandler.java Source code

Introduction

Here is the source code for com.music.web.websocket.GameHandler.java

Source

/*
 * Computoser is a music-composition algorithm and a website to present the results
 * Copyright (C) 2012-2014  Bozhidar Bozhanov
 *
 * Computoser is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * Computoser is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with Computoser.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.music.web.websocket;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;

import javax.inject.Inject;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.AbstractWebSocketHandler;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.music.service.PieceService;
import com.music.web.websocket.dto.GameEvent;
import com.music.web.websocket.dto.GameEventType;
import com.music.web.websocket.dto.GameMessage;

@Component
public class GameHandler extends AbstractWebSocketHandler {

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

    private ObjectMapper mapper = new ObjectMapper();

    @Inject
    private PieceService pieceService;

    private Map<String, Game> games = new ConcurrentHashMap<>();
    private Map<String, Game> pendingGames = new ConcurrentHashMap<>();
    private Map<String, Player> players = new ConcurrentHashMap<>();
    private Map<String, Game> playerGames = new ConcurrentHashMap<>();

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        Player player = new Player(session);
        players.put(session.getId(), player);
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        leave(session.getId());
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage textMessage) throws Exception {
        try {
            GameMessage message = getMessage(textMessage);
            switch (message.getAction()) {
            case INITIALIZE:
                initialize(message, session);
                break;
            case JOIN:
                join(message.getGameId(), message.getPlayerName(), session);
                break;
            case LEAVE:
                leave(session.getId());
                break;
            case START:
                startGame(message);
                break;
            case ANSWER:
                answer(message, session.getId());
                break;
            case JOIN_RANDOM:
                joinRandomGame(message.getPlayerName(), session);
                break;
            }
        } catch (Exception ex) {
            logger.error("Exception occurred while handling message", ex);
        }
    }

    private void startGame(GameMessage message) {
        Game game = games.get(message.getGameId());
        if (game != null) {
            game.start();
        }
        pendingGames.remove(game.getId());
    }

    private void answer(GameMessage message, String playerId) {
        Game game = games.get(message.getGameId());
        Player player = game.getPlayers().get(playerId);
        player.answer(game, message.getAnswer());
    }

    private void leave(String playerId) {
        players.remove(playerId);
        Game game = playerGames.remove(playerId);
        if (game != null) {
            game.playerLeft(playerId);
            // if this is the last player, remove the orphaned game
            if (game.getPlayers().isEmpty()) {
                games.remove(game.getId());
                pendingGames.remove(game.getId());
            }
        }
    }

    private void join(String gameId, String playerName, WebSocketSession session) {
        Game game = games.get(gameId);
        if (game == null) {
            sendMessage(new GameEvent(GameEventType.NO_GAME_AVAILABLE), session);
        }
        if (!game.isStarted()) {
            String playerId = session.getId();
            Player player = players.get(playerId);
            player.setName(playerName);
            playerGames.put(playerId, game);
            if (!game.playerJoined(player)) {
                sendMessage(new GameEvent(GameEventType.PLAYER_NAME_TAKEN), session);
            }
        } else {
            sendMessage(new GameEvent(GameEventType.GAME_ALREADY_STARTED), session);
        }
    }

    public static void sendMessage(GameEvent event, WebSocketSession session) {
        try {
            session.sendMessage(new TextMessage(event.toJson()));
        } catch (IOException e) {
            throw new IllegalStateException(e);
        }
    }

    private void joinRandomGame(String playerName, WebSocketSession session) {
        if (pendingGames.isEmpty()) {
            sendMessage(new GameEvent(GameEventType.NO_GAME_AVAILABLE), session);
            return;
        }
        List<Game> games = new ArrayList<>(this.pendingGames.values());
        Random random = new Random();

        Game game = pendingGames.get(random.nextInt(games.size()));
        join(game.getId(), playerName, session);
        GameEvent event = new GameEvent(GameEventType.RANDOM_GAME_JOINED);
        event.setGameId(game.getId());
        sendMessage(event, session);
    }

    private void initialize(GameMessage message, WebSocketSession session) {
        String gameId = UUID.randomUUID().toString();
        Game game = new Game(gameId, pieceService);
        games.put(gameId, game);
        pendingGames.put(gameId, game);

        join(gameId, message.getPlayerName(), session);
        GameEvent event = new GameEvent(GameEventType.GAME_INITIALIZED);
        event.setGameId(gameId);
        sendMessage(event, session);
    }

    public GameMessage getMessage(TextMessage message) {
        try {
            return mapper.readValue(message.getPayload(), GameMessage.class);
        } catch (IOException e) {
            throw new IllegalStateException(e);
        }
    }

    public Map<String, Game> getGames() {
        return games;
    }

    public void setPieceService(PieceService pieceService) {
        this.pieceService = pieceService;
    }
}