You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
2.0 KiB
Go
106 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/jordanorelli/blammo"
|
|
)
|
|
|
|
type player struct {
|
|
*blammo.Log
|
|
conn *websocket.Conn
|
|
server *server
|
|
id int
|
|
username string
|
|
outbox chan string
|
|
}
|
|
|
|
func (p *player) run() {
|
|
p.Info("starting run cycle")
|
|
defer p.Info("run cycle done")
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
go p.writeMessages(ctx)
|
|
p.readMessages(cancel)
|
|
p.server.leave <- p
|
|
p.outbox = nil
|
|
p.conn.Close()
|
|
}
|
|
|
|
func (p *player) readMessages(cancel func()) {
|
|
defer cancel()
|
|
for {
|
|
_, rd, err := p.conn.NextReader()
|
|
if err != nil {
|
|
if closeError, ok := err.(*websocket.CloseError); ok {
|
|
switch closeError.Code {
|
|
case websocket.CloseNormalClosure:
|
|
p.Info("client disconnected: %s", closeError.Text)
|
|
default:
|
|
p.Error("client disconnected weirdly: %#v", *closeError)
|
|
}
|
|
} else {
|
|
p.Error("nextreader error: %s", err)
|
|
}
|
|
return
|
|
}
|
|
|
|
b, err := ioutil.ReadAll(rd)
|
|
if err != nil {
|
|
p.Error("read error: %s", err)
|
|
} else {
|
|
p.Child("rcv").Info(string(b))
|
|
}
|
|
p.server.inbox <- message{
|
|
from: p,
|
|
text: string(b),
|
|
}
|
|
}
|
|
}
|
|
|
|
func (p *player) writeMessages(ctx context.Context) {
|
|
p.Info("writeMessages loop start")
|
|
defer p.Info("writeMessage loop end")
|
|
|
|
ticker := time.NewTicker(1 * time.Second)
|
|
n := 0
|
|
|
|
for {
|
|
select {
|
|
case t := <-ticker.C:
|
|
n++
|
|
w, err := p.conn.NextWriter(websocket.TextMessage)
|
|
if err != nil {
|
|
p.Error("error getting writer: %v", err)
|
|
return
|
|
}
|
|
|
|
fmt.Fprintf(w, "tick %d: %v", n, t)
|
|
if err := w.Close(); err != nil {
|
|
p.Error("close frame error: %v", err)
|
|
return
|
|
}
|
|
case msg := <-p.outbox:
|
|
p.Info("writing message from outbox: %s", msg)
|
|
w, err := p.conn.NextWriter(websocket.TextMessage)
|
|
if err != nil {
|
|
p.Error("error getting writer: %v", err)
|
|
return
|
|
}
|
|
|
|
w.Write([]byte(msg))
|
|
if err := w.Close(); err != nil {
|
|
p.Error("close frame error: %v", err)
|
|
return
|
|
}
|
|
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|