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: nilix
GPG key ID: 3B7FBC22144E6398
4 changed files with 140 additions and 121 deletions

View file

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

22
main.go
View file

@ -1,14 +1,14 @@
package main
import (
"context"
"nilfm.cc/git/felt/gametable"
"net"
"net/http"
"os"
"os/signal"
"time"
"log"
"context"
"log"
"net"
"net/http"
"nilfm.cc/git/felt/gametable"
"os"
"os/signal"
"time"
)
func main() {
@ -26,9 +26,9 @@ func run() error {
gt := gametable.New()
s := &http.Server{
Handler: gt,
ReadTimeout: time.Second * 10,
WriteTimeout: time.Second * 10,
Handler: gt,
ReadTimeout: time.Second * 10,
WriteTimeout: time.Second * 10,
}
errc := make(chan error, 1)

View file

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

View file

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