// Copyright 2014, 2015 by Sascha L. Teichmann // Use of this source code is governed by the MIT license // that can be found in the LICENSE file. package main import ( "bufio" "bytes" "encoding/json" "html/template" "log" "math" "net/http" "os" "sort" "sync" "time" ) const sleepInterval = time.Second * 5 var geoJSONTmpl = template.Must(template.New("geojson").Parse( `{ "type": "Feature", "geometry": { "type": "Point", "coordinates": [{{.Z}}, {{.X}}] }, "properties": { "name": "{{.Name | html }}" } }`)) type player struct { X float64 `json:"x"` Y float64 `json:"y"` Z float64 `json:"z"` Name string `json:"name"` } type players struct { fifo string wsf *websocketForwarder pls []*player mu sync.RWMutex } func newPlayers(fifo string, wsf *websocketForwarder) *players { return &players{fifo: fifo, wsf: wsf, pls: []*player{}} } func (p *player) MarshalJSON() ([]byte, error) { var buf bytes.Buffer if err := geoJSONTmpl.Execute(&buf, p); err != nil { return nil, err } return buf.Bytes(), nil } func (p *player) same(o *player) bool { return p.Name == o.Name && math.Abs(p.X-o.X) < 0.000001 && math.Abs(p.Y-o.Y) < 0.000001 && math.Abs(p.Z-o.Z) < 0.000001 } type sortPlayersByName []*player func (pls sortPlayersByName) Len() int { return len(pls) } func (pls sortPlayersByName) Less(i, j int) bool { return pls[i].Name < pls[j].Name } func (pls sortPlayersByName) Swap(i, j int) { pls[i], pls[j] = pls[j], pls[i] } func (ps *players) readFromFIFO() ([]*player, error) { file, err := os.Open(ps.fifo) if err != nil { return nil, err } defer file.Close() reader := bufio.NewReader(file) decoder := json.NewDecoder(reader) var pls []*player if err = decoder.Decode(&pls); err != nil { return nil, err } return pls, nil } func samePlayers(a, b []*player) bool { if len(a) != len(b) { return false } for i, p := range a { if !p.same(b[i]) { return false } } return true } func (ps *players) run() { for { pls, err := ps.readFromFIFO() if err != nil { //log.Printf("err: %s\n", err) time.Sleep(sleepInterval) continue } if pls == nil { //log.Println("no players") continue } //log.Printf("%+q\n", pls) sort.Sort(sortPlayersByName(pls)) var change bool ps.mu.Lock() //log.Printf("%+q\n", pls) //log.Printf("%+q\n", ps.pls) if change = !samePlayers(pls, ps.pls); change { ps.pls = pls } ps.mu.Unlock() if change && ps.wsf != nil { // TODO: Throttle this! ps.wsf.BroadcastPlayers(pls) } } } func (ps *players) ServeHTTP(rw http.ResponseWriter, r *http.Request) { rw.Header().Set("Content-Type", "application/json") var pls []*player ps.mu.RLock() pls = ps.pls ps.mu.RUnlock() encoder := json.NewEncoder(rw) if err := encoder.Encode(pls); err != nil { log.Printf("error: sending JSON failed: %s\n", err) } }