88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package postgres
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.snrd.de/Spaenny/steamServer"
|
|
"github.com/google/uuid"
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type ServerStore struct {
|
|
*sqlx.DB
|
|
}
|
|
|
|
func (s *ServerStore) Server(id uuid.UUID) (steamServer.Server, error) {
|
|
var ser steamServer.Server
|
|
if err := s.Get(&ser, `SELECT * FROM servers WHERE id = $1`, id); err != nil {
|
|
return steamServer.Server{}, fmt.Errorf("error getting server: %w", err)
|
|
}
|
|
return ser, nil
|
|
}
|
|
|
|
func (s *ServerStore) Servers() ([]steamServer.Server, error) {
|
|
var ss []steamServer.Server
|
|
var query = `SELECT * FROM servers`
|
|
if err := s.Select(&ss, query); err != nil {
|
|
return []steamServer.Server{}, fmt.Errorf("error gettings servers: %w", err)
|
|
}
|
|
return ss, nil
|
|
}
|
|
|
|
func (s *ServerStore) ServersByRegion(regionID uuid.UUID) ([]steamServer.Server, error) {
|
|
var ss []steamServer.Server
|
|
var query = `
|
|
SELECT * FROM servers WHERE region_id = $1`
|
|
if err := s.Select(&ss, query, regionID); err != nil {
|
|
return []steamServer.Server{}, fmt.Errorf("error gettings region servers: %w", err)
|
|
}
|
|
return ss, nil
|
|
}
|
|
|
|
func (s *ServerStore) ServersByRegionAndGamemode(regionID uuid.UUID, gamemodeID uuid.UUID) ([]steamServer.Server, error) {
|
|
var ss []steamServer.Server
|
|
var query = `SELECT * FROM servers WHERE gamemode_id = $1 AND region_id = $2`
|
|
if err := s.Select(&ss, query, regionID, gamemodeID); err != nil {
|
|
return []steamServer.Server{}, fmt.Errorf("error gettings region servers: %w", err)
|
|
}
|
|
return ss, nil
|
|
}
|
|
|
|
func (s *ServerStore) ServersByGamemode(gamemodeID uuid.UUID) ([]steamServer.Server, error) {
|
|
var ss []steamServer.Server
|
|
|
|
var query = `SELECT * FROM servers WHERE gamemode_id = $1`
|
|
|
|
if err := s.Select(&ss, query, gamemodeID); err != nil {
|
|
return []steamServer.Server{}, fmt.Errorf("error gettings gamemode servers: %w", err)
|
|
}
|
|
return ss, nil
|
|
}
|
|
|
|
func (s *ServerStore) CreateServer(ser *steamServer.Server) error {
|
|
if err := s.Get(ser, `INSERT INTO servers VALUES($1, $2, $3, $4, $5, $6, $7, $8, $9) RETURNUNG *`,
|
|
ser.ID,
|
|
ser.RegionID,
|
|
ser.GamemodeID,
|
|
ser.AppID,
|
|
ser.Name,
|
|
ser.ServerIP,
|
|
ser.Map,
|
|
ser.Players,
|
|
ser.MaxPlayers); err != nil {
|
|
return fmt.Errorf("error creating server: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerStore) UpdateServerByQuery(ser *steamServer.Server, server_ip string) error {
|
|
if err := s.Get(s, `UPDATE "servers" SET "name" = $1, "map" = $2, "players" = $3, "max_players" = $4 WHERE "server_ip" = $5 RETURNING *`,
|
|
ser.Name,
|
|
ser.Map,
|
|
ser.Players,
|
|
ser.MaxPlayers,
|
|
server_ip); err != nil {
|
|
return fmt.Errorf("error updating server: %w", err)
|
|
}
|
|
return nil
|
|
}
|