Feat: settings repo

This commit is contained in:
Grail Finder
2025-07-04 13:32:59 +03:00
parent 058d501774
commit 6be365473c
9 changed files with 72 additions and 19 deletions

View File

@ -65,6 +65,9 @@ func (cm *CronManager) CleanupRooms() {
if err := cm.repo.RoomDeleteByID(ctx, room.ID); err != nil { if err := cm.repo.RoomDeleteByID(ctx, room.ID); err != nil {
cm.log.Error("failed to delete empty room", "room_id", room.ID, "err", err) cm.log.Error("failed to delete empty room", "room_id", room.ID, "err", err)
} }
if err := cm.repo.SettingsDeleteByRoomID(ctx, room.ID); err != nil {
cm.log.Error("failed to delete settings for empty room", "room_id", room.ID, "err", err)
}
continue continue
} }
@ -92,6 +95,9 @@ func (cm *CronManager) CleanupRooms() {
if err := cm.repo.RoomDeleteByID(ctx, room.ID); err != nil { if err := cm.repo.RoomDeleteByID(ctx, room.ID); err != nil {
cm.log.Error("failed to delete room after creator left", "room_id", room.ID, "err", err) cm.log.Error("failed to delete room after creator left", "room_id", room.ID, "err", err)
} }
if err := cm.repo.SettingsDeleteByRoomID(ctx, room.ID); err != nil {
cm.log.Error("failed to delete settings after creator left", "room_id", room.ID, "err", err)
}
} }
} }

View File

@ -200,7 +200,7 @@ func HandleStartGame(w http.ResponseWriter, r *http.Request) {
// Save action history // Save action history
action.RoomID = fi.Room.ID action.RoomID = fi.Room.ID
action.CreatedAt = time.Now() action.CreatedAt = time.Now()
if err := repo.CreateAction(ctx, fi.Room.ID, &action); err != nil { if err := repo.CreateAction(ctx, &action); err != nil {
if err := tx.Rollback(); err != nil { if err := tx.Rollback(); err != nil {
log.Error("failed to rollback transaction", "error", err) log.Error("failed to rollback transaction", "error", err)
} }

View File

@ -28,19 +28,25 @@ func StartTurnTimer(roomID string, duration time.Duration) {
done := make(chan bool) done := make(chan bool)
timers[roomID] = &roomTimer{ticker: ticker, done: done} timers[roomID] = &roomTimer{ticker: ticker, done: done}
go func() { go func() {
settings, err := repo.SettingsGetByRoomID(context.Background(), roomID)
if err != nil {
log.Error("failed to get settings by room id", "error", err)
StopTurnTimer(roomID)
return
}
for { for {
select { select {
case <-done: case <-done:
return return
case <-ticker.C: case <-ticker.C:
room, err := repo.RoomGetByID(context.Background(), roomID) if settings.TurnSecondsLeft <= 0 {
if err != nil {
log.Error("failed to get room by id", "error", err)
StopTurnTimer(roomID)
return
}
if room.Settings.TurnSecondsLeft <= 0 {
log.Info("turn time is over", "room_id", roomID) log.Info("turn time is over", "room_id", roomID)
room, err := repo.RoomGetByID(context.Background(), roomID)
if err != nil {
log.Error("failed to get room by id", "error", err)
StopTurnTimer(roomID)
return
}
room.ChangeTurn() room.ChangeTurn()
room.MimeDone = false room.MimeDone = false
if err := repo.RoomUpdate(context.Background(), room); err != nil { if err := repo.RoomUpdate(context.Background(), room); err != nil {
@ -51,11 +57,11 @@ func StartTurnTimer(roomID string, duration time.Duration) {
StopTurnTimer(roomID) StopTurnTimer(roomID)
return return
} }
room.Settings.TurnSecondsLeft-- settings.TurnSecondsLeft--
// if err := saveRoom(room); err != nil { if err := repo.SettingsUpdate(context.Background(), settings); err != nil {
// log.Error("failed to save room", "error", err) log.Error("failed to update settings", "error", err)
// } }
notify(models.NotifyRoomUpdatePrefix+room.ID, "") notify(models.NotifyRoomUpdatePrefix+roomID, "")
} }
} }
}() }()

View File

@ -65,6 +65,7 @@ CREATE TABLE settings (
language TEXT NOT NULL DEFAULT 'en', language TEXT NOT NULL DEFAULT 'en',
room_pass TEXT NOT NULL DEFAULT '', room_pass TEXT NOT NULL DEFAULT '',
turn_time INTEGER NOT NULL DEFAULT 60, -- seconds turn_time INTEGER NOT NULL DEFAULT 60, -- seconds
turn_seconds_left INTEGER NOT NULL DEFAULT 0,
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
FOREIGN KEY (room_id) REFERENCES rooms(id) ON DELETE CASCADE FOREIGN KEY (room_id) REFERENCES rooms(id) ON DELETE CASCADE
); );

View File

@ -390,7 +390,7 @@ type GameSettings struct {
RoomID string `db:"room_id"` RoomID string `db:"room_id"`
Language string `json:"language" example:"en" form:"language" db:"language"` Language string `json:"language" example:"en" form:"language" db:"language"`
RoomPass string `json:"room_pass" db:"room_pass"` RoomPass string `json:"room_pass" db:"room_pass"`
TurnSecondsLeft uint32 `db:"-"` TurnSecondsLeft uint32 `db:"turn_seconds_left"`
RoundTime uint32 `json:"round_time" db:"turn_time"` RoundTime uint32 `json:"round_time" db:"turn_time"`
CreatedAt time.Time `db:"created_at"` CreatedAt time.Time `db:"created_at"`
} }

View File

@ -50,9 +50,10 @@ func TestActionsRepo_CreateAction(t *testing.T) {
WordColor: "red", WordColor: "red",
Number: "3", Number: "3",
CreatedAt: time.Now(), CreatedAt: time.Now(),
RoomID: roomID,
} }
err := repo.CreateAction(context.Background(), roomID, action) err := repo.CreateAction(context.Background(), action)
assert.NoError(t, err) assert.NoError(t, err)
var retrievedAction models.Action var retrievedAction models.Action
@ -75,6 +76,7 @@ func TestActionsRepo_ListActions(t *testing.T) {
Word: "apple", Word: "apple",
WordColor: "red", WordColor: "red",
Number: "3", Number: "3",
RoomID: roomID,
CreatedAt: time.Now().Add(-2 * time.Second), CreatedAt: time.Now().Add(-2 * time.Second),
} }
action2 := &models.Action{ action2 := &models.Action{
@ -84,6 +86,7 @@ func TestActionsRepo_ListActions(t *testing.T) {
Word: "banana", Word: "banana",
WordColor: "blue", WordColor: "blue",
Number: "0", Number: "0",
RoomID: roomID,
CreatedAt: time.Now().Add(-1 * time.Second), CreatedAt: time.Now().Add(-1 * time.Second),
} }

View File

@ -17,6 +17,7 @@ type AllRepos interface {
PlayersRepo PlayersRepo
SessionsRepo SessionsRepo
WordCardsRepo WordCardsRepo
SettingsRepo
InitTx(ctx context.Context) (context.Context, *sqlx.Tx, error) InitTx(ctx context.Context) (context.Context, *sqlx.Tx, error)
} }
@ -32,6 +33,11 @@ func NewRepoProvider(pathToDB string) *RepoProvider {
slog.Error("Unable to connect to database", "error", err) slog.Error("Unable to connect to database", "error", err)
os.Exit(1) os.Exit(1)
} }
_, err = db.Exec("PRAGMA foreign_keys = ON;")
if err != nil {
slog.Error("Unable to enable foreign keys", "error", err)
os.Exit(1)
}
slog.Info("Successfully connected to database") slog.Info("Successfully connected to database")
rp := &RepoProvider{ rp := &RepoProvider{
DB: db, DB: db,

View File

@ -60,10 +60,6 @@ func (p *RepoProvider) RoomDeleteByID(ctx context.Context, id string) error {
func (p *RepoProvider) RoomUpdate(ctx context.Context, r *models.Room) error { func (p *RepoProvider) RoomUpdate(ctx context.Context, r *models.Room) error {
db := getDB(ctx, p.DB) db := getDB(ctx, p.DB)
_, err := db.ExecContext(ctx, `UPDATE rooms SET team_turn = ?, this_turn_limit = ?, opened_this_turn = ?, blue_counter = ?, red_counter = ?, red_turn = ?, mime_done = ?, is_running = ?, is_over = ?, team_won = ?, room_link = ? WHERE id = ?`, r.TeamTurn, r.ThisTurnLimit, r.OpenedThisTurn, r.BlueCounter, r.RedCounter, r.RedTurn, r.MimeDone, r.IsRunning, r.IsOver, r.TeamWon, r.RoomLink, r.ID) _, err := db.ExecContext(ctx, `UPDATE rooms SET team_turn = ?, this_turn_limit = ?, opened_this_turn = ?, blue_counter = ?, red_counter = ?, red_turn = ?, mime_done = ?, is_running = ?, is_over = ?, team_won = ?, room_link = ? WHERE id = ?`, r.TeamTurn, r.ThisTurnLimit, r.OpenedThisTurn, r.BlueCounter, r.RedCounter, r.RedTurn, r.MimeDone, r.IsRunning, r.IsOver, r.TeamWon, r.RoomLink, r.ID)
if err != nil {
return err
}
_, err = db.ExecContext(ctx, `UPDATE settings SET language = ?, room_pass = ?, turn_time = ? WHERE room_id = ?`, r.Settings.Language, r.Settings.RoomPass, r.Settings.RoundTime, r.ID)
return err return err
} }

35
repos/settings.go Normal file
View File

@ -0,0 +1,35 @@
package repos
import (
"context"
"gralias/models"
"github.com/jmoiron/sqlx"
)
type SettingsRepo interface {
SettingsGetByRoomID(ctx context.Context, roomID string) (*models.GameSettings, error)
SettingsUpdate(ctx context.Context, settings *models.GameSettings) error
SettingsDeleteByRoomID(ctx context.Context, roomID string) error
}
func (p *RepoProvider) SettingsGetByRoomID(ctx context.Context, roomID string) (*models.GameSettings, error) {
settings := &models.GameSettings{}
err := sqlx.GetContext(ctx, p.DB, settings, `SELECT * FROM settings WHERE room_id = ?`, roomID)
if err != nil {
return nil, err
}
return settings, nil
}
func (p *RepoProvider) SettingsUpdate(ctx context.Context, s *models.GameSettings) error {
db := getDB(ctx, p.DB)
_, err := db.ExecContext(ctx, `UPDATE settings SET language = ?, room_pass = ?, turn_time = ?, turn_seconds_left = ? WHERE room_id = ?`, s.Language, s.RoomPass, s.RoundTime, s.TurnSecondsLeft, s.RoomID)
return err
}
func (p *RepoProvider) SettingsDeleteByRoomID(ctx context.Context, roomID string) error {
db := getDB(ctx, p.DB)
_, err := db.ExecContext(ctx, `DELETE FROM settings WHERE room_id = ?`, roomID)
return err
}