diff --git a/internal/database/vote.go b/internal/database/vote.go index b7e334e..c64d044 100644 --- a/internal/database/vote.go +++ b/internal/database/vote.go @@ -12,13 +12,13 @@ const VALUESTEMPLATE string = "(%s, %s, %s, %d)" const INITVOTEDB string = "CREATE TABLE IF NOT EXISTS vote(id varchar(16) PRIMARY KEY, category TEXT NOT NULL, name TEXT NOT NULL, votes int NOT NULL)" const INSERTVOTEENTRY string = "INSERT INTO vote(id, category, name, votes) VALUES($1, $2, $3, $4)" -const BULKINSERTVOTEENTRY string = "INSERT INTO vote(id, name, votes) VALUES" +const BULKINSERTVOTEENTRY string = "INSERT INTO vote(id, category, name, votes) VALUES" const SETVOTE string = "UPDATE vote SET votes = $1 WHERE id = $2" const INCVOTE string = "UPDATE vote SET votes = votes + 1 WHERE id = $1" const DELVOTEENTRY string = "DELETE FROM vote WHERE id = $1" const BULKDELVOTEENTRY string = "DELETE FROM vote WHERE key IN (%s)" const GETVOTES string = "SELECT * FROM vote ORDER BY votes DESC" -const GETVOTEOPTIONS string = "SELECT id, name FROM vote" +const GETVOTEOPTIONS string = "SELECT id, category, name FROM vote" const GETVOTE string = "SELECT * FROM vote WHERE id = $1" type VoteEntry struct { @@ -43,7 +43,7 @@ func InsertVoteEntry(name string, category string) (VoteEntry, error) { if err != nil { return VoteEntry{}, err } - _, err = db.Exec(INSERTVOTEENTRY, id, name, 0) + _, err = db.Exec(INSERTVOTEENTRY, id, category, name, 0) if err != nil { return VoteEntry{}, err