server/control.go

391 lines
9.8 KiB
Go
Raw Normal View History

2014-04-23 14:28:13 -07:00
package server
2013-08-28 23:46:12 -07:00
import (
2013-09-01 23:00:09 -07:00
"encoding/json"
2013-11-13 23:45:02 -08:00
"errors"
"fmt"
"io/ioutil"
2013-09-28 23:00:29 -07:00
"log"
2013-08-28 23:46:12 -07:00
"net/http"
"os"
2013-09-28 23:00:29 -07:00
"runtime/pprof"
"strings"
2014-03-29 01:57:29 -07:00
"sync"
"github.com/elazarl/go-bindata-assetfs"
"golang.org/x/net/websocket"
"mcquay.me/idg"
2013-08-28 23:46:12 -07:00
)
2014-04-14 00:26:41 -07:00
// JsonHandler is a function type that allows setting the Content-Type
// appropriately for views destined to serve JSON
2013-08-28 23:46:12 -07:00
type JsonHandler func(http.ResponseWriter, *http.Request)
2014-04-14 00:26:41 -07:00
// ServeHTTP is JsonHandler's http.Handler implementation.
2013-08-28 23:46:12 -07:00
func (h JsonHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Type", "application/json")
h(w, req)
}
2013-09-01 23:00:09 -07:00
2014-04-14 00:26:41 -07:00
// Controller is the shepherd of a collection of games. The main package in
2014-04-23 14:28:13 -07:00
// server simply populates one of these and starts an http server.
type Controller struct {
2014-11-24 14:42:11 -08:00
Idg *idg.Generator
Conf Config
Games MapLock
Memprofile string
Profile string
}
var prefix map[string]string
2014-04-14 00:26:41 -07:00
// NewController takes a populated Config, and some parameters to determine
// what sorts of profiling to deal with and returns a freshly populated
// Controller.
func NewController(conf Config, mprof, pprof, staticFiles string) *http.ServeMux {
c := &Controller{
2014-11-24 14:42:11 -08:00
Idg: idg.NewGenerator(),
Conf: conf,
Games: MapLock{
M: make(map[string]*Game),
},
Memprofile: mprof,
Profile: pprof,
}
go c.Run()
prefix = map[string]string{
"ui": "/ui/",
"websocket": "/ws/",
"list": "/api/v0/game/list/",
"start": "/api/v0/game/start/",
"stats": "/api/v0/game/stats/",
"stop": "/api/v0/game/stop/",
"bandwidth": "/api/v0/game/bw/",
"fsu": "/api/v0/fsu/",
"info": "/api/v0/info/",
}
sm := http.NewServeMux()
sm.HandleFunc(
"/",
func(w http.ResponseWriter, r *http.Request) {
http.Redirect(w, r, prefix["ui"], http.StatusMovedPermanently)
},
)
if staticFiles == "" {
sm.Handle(
prefix["ui"],
http.FileServer(
&assetfs.AssetFS{
Asset: Asset,
AssetDir: AssetDir,
},
),
)
} else {
sm.Handle(
prefix["ui"],
http.StripPrefix(
prefix["ui"],
http.FileServer(http.Dir(staticFiles)),
),
)
}
sm.Handle(prefix["websocket"], websocket.Handler(c.AddPlayer))
sm.Handle(prefix["list"], JsonHandler(c.ListGames))
sm.Handle(prefix["start"], JsonHandler(c.StartGame))
sm.Handle(prefix["stats"], JsonHandler(c.GameStats))
sm.Handle(prefix["stop"], JsonHandler(c.StopGame))
sm.Handle(prefix["bandwidth"], JsonHandler(c.BW))
sm.HandleFunc(prefix["fsu"], c.KillServer)
sm.HandleFunc(prefix["info"], c.Info)
return sm
}
2014-04-14 00:26:41 -07:00
// TODO Eventually this thing will have a select loop for dealing with game
// access in a more lock-free manner?
func (c *Controller) Run() {
c.Idg.Run()
}
2014-04-14 00:26:41 -07:00
// StartGame is the http route responsible for responding to requests to start
// games under this controller. Creates a default Config object, and populates
// it according to data POSTed.
func (c *Controller) StartGame(w http.ResponseWriter, req *http.Request) {
2013-10-14 00:17:12 -07:00
log.Println("asked to create a game")
requested_game_name := c.Idg.Hash()
2014-04-26 14:56:12 -07:00
width, height := float64(c.Conf.Width), float64(c.Conf.Height)
2014-04-26 08:56:40 -07:00
obstacleCount := 0
obstacles := []Obstacle{}
maxPoints := c.Conf.MaxPoints
mode := "deathmatch"
// here we determine if we are going to run with defaults or pick them off
// a posted json blob
if req.Method == "POST" {
body, err := ioutil.ReadAll(req.Body)
if err != nil {
2014-04-14 00:39:42 -07:00
log.Printf("unable to read request body: %v", err)
}
req.Body.Close()
2013-11-13 20:38:57 -08:00
cfg := struct {
2013-11-29 00:10:49 -08:00
Name string `json:"name"`
Config
}{}
2013-11-13 20:38:57 -08:00
err = json.Unmarshal(body, &cfg)
if err != nil {
if err := json.NewEncoder(w).Encode(NewFailure(err.Error())); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
return
}
2013-11-13 20:38:57 -08:00
requested_game_name = cfg.Name
2014-04-26 14:56:12 -07:00
width = float64(cfg.Width)
height = float64(cfg.Height)
2014-04-26 08:56:40 -07:00
obstacleCount = cfg.ObstacleCount
obstacles = cfg.Obstacles
2013-11-29 00:10:49 -08:00
maxPoints = cfg.MaxPoints
mode = cfg.Mode
}
g := c.Games.Get(requested_game_name)
if g == nil {
log.Printf("Game '%s' non-existant; making it now", requested_game_name)
var err error
g, err = NewGame(
requested_game_name,
width,
height,
c.Conf.Tick,
maxPoints,
mode,
)
g.obstacleCount = obstacleCount
g.obstacles = obstacles
g.defaultObstacles = obstacles
if len(g.defaultObstacles) == 0 {
g.obstacles = GenerateObstacles(
g.obstacleCount,
g.width,
g.height,
)
} else {
g.obstacles = c.Conf.Obstacles
}
if err != nil {
log.Printf("problem creating game: %s: %s", requested_game_name, err)
b, _ := json.Marshal(NewFailure("game creation failure"))
http.Error(w, string(b), http.StatusConflict)
return
}
go g.run()
c.Games.Add(g)
} else {
2013-11-29 00:10:49 -08:00
log.Printf("Game '%s' already exists: %p", requested_game_name, g)
b, _ := json.Marshal(NewFailure("game already exists"))
http.Error(w, string(b), http.StatusConflict)
return
}
2013-09-01 23:00:09 -07:00
game_json := struct {
Id string `json:"id"`
}{
Id: g.id,
}
if err := json.NewEncoder(w).Encode(game_json); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
2014-04-14 00:26:41 -07:00
// ListGames makes a reasonable JSON response based on the games currently
// being run.
func (c *Controller) ListGames(w http.ResponseWriter, req *http.Request) {
2013-10-14 00:17:12 -07:00
log.Println("games list requested")
c.Games.RLock()
defer c.Games.RUnlock()
type pout struct {
Name string `json:"name"`
Id string `json:"id"`
}
2013-09-07 19:12:46 -07:00
type gl struct {
Id string `json:"id"`
Players []pout `json:"players"`
2013-09-07 19:12:46 -07:00
}
ids := make([]gl, 0)
for id, g := range c.Games.M {
players := make([]pout, 0)
2013-11-08 21:25:42 -08:00
// TODO - players instead of robots?
for p := range g.players {
2013-11-08 21:25:42 -08:00
for _, r := range p.Robots {
players = append(players, pout{
Name: r.Name,
Id: r.Id,
})
}
2013-09-07 19:12:46 -07:00
}
ids = append(ids, gl{
2013-09-07 19:12:46 -07:00
Id: id,
Players: players,
})
2013-09-01 23:00:09 -07:00
}
if err := json.NewEncoder(w).Encode(ids); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
2014-04-14 00:26:41 -07:00
// GameStats provides an control mechanism to query for the stats of a single
// game
func (c *Controller) GameStats(w http.ResponseWriter, req *http.Request) {
2013-11-15 08:54:52 -08:00
// TODO: wrap this up in something similar to the JsonHandler to verify the
// url? Look at gorilla routing?
key, err := c.getGameId(req.URL.Path)
2013-11-13 23:45:02 -08:00
if err != nil {
2013-11-15 08:54:52 -08:00
b, _ := json.Marshal(NewFailure(err.Error()))
http.Error(w, string(b), http.StatusBadRequest)
return
2013-11-13 23:45:02 -08:00
}
log.Printf("requested stats for game: %s", key)
c.Games.RLock()
g, ok := c.Games.M[key]
c.Games.RUnlock()
2013-11-13 23:45:02 -08:00
if !ok {
2013-11-15 08:54:52 -08:00
b, _ := json.Marshal(NewFailure("game not found"))
http.Error(w, string(b), http.StatusNotFound)
return
}
2014-01-16 00:02:59 -08:00
g.stats.RLock()
defer g.stats.RUnlock()
2014-01-16 00:13:02 -08:00
if err := json.NewEncoder(w).Encode(g.stats.PlayerStats); err != nil {
2013-11-13 23:45:02 -08:00
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
2014-04-14 00:26:41 -07:00
// BW provides a route to query for current bandwidth utilization for a single
// game.
func (c *Controller) BW(w http.ResponseWriter, req *http.Request) {
// TODO: wrap this up in something similar to the JsonHandler to verify the
// url? Look at gorilla routing?
key, err := c.getGameId(req.URL.Path)
if err != nil {
b, _ := json.Marshal(NewFailure(err.Error()))
http.Error(w, string(b), http.StatusBadRequest)
return
}
log.Printf("requested bandwidth for game: %s", key)
c.Games.RLock()
g, ok := c.Games.M[key]
c.Games.RUnlock()
if !ok {
b, _ := json.Marshal(NewFailure("game not found"))
http.Error(w, string(b), http.StatusNotFound)
return
}
s := map[string][]float64{
"tx": <-g.bw.Tx,
"rx": <-g.bw.Rx,
}
if err := json.NewEncoder(w).Encode(s); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
2014-04-14 00:26:41 -07:00
// StopGame is the only mechanism to decrease the number of running games in
// a Controller
func (c *Controller) StopGame(w http.ResponseWriter, req *http.Request) {
key, err := c.getGameId(req.URL.Path)
2013-11-13 23:45:02 -08:00
if err != nil {
2013-11-15 08:54:52 -08:00
b, _ := json.Marshal(NewFailure(err.Error()))
http.Error(w, string(b), http.StatusBadRequest)
return
2013-11-13 23:45:02 -08:00
}
c.Games.Lock()
g, ok := c.Games.M[key]
defer c.Games.Unlock()
if !ok {
http.NotFound(w, req)
return
}
2013-11-13 23:45:02 -08:00
g.kill <- true
delete(c.Games.M, key)
2013-11-13 23:45:02 -08:00
message := struct {
Ok bool `json:"ok"`
Message string `json:"message"`
}{
Ok: true,
Message: fmt.Sprintf("Successfully stopped game: %s", key),
}
if err := json.NewEncoder(w).Encode(message); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
log.Printf("returning from StopGame")
}
2013-09-28 23:00:29 -07:00
2014-04-23 14:28:13 -07:00
// KillServer is my favorite method of all the methods in server: it shuts
2014-04-14 00:26:41 -07:00
// things down respecting profiling requests.
func (c *Controller) KillServer(w http.ResponseWriter, req *http.Request) {
if c.Profile != "" {
2013-09-28 23:03:42 -07:00
log.Print("trying to stop cpu profile")
2013-09-28 23:00:29 -07:00
pprof.StopCPUProfile()
2013-09-28 23:03:42 -07:00
log.Print("stopped cpu profile")
2013-09-28 23:00:29 -07:00
}
if c.Memprofile != "" {
log.Print("trying to dump memory profile")
f, err := os.Create(c.Memprofile)
if err != nil {
log.Fatal(err)
}
pprof.WriteHeapProfile(f)
f.Close()
log.Print("stopped memory profile dump")
}
2013-09-28 23:03:42 -07:00
log.Fatal("shit got fucked up")
2013-09-28 23:00:29 -07:00
}
2013-11-07 22:05:20 -08:00
2014-04-14 00:26:41 -07:00
// getGameId trims the gameid off of the url. This is hokey, and makes me miss
// django regex-specified routes.
func (c *Controller) getGameId(path string) (string, error) {
2013-11-13 23:45:02 -08:00
var err error
trimmed := strings.Trim(path, "/")
fullPath := strings.Split(trimmed, "/")
if len(fullPath) != 3 {
return "", errors.New("improperly formed url")
}
key := fullPath[2]
return key, err
}
2014-03-29 01:57:29 -07:00
// MapLock is simply a map and a RWMutex
2014-04-14 00:26:41 -07:00
// TODO: obviate the need for this in Controller.Run
2014-03-29 01:57:29 -07:00
type MapLock struct {
M map[string]*Game
sync.RWMutex
}
// get is a function that returns a game if found, and creates one if
// not found and force is true. In order to get a hash (rather than use
// the string you pass) send "" for id.
func (ml *MapLock) Get(id string) *Game {
2014-03-29 01:57:29 -07:00
ml.Lock()
g, _ := ml.M[id]
ml.Unlock()
return g
}
// add is used to insert a new game into this MapLock
func (ml *MapLock) Add(g *Game) {
2014-03-29 01:57:29 -07:00
ml.Lock()
ml.M[g.id] = g
ml.Unlock()
}