Feat: nullable roomid

This commit is contained in:
Grail Finder
2025-07-03 12:29:39 +03:00
parent 2a593739ae
commit d8338fe382
6 changed files with 25 additions and 14 deletions

View File

@ -300,11 +300,14 @@ func recoverBots() {
func recoverBot(bm models.Player) error { func recoverBot(bm models.Player) error {
// check if room still exists // check if room still exists
if _, err := repo.RoomGetByID(context.Background(), bm.RoomID); err != nil { if bm.RoomID == nil {
return fmt.Errorf("no such room: %s; err: %w", bm.RoomID, err) return errors.New("bot has no room id")
}
if _, err := repo.RoomGetByID(context.Background(), *bm.RoomID); err != nil {
return fmt.Errorf("no such room: %s; err: %w", *bm.RoomID, err)
} }
log.Debug("recovering bot", "bot", bm) log.Debug("recovering bot", "bot", bm)
_, err := llmapi.NewBot(string(bm.Role), string(bm.Team), bm.Username, bm.RoomID, cfg, true) _, err := llmapi.NewBot(string(bm.Role), string(bm.Team), bm.Username, *bm.RoomID, cfg, true)
if err != nil { if err != nil {
return err return err
} }

View File

@ -110,8 +110,11 @@ func HandleFrontLogin(w http.ResponseWriter, r *http.Request) {
// room.PlayerList = append(room.PlayerList, fi.State.Username) // room.PlayerList = append(room.PlayerList, fi.State.Username)
fi.Room = room fi.Room = room
fi.List = nil fi.List = nil
fi.State.RoomID = room.ID fi.State.RoomID = &room.ID
repo.PlayerSetRoomID(r.Context(), fi.State.Username, room.ID) if err := repo.PlayerSetRoomID(r.Context(), fi.State.Username, room.ID); err != nil {
abortWithError(w, err.Error())
return
}
// repo.RoomUpdate() // repo.RoomUpdate()
// save full info instead // save full info instead
// if err := saveFullInfo(r.Context(), fi); err != nil { // if err := saveFullInfo(r.Context(), fi); err != nil {
@ -191,6 +194,7 @@ func makeCookie(username string, remote string) (*http.Cookie, error) {
return cookie, nil return cookie, nil
} }
//nolint: unused
func cacheGetSession(key string) (*models.Session, error) { func cacheGetSession(key string) (*models.Session, error) {
userSessionB, err := cache.MemCache.Get(key) userSessionB, err := cache.MemCache.Get(key)
if err != nil { if err != nil {

View File

@ -38,7 +38,7 @@ func HandleCreateRoom(w http.ResponseWriter, r *http.Request) {
abortWithError(w, msg) abortWithError(w, msg)
return return
} }
fi.State.RoomID = room.ID fi.State.RoomID = &room.ID
fi.Room = room fi.Room = room
if err := repo.RoomCreate(r.Context(), room); err != nil { if err := repo.RoomCreate(r.Context(), room); err != nil {
log.Error("failed to create a room", "error", err) log.Error("failed to create a room", "error", err)
@ -223,7 +223,7 @@ func HandleJoinRoom(w http.ResponseWriter, r *http.Request) {
return return
} }
// room.PlayerList = append(room.PlayerList, fi.State.Username) // room.PlayerList = append(room.PlayerList, fi.State.Username)
fi.State.RoomID = room.ID fi.State.RoomID = &room.ID
fi.Room = room fi.Room = room
fi.List = nil fi.List = nil
if err := saveFullInfo(r.Context(), fi); err != nil { if err := saveFullInfo(r.Context(), fi); err != nil {

View File

@ -2,8 +2,8 @@ package handlers
import ( import (
"context" "context"
"fmt"
"gralias/models" "gralias/models"
"strconv"
"sync" "sync"
"time" "time"
) )
@ -46,7 +46,7 @@ func StartTurnTimer(roomID string, duration time.Duration) {
if err := repo.RoomUpdate(context.Background(), room); err != nil { if err := repo.RoomUpdate(context.Background(), room); err != nil {
log.Error("failed to save room", "error", err) log.Error("failed to save room", "error", err)
} }
notify(models.NotifyTurnTimerPrefix+room.ID, fmt.Sprintf("%d", room.Settings.TurnSecondsLeft)) notify(models.NotifyTurnTimerPrefix+room.ID, strconv.FormatUint(uint64(room.Settings.TurnSecondsLeft), 10))
notifyBotIfNeeded(room) notifyBotIfNeeded(room)
StopTurnTimer(roomID) StopTurnTimer(roomID)
return return

View File

@ -49,6 +49,7 @@ func convertToSliceOfStrings(value any) ([]string, error) {
} }
} }
//nolint: unused
func (b *Bot) checkGuesses(tempMap map[string]any, room *models.Room) error { func (b *Bot) checkGuesses(tempMap map[string]any, room *models.Room) error {
guesses, err := convertToSliceOfStrings(tempMap["guesses"]) guesses, err := convertToSliceOfStrings(tempMap["guesses"])
if err != nil { if err != nil {
@ -539,5 +540,5 @@ func (b *Bot) CallLLM(prompt string) ([]byte, error) {
return body, nil return body, nil
} }
// This line should not be reached because each error path returns in the loop. // This line should not be reached because each error path returns in the loop.
return nil, fmt.Errorf("unknown error in retry loop") return nil, errors.New("unknown error in retry loop")
} }

View File

@ -38,8 +38,9 @@ func TestPlayersRepo_AddPlayer(t *testing.T) {
repo := &RepoProvider{DB: db} repo := &RepoProvider{DB: db}
roomID := "test_room_player_1"
player := &models.Player{ player := &models.Player{
RoomID: "test_room_player_1", RoomID: &roomID,
Username: "test_player_1", Username: "test_player_1",
Team: "blue", Team: "blue",
Role: "player", Role: "player",
@ -61,8 +62,9 @@ func TestPlayersRepo_GetPlayer(t *testing.T) {
repo := &RepoProvider{DB: db} repo := &RepoProvider{DB: db}
roomID := "test_room_player_2"
player := &models.Player{ player := &models.Player{
RoomID: "test_room_player_2", RoomID: &roomID,
Username: "test_player_2", Username: "test_player_2",
Team: "red", Team: "red",
Role: "player", Role: "player",
@ -84,8 +86,9 @@ func TestPlayersRepo_DeletePlayer(t *testing.T) {
repo := &RepoProvider{DB: db} repo := &RepoProvider{DB: db}
roomID := "test_room_player_3"
player := &models.Player{ player := &models.Player{
RoomID: "test_room_player_3", RoomID: &roomID,
Username: "test_player_3", Username: "test_player_3",
Team: "blue", Team: "blue",
Role: "player", Role: "player",
@ -95,7 +98,7 @@ func TestPlayersRepo_DeletePlayer(t *testing.T) {
_, err := db.Exec(`INSERT INTO players (room_id, username, team, role, is_bot) VALUES (?, ?, ?, ?, ?)`, player.RoomID, player.Username, player.Team, player.Role, player.IsBot) _, err := db.Exec(`INSERT INTO players (room_id, username, team, role, is_bot) VALUES (?, ?, ?, ?, ?)`, player.RoomID, player.Username, player.Team, player.Role, player.IsBot)
assert.NoError(t, err) assert.NoError(t, err)
err = repo.PlayerDelete(context.Background(), player.RoomID, player.Username) err = repo.PlayerDelete(context.Background(), *player.RoomID, player.Username)
assert.NoError(t, err) assert.NoError(t, err)
var count int var count int