pg

backend/handlers.go

@@ -31,7 +31,7 @@ type Sublist struct {
 }
 
 type api struct {
-	db  *sql.DB
+	db  querier
 	hub *hub
 }
 
@@ -132,12 +132,11 @@ func (a *api) createSublist(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	res, err := a.db.Exec("INSERT INTO sublists (list_id, name, color) VALUES (?, ?, ?)", listID, in.Name, in.Color)
+	id, err := a.db.insertReturningID("INSERT INTO sublists (list_id, name, color) VALUES (?, ?, ?)", listID, in.Name, in.Color)
 	if err != nil {
 		writeError(w, http.StatusInternalServerError, err.Error())
 		return
 	}
-	id, _ := res.LastInsertId()
 
 	var s Sublist
 	err = a.db.QueryRow("SELECT id, name, color, created_at FROM sublists WHERE id = ?", id).
@@ -238,12 +237,11 @@ func (a *api) createItem(w http.ResponseWriter, r *http.Request) {
 	}
 	position := maxPosition.Float64 + 1
 
-	res, err := a.db.Exec("INSERT INTO items (list_id, name, quantity, position, sublist_id) VALUES (?, ?, ?, ?, ?)", listID, in.Name, in.Quantity, position, in.SublistID)
+	id, err := a.db.insertReturningID("INSERT INTO items (list_id, name, quantity, position, sublist_id) VALUES (?, ?, ?, ?, ?)", listID, in.Name, in.Quantity, position, in.SublistID)
 	if err != nil {
 		writeError(w, http.StatusInternalServerError, err.Error())
 		return
 	}
-	id, _ := res.LastInsertId()
 
 	var it Item
 	err = a.db.QueryRow("SELECT id, name, quantity, done, position, sublist_id, created_at FROM items WHERE id = ?", id).