server/botserv/main.go

69 lines
1.8 KiB
Go
Raw Normal View History

2013-08-19 20:43:26 -07:00
package main
import (
"flag"
"log"
"math/rand"
"net/http"
_ "net/http/pprof"
2013-09-28 23:00:29 -07:00
"os"
"runtime/pprof"
2013-08-19 20:43:26 -07:00
"time"
2013-11-29 00:10:49 -08:00
2014-04-23 14:54:39 -07:00
"bitbucket.org/hackerbots/server"
"golang.org/x/net/websocket"
2013-08-19 20:43:26 -07:00
)
var addr = flag.String("addr", ":8666", "http service address")
2013-09-28 23:00:29 -07:00
var profile = flag.String("pprof", "", "if specified will run with pprof")
var mprofile = flag.String("mprof", "", "if specified will dump a memory profile")
var netprofile = flag.Bool("netprof", false, "if specified will run with net/http/pprof on :8667")
2013-11-13 20:38:57 -08:00
var verbose = flag.Bool("verbose", false, "")
var config = flag.String("config", "~/.config/hackerbots/server.json", "location of config file")
2013-08-19 20:43:26 -07:00
func main() {
2014-01-13 20:54:01 -08:00
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
log.Printf("Starting Server...")
2013-11-13 20:38:57 -08:00
var err error
2013-08-19 20:43:26 -07:00
rand.Seed(time.Now().UnixNano())
flag.Parse()
2013-09-28 23:00:29 -07:00
if *profile != "" {
f, err := os.Create(*profile)
if err != nil {
log.Fatal(err)
}
pprof.StartCPUProfile(f)
}
if *netprofile {
go func() {
log.Println(http.ListenAndServe("localhost:8667", nil))
}()
log.Println("serving profile info at http://localhost:8667/debug/pprof/")
}
2013-08-19 20:43:26 -07:00
2014-04-23 14:54:39 -07:00
conf, err := server.LoadConfig(*config)
2013-11-13 20:38:57 -08:00
if err != nil {
log.Fatal(err)
}
sm := http.NewServeMux()
2014-04-23 14:54:39 -07:00
c := server.NewController(conf, *mprofile, *profile)
go c.Run()
2014-04-23 14:54:39 -07:00
sm.Handle("/", server.JsonHandler(c.Index))
sm.Handle("/ws/", websocket.Handler(c.AddPlayer))
2014-04-23 14:54:39 -07:00
sm.Handle("/game/start/", server.JsonHandler(c.StartGame))
sm.Handle("/game/list/", server.JsonHandler(c.ListGames))
sm.Handle("/game/stats/", server.JsonHandler(c.GameStats))
sm.Handle("/game/bw/", server.JsonHandler(c.BW))
sm.Handle("/game/stop/", server.JsonHandler(c.StopGame))
sm.HandleFunc("/fuck/shit/up/", c.KillServer)
2013-08-19 20:43:26 -07:00
2013-11-13 20:38:57 -08:00
err = http.ListenAndServe(*addr, sm)
2013-08-19 20:43:26 -07:00
if err != nil {
log.Fatal("unable to start server")
}
}