Merge branch 'main' into 'main'
add code See merge request spfn/match-result-collection-server!1
This commit is contained in:
commit
705eae5067
3 changed files with 176 additions and 0 deletions
13
go.mod
Normal file
13
go.mod
Normal file
|
|
@ -0,0 +1,13 @@
|
||||||
|
module goresults
|
||||||
|
|
||||||
|
go 1.25.0
|
||||||
|
|
||||||
|
require github.com/jackc/pgx/v5 v5.9.1
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/jackc/pgpassfile v1.0.0 // indirect
|
||||||
|
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
|
||||||
|
github.com/jackc/puddle/v2 v2.2.2 // indirect
|
||||||
|
golang.org/x/sync v0.17.0 // indirect
|
||||||
|
golang.org/x/text v0.29.0 // indirect
|
||||||
|
)
|
||||||
26
go.sum
Normal file
26
go.sum
Normal file
|
|
@ -0,0 +1,26 @@
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
|
||||||
|
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
|
||||||
|
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
|
||||||
|
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=
|
||||||
|
github.com/jackc/pgx/v5 v5.9.1 h1:uwrxJXBnx76nyISkhr33kQLlUqjv7et7b9FjCen/tdc=
|
||||||
|
github.com/jackc/pgx/v5 v5.9.1/go.mod h1:mal1tBGAFfLHvZzaYh77YS/eC6IX9OWbRV1QIIM0Jn4=
|
||||||
|
github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo=
|
||||||
|
github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
|
||||||
|
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
|
||||||
|
golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=
|
||||||
|
golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||||
|
golang.org/x/text v0.29.0 h1:1neNs90w9YzJ9BocxfsQNHKuAT4pkghyXc4nhZ6sJvk=
|
||||||
|
golang.org/x/text v0.29.0/go.mod h1:7MhJOA9CD2qZyOKYazxdYMF85OwPdEr9jTtBpO7ydH4=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
137
main.go
Normal file
137
main.go
Normal file
|
|
@ -0,0 +1,137 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/jackc/pgx/v5/pgxpool"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CompetitionPostResults struct {
|
||||||
|
SplatfestID uint32 `json:"splatfest_id"`
|
||||||
|
Score uint32 `json:"score"`
|
||||||
|
TeamID uint8 `json:"team_id"`
|
||||||
|
TeamWin uint8 `json:"team_win"`
|
||||||
|
User uint32 `json:"user"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var db *pgxpool.Pool
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
var err error
|
||||||
|
connStr := os.Getenv("DATABASE_URL")
|
||||||
|
db, err = pgxpool.New(context.Background(), connStr)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Unable to connect to database: %v\n", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
|
||||||
|
http.HandleFunc("/results/get", handleGetResults)
|
||||||
|
http.HandleFunc("/results/votes", handleGetVotes)
|
||||||
|
http.HandleFunc("/results/post", handlePostResults)
|
||||||
|
|
||||||
|
log.Println("Server starting on :8000...")
|
||||||
|
log.Fatal(http.ListenAndServe(":8000", nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
// GET /results/get?splatfest_id=<id>
|
||||||
|
func handleGetResults(w http.ResponseWriter, r *http.Request) {
|
||||||
|
festID := r.URL.Query().Get("splatfest_id")
|
||||||
|
|
||||||
|
rows, err := db.Query(context.Background(),
|
||||||
|
"SELECT splatfest_id, score, team_id, team_win, user_id FROM results WHERE splatfest_id = $1", festID)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
var results []CompetitionPostResults
|
||||||
|
for rows.Next() {
|
||||||
|
var res CompetitionPostResults
|
||||||
|
err := rows.Scan(&res.SplatfestID, &res.Score, &res.TeamID, &res.TeamWin, &res.User)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
results = append(results, res)
|
||||||
|
}
|
||||||
|
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
json.NewEncoder(w).Encode(results)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GET /results/votes?splatfest_id=<id>
|
||||||
|
func handleGetVotes(w http.ResponseWriter, r *http.Request) {
|
||||||
|
festID := r.URL.Query().Get("splatfest_id")
|
||||||
|
|
||||||
|
rows, err := db.Query(context.Background(),
|
||||||
|
"SELECT team_id, COUNT(user_id) FROM user_votes WHERE splatfest_id = $1 GROUP BY team_id ORDER BY team_id ASC", festID)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
// genuinely i can't be assed anymore
|
||||||
|
voteCounts := make(map[uint8]uint32)
|
||||||
|
maxID := 0
|
||||||
|
for rows.Next() {
|
||||||
|
var tid uint8
|
||||||
|
var count uint32
|
||||||
|
if err := rows.Scan(&tid, &count); err == nil {
|
||||||
|
voteCounts[tid] = count
|
||||||
|
if int(tid) > maxID {
|
||||||
|
maxID = int(tid)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
resultArr := make([]string, maxID+1)
|
||||||
|
for i := 0; i <= maxID; i++ {
|
||||||
|
resultArr[i] = strconv.FormatUint(uint64(voteCounts[uint8(i)]), 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Fprintf(w, "[%s]", strings.Join(resultArr, ","))
|
||||||
|
}
|
||||||
|
|
||||||
|
// POST /results/post
|
||||||
|
func handlePostResults(w http.ResponseWriter, r *http.Request) {
|
||||||
|
var res CompetitionPostResults
|
||||||
|
if err := json.NewDecoder(r.Body).Decode(&res); err != nil {
|
||||||
|
http.Error(w, "Invalid JSON", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
tx, err := db.Begin(ctx)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "Transaction error", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer tx.Rollback(ctx)
|
||||||
|
|
||||||
|
_, err = tx.Exec(ctx,
|
||||||
|
"INSERT INTO results (splatfest_id, score, team_id, team_win, user_id) VALUES ($1, $2, $3, $4, $5)",
|
||||||
|
res.SplatfestID, res.Score, res.TeamID, res.TeamWin, res.User)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("DB Error results: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = tx.Exec(ctx,
|
||||||
|
"INSERT INTO user_votes (user_id, splatfest_id, team_id) VALUES ($1, $2, $3) ON CONFLICT DO NOTHING",
|
||||||
|
res.User, res.SplatfestID, res.TeamID)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("DB Error votes: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
tx.Commit(ctx)
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue