Feat: add player interface

This commit is contained in:
Grail Finder
2025-07-01 13:38:20 +03:00
parent 83d3a19d05
commit 70f83f1002
3 changed files with 70 additions and 7 deletions

View File

@ -1,11 +1,31 @@
package repos
import "github.com/jackc/pgx/v5/pgxpool"
import (
"context"
"log/slog"
"os"
"github.com/jackc/pgx/v5/pgxpool"
)
type AllRepos interface {
RoomsRepo
ActionsRepo
PlayersRepo
}
type RepoProvider struct {
DB *pgxpool.Pool
}
func NewRepoProvider(pathToDB string) *RepoProvider {
dbpool, err := pgxpool.New(context.Background(), pathToDB)
if err != nil {
slog.Error("Unable to connect to database", "error", err)
os.Exit(1)
}
slog.Info("Successfully connected to database")
return &RepoProvider{
DB: dbpool,
}
}

31
repos/players.go Normal file
View File

@ -0,0 +1,31 @@
package repos
import (
"context"
"gralias/models"
)
type PlayersRepo interface {
GetPlayer(roomID, username string) (*models.Player, error)
AddPlayer(player *models.Player) error
DeletePlayer(roomID, username string) error
}
func (p *RepoProvider) GetPlayer(roomID, username string) (*models.Player, error) {
var player models.Player
err := p.DB.QueryRow(context.Background(), "SELECT id, room_id, username, team, role, is_bot FROM players WHERE room_id = $1 AND username = $2", roomID, username).Scan(&player.ID, &player.RoomID, &player.Username, &player.Team, &player.Role, &player.IsBot)
if err != nil {
return nil, err
}
return &player, nil
}
func (p *RepoProvider) AddPlayer(player *models.Player) error {
_, err := p.DB.Exec(context.Background(), "INSERT INTO players (room_id, username, team, role, is_bot) VALUES ($1, $2, $3, $4, $5)", player.RoomID, player.Username, player.Team, player.Role, player.IsBot)
return err
}
func (p *RepoProvider) DeletePlayer(roomID, username string) error {
_, err := p.DB.Exec(context.Background(), "DELETE FROM players WHERE room_id = $1 AND username = $2", roomID, username)
return err
}