gofmt, rid code of typescript confusion, little bit of db code

This commit is contained in:
Iris Lightshard 2022-11-10 23:00:05 -07:00
parent 25e51fb2d5
commit e59f3e1609
Signed by: Iris Lightshard
GPG key ID: 3B7FBC22144E6398
4 changed files with 140 additions and 121 deletions

View file

@ -1,47 +1,47 @@
package gametable package gametable
import ( import (
"context" "context"
"nhooyr.io/websocket" "errors"
"golang.org/x/time/rate" "golang.org/x/time/rate"
"io/ioutil" "io/ioutil"
"time" "log"
"sync" "net/http"
"net/http" "nhooyr.io/websocket"
"log" "sync"
"errors" "time"
) )
type Subscriber struct { type Subscriber struct {
msgs chan []byte msgs chan []byte
closeSlow func() closeSlow func()
} }
type GameTableServer struct { type GameTableServer struct {
subscribeMessageBuffer int subscribeMessageBuffer int
publishLimiter *rate.Limiter publishLimiter *rate.Limiter
logf func(f string, v ...interface{}) logf func(f string, v ...interface{})
serveMux http.ServeMux serveMux http.ServeMux
subscribersLock sync.Mutex subscribersLock sync.Mutex
subscribers map[*Subscriber]struct{} subscribers map[*Subscriber]struct{}
} }
func New() *GameTableServer { func New() *GameTableServer {
srvr := &GameTableServer { srvr := &GameTableServer{
subscribeMessageBuffer: 16, subscribeMessageBuffer: 16,
logf: log.Printf, logf: log.Printf,
subscribers: make(map[*Subscriber]struct{}), subscribers: make(map[*Subscriber]struct{}),
publishLimiter: rate.NewLimiter(rate.Every(time.Millisecond*100), 8), publishLimiter: rate.NewLimiter(rate.Every(time.Millisecond*100), 8),
} }
srvr.serveMux.Handle("/", http.FileServer(http.Dir("./static"))) srvr.serveMux.Handle("/", http.FileServer(http.Dir("./static")))
srvr.serveMux.HandleFunc("/subscribe", srvr.subscribeHandler) srvr.serveMux.HandleFunc("/subscribe", srvr.subscribeHandler)
srvr.serveMux.HandleFunc("/publish", srvr.publishHandler) srvr.serveMux.HandleFunc("/publish", srvr.publishHandler)
return srvr return srvr
} }
func (self *GameTableServer) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (self *GameTableServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
self.serveMux.ServeHTTP(w, r) self.serveMux.ServeHTTP(w, r)
} }
func (self *GameTableServer) subscribeHandler(w http.ResponseWriter, r *http.Request) { func (self *GameTableServer) subscribeHandler(w http.ResponseWriter, r *http.Request) {
@ -112,7 +112,7 @@ func (self *GameTableServer) publish(msg []byte) {
self.subscribersLock.Lock() self.subscribersLock.Lock()
defer self.subscribersLock.Unlock() defer self.subscribersLock.Unlock()
// decode message and store in DB // decode message and store in DB
self.publishLimiter.Wait(context.Background()) self.publishLimiter.Wait(context.Background())
@ -142,4 +142,4 @@ func writeTimeout(ctx context.Context, timeout time.Duration, c *websocket.Conn,
defer cancel() defer cancel()
return c.Write(ctx, websocket.MessageText, msg) return c.Write(ctx, websocket.MessageText, msg)
} }

28
main.go
View file

@ -1,14 +1,14 @@
package main package main
import ( import (
"context" "context"
"nilfm.cc/git/felt/gametable" "log"
"net" "net"
"net/http" "net/http"
"os" "nilfm.cc/git/felt/gametable"
"os/signal" "os"
"time" "os/signal"
"log" "time"
) )
func main() { func main() {
@ -23,14 +23,14 @@ func run() error {
if err != nil { if err != nil {
return err return err
} }
gt := gametable.New() gt := gametable.New()
s := &http.Server{ s := &http.Server{
Handler: gt, Handler: gt,
ReadTimeout: time.Second * 10, ReadTimeout: time.Second * 10,
WriteTimeout: time.Second * 10, WriteTimeout: time.Second * 10,
} }
errc := make(chan error, 1) errc := make(chan error, 1)
go func() { go func() {
errc <- s.Serve(l) errc <- s.Serve(l)
@ -49,4 +49,4 @@ func run() error {
defer cancel() defer cancel()
return s.Shutdown(ctx) return s.Shutdown(ctx)
} }

View file

@ -1,25 +1,25 @@
package models package models
import ( import (
"time" "time"
) )
type TableKey struct { type TableKey struct {
name: string Name string
passcode: string Passcode string
} }
type DiceRoll struct { type DiceRoll struct {
faces: uint8 Faces uint8
roll: uint8[] Roll []uint8
player: string Player string
note: string Note string
timestamp: time.Time Timestamp time.Time
} }
type Token struct { type Token struct {
id: string Id string
spriteUri: string SpriteUri string
x: int X int
y: int Y int
} }

View file

@ -1,76 +1,95 @@
package dbengine package dbengine
import ( import (
"context" "context"
"time" "go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/mongo/options" "nilfm.cc/git/felt/models"
"nilfm.cc/git/felt/models" "time"
) )
interface DbAdapter { type DbAdapter interface {
Init(mongoUri: string): error Init(mongoUri string) error
CreateTable(table: models.TableKey): error CreateTable(table models.TableKey) error
DestroyTable(table: models.TableKey): error DestroyTable(table models.TableKey) error
InsertDiceRoll(table: models.TableKey, diceRoll: models.DiceRoll): error InsertDiceRoll(table models.TableKey, diceRoll models.DiceRoll) error
GetDiceRolls(table: models.TableKey): models.DiceRoll[], error GetDiceRolls(table models.TableKey) ([]models.DiceRoll, error)
SetMapImageUrl(table: models.TableKey, url: string): error SetMapImageUrl(table models.TableKey, url string) error
GetMapImageUrl(table: models.TableKey): string, error GetMapImageUrl(table models.TableKey) (string, error)
AddToken(table: models.TableKey, token: models.Token): error AddToken(table models.TableKey, token models.Token) error
RemoveToken(table: models.TableKey, tokenId: string): error RemoveToken(table models.TableKey, tokenId string) error
ModifyToken(table: models.TableKey, token: models.Token): error ModifyToken(table models.TableKey, token models.Token) error
GetTokens(table: models.TableKey): models.Token[], error GetTokens(table models.TableKey) ([]models.Token, error)
} }
type DbEngine struct { type DbEngine struct {
client: mongo.Client client mongo.Client
} }
func (self *DbEngine) Init(mongoUri: string) error { func (self *DbEngine) mkCtx(timeoutSec int) context.Context {
client, err := mongo.NewClient(options.Client().ApplyURI(mongoUri)) return context.WithTimeout(context.Background(), 10*time.Second)
if err != nil {
return err
}
self.client = client
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
err = client.Connect(ctx)
if err != nil {
return err
}
defer client.Disconnect(ctx)
db := client.Database("felt")
err = self.ensureCollections(db)
return err
} }
func (self *DbEngine) ensureCollections(db: mongo.Database) error { func (self *DbEngine) Init(mongoUri string) error {
tables := db.Collection("tables") client, err := mongo.NewClient(options.Client().ApplyURI(mongoUri))
if tables == nil { if err != nil {
createCmd := bson.D{ return err
{"create", "tables"}, }
{"clusteredIndex", { self.client = client
{"key", {"name"}}, ctx, _ := self.mkCtx(10)
{"unique", true},
{"name", "idx_tables_unique_names"} err = client.Connect(ctx)
}} if err != nil {
} return err
}
var createResult bson.M defer client.Disconnect(ctx)
err := db.RunCommand(
context.WithTimeout(context.Background(), 10*time.Second), db := client.Database("felt")
createCmd).Decode(&createResult)
err = self.ensureCollections(db)
if err != nil { return err
return err }
}
} func (self *DbEngine) ensureCollections(db mongo.Database) error {
return nil tables := db.Collection("tables")
if tables == nil {
createCmd := bson.D{
{"create", "tables"},
{"clusteredIndex", {
{"key", {"name"}},
{"unique", true},
{"name", "idx_tables_unique_names"},
}},
}
var createResult bson.M
err := db.RunCommand(
self.mkCtx(10),
createCmd).Decode(&createResult)
if err != nil {
return err
}
}
return nil
}
func (self *DbEngine) CreateTable(table models.TableKey) error {
tables := self.db.Collection("tables")
if tables != nil {
_, err := tables.insertOne(self.mkCtx(10), bson.D{
{"name", table.Name},
{"passcode", table.Passcode},
{"mapUri", ""},
{"diceRolls", bson.A{}},
{"tokens", bson.A{}},
})
}
return err
} }