84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
)
|
|
|
|
type PlayerService struct {
|
|
db *DB
|
|
}
|
|
type PlayerServiceQuery struct {
|
|
Name *string `db:"name"`
|
|
ID *int64 `db:"id"`
|
|
}
|
|
|
|
func (ps *PlayerService) Query(query PlayerServiceQuery) ([]Player, error) {
|
|
res := []Player{}
|
|
sqlQuery := "SELECT id, name, guild FROM player"
|
|
whereQuery := BuildWhereQuery(query)
|
|
if whereQuery != "" {
|
|
sqlQuery += " " + whereQuery
|
|
}
|
|
|
|
rows, err := ps.db.readConn.Query(sqlQuery)
|
|
if err != nil {
|
|
return res, fmt.Errorf("failed getting players for query %q: %v", sqlQuery, err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
player := Player{}
|
|
err := rows.Scan(&player.ID, &player.Name, &player.Guild.ID)
|
|
if err != nil {
|
|
return res, fmt.Errorf("failed scanning player: %v", err)
|
|
}
|
|
res = append(res, player)
|
|
}
|
|
|
|
return res, nil
|
|
}
|
|
|
|
func (ps *PlayerService) Create(name string, guild Guild) (Player, error) {
|
|
log.Printf("Creating player %s in guild %s", name, guild.Name)
|
|
player := Player{}
|
|
res, err := ps.db.writeConn.Exec("insert into player (name, guild) values (?, ?)", name, guild.ID)
|
|
if err != nil {
|
|
return player, fmt.Errorf("failed to insert player: %v", err)
|
|
}
|
|
|
|
id, err := res.LastInsertId()
|
|
if err != nil {
|
|
return player, fmt.Errorf("failed to get last insert id: %v", err)
|
|
}
|
|
log.Printf("Created player %s in guild %s with id %d", name, guild.Name, id)
|
|
|
|
qres, err := ps.Query(PlayerServiceQuery{ID: &id})
|
|
if err != nil {
|
|
return player, fmt.Errorf("failed getting player for id %d: %v", id, err)
|
|
}
|
|
if len(qres) > 1 {
|
|
return player, fmt.Errorf("more than one player found for id %d", id)
|
|
}
|
|
|
|
return qres[0], nil
|
|
}
|
|
|
|
func (ps *PlayerService) GetOrCreate(name string, guild Guild) (Player, error) {
|
|
player := Player{}
|
|
players, err := ps.Query(PlayerServiceQuery{Name: &name})
|
|
if len(players) > 1 {
|
|
return player, fmt.Errorf("more than one player found for name %s", name)
|
|
}
|
|
if err != nil || len(players) == 0 {
|
|
player, err = ps.Create(name, guild)
|
|
if err != nil {
|
|
return player, fmt.Errorf("failed creating player: %v", err)
|
|
}
|
|
}
|
|
if len(players) == 1 {
|
|
player = players[0]
|
|
}
|
|
return player, nil
|
|
}
|