package main import "fmt" 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) { 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) } 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 err != nil { if len(players) > 1 { return player, fmt.Errorf("more than one player found for name %s", name) } player, err = ps.Create(name, guild) if err != nil { return player, fmt.Errorf("failed creating player: %v", err) } } return player, nil }