From 7f5f8fc0fd5a5b99b6643df3e3f7293fea9a761e Mon Sep 17 00:00:00 2001 From: red binder Date: Sun, 12 Apr 2026 16:50:06 +0200 Subject: [PATCH] add code --- go.mod | 13 ++++++ go.sum | 26 +++++++++++ main.go | 137 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 176 insertions(+) create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..2e0e12f --- /dev/null +++ b/go.mod @@ -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 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..8e29ab9 --- /dev/null +++ b/go.sum @@ -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= diff --git a/main.go b/main.go new file mode 100644 index 0000000..8744bc7 --- /dev/null +++ b/main.go @@ -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= +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= +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) +}