pinguin-server/games/gameserver.go

package games

import (
	"encoding/json"

	extStomp "github.com/go-stomp/stomp"
)

type GameServer struct {
	addGameChan   chan addGameChan
	listGamesChan chan gamesChan
	getGameChan   chan getGame
}

type addGameChan struct {
	game *Game
	id chan int64
}

type gamesChan chan []*Game

type getGame struct {
	id         int64
	returnChan chan *Game
}

func New() GameServer {

	gameServer := GameServer{
		addGameChan:   make(chan addGameChan),
		listGamesChan: make(chan gamesChan),
		getGameChan:   make(chan getGame),
	}

	go serveGames(gameServer)
	return gameServer
}

func serveGames(gameServer GameServer) {
	gamesMap := make(map[int64]*Game)
	gameCounter := int64(0)

	for {
		select {
		case addGameChan := <-gameServer.addGameChan:
			addGameChan.game.Id = gameCounter
			gameCounter++
			gamesMap[addGameChan.game.Id] = addGameChan.game
			addGameChan.id <- addGameChan.game.Id
		case r := <-gameServer.listGamesChan:
			v := make([]*Game, 0, len(gamesMap))
			for _, value := range gamesMap {
				v = append(v, value)
			}
			r <- v
		case r := <-gameServer.getGameChan:
			r.returnChan <- gamesMap[r.id]
		}
	}
}

type stompGameStart struct {
	Type   string `json:"type"`
	GameId int64  `json:"gameId"`
}

func (games GameServer) AddGame(x, y int, stompConnection *extStomp.Conn) *Game {
	game2 := NewGame(x, y)
	addGame := addGameChan{
		game: game2,
		id: make(chan int64),
	}
	
	games.addGameChan <- addGame
	gameId := <- addGame.id
	
	stompGameStart := stompGameStart{
		Type:   "GAME_START",
		GameId: gameId,
	}
	b, _ := json.Marshal(stompGameStart)
	stompConnection.Send("/topic/go-battlecamp.game", "application/json;charset=utf-8", b, extStomp.SendOpt.NoContentLength)
	return game2
}

func (games GameServer) ListGames() []*Game {
	r := make(chan []*Game)
	games.listGamesChan <- r
	return <-r
}

func (games GameServer) GetGame(id int64) *Game {
	getGame := getGame{
		id:         id,
		returnChan: make(chan *Game),
	}
	games.getGameChan <- getGame
	return <-getGame.returnChan
}