Update db schema to persist polls and choices

server/server.go

@@ -19,6 +19,12 @@ var (
 	db *sql.DB
 )
 
+type ChoicesPost struct {
+	PollId   int      `json:"pollId"`
+	Username string   `json:"username"`
+	Choices  [][]bool `json:"choices"`
+}
+
 func initDatabase() {
 	var err error
 	db, err = sql.Open("sqlite3", DB_FILE)
@@ -26,17 +32,31 @@ func initDatabase() {
 		log.Fatalf("[!] Could not open database file %s, error: %s\n", DB_FILE, err)
 	}
 
-	const query = `
-		CREATE TABLE IF NOT EXISTS choices (
+	const createPollsQuery = `
+		CREATE TABLE IF NOT EXISTS polls (
 			id INTEGER PRIMARY KEY AUTOINCREMENT,
-			username TEXT NOT NULL,
-			choices TEXT NOT NULL
+			name TEXT NOT NULL
 		);
 	`
 
-	_, err = db.Exec(query)
+	_, err = db.Exec(createPollsQuery)
 	if err != nil {
-		log.Fatalf("[!] Could not create database table, error: %s\n", err)
+		log.Fatalf("[!] Could not create database table polls, error: %s\n", err)
+	}
+
+	const createChoicesQuery = `
+		CREATE TABLE IF NOT EXISTS choices (
+			id INTEGER PRIMARY KEY AUTOINCREMENT,
+			pollid INTEGER NOT NULL,
+			username TEXT NOT NULL,
+			choices TEXT NOT NULL,
+			FOREIGN KEY(pollid) REFERENCES polls(id)
+		);
+	`
+
+	_, err = db.Exec(createChoicesQuery)
+	if err != nil {
+		log.Fatalf("[!] Could not create database table choices, error: %s\n", err)
 	}
 }
 
@@ -53,25 +73,25 @@ func handlePostSubmit(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Access-Control-Allow-Methods", "OPTIONS, GET, POST")
 	w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Access-Control-Allow-Origin")
 
-	var choices [][]bool
+	var choicesPost ChoicesPost
 
-	err := json.NewDecoder(r.Body).Decode(&choices)
+	err := json.NewDecoder(r.Body).Decode(&choicesPost)
 	if err != nil {
 		log.Printf("[!] Error decoding request Body %v, error: %s\n", r.Body, err)
 		w.WriteHeader(http.StatusBadRequest)
 	}
 
-	choicesJson, err := json.Marshal(choices)
+	choicesJson, err := json.Marshal(choicesPost.Choices)
 	if err != nil {
 		log.Printf("[!] Could not marshal choices, error: %s\n", err)
 	}
 
 	const query = `
-		INSERT INTO choices (username, choices)
-		VALUES (?, ?);
+		INSERT INTO choices (pollid, username, choices)
+		VALUES (?, ?, ?);
 	`
 
-	_, err = db.Exec(query, "DUMMY USERNAME", string(choicesJson))
+	_, err = db.Exec(query, choicesPost.PollId, choicesPost.Username, string(choicesJson))
 	if err != nil {
 		log.Printf("[!] Could not insert choices into database, error: %s\n", err)
 	}