Implement SQL queries for cake updates and inserts

This commit is contained in:
bronku 2025-03-27 08:08:45 +01:00
parent d70144c39a
commit d9bcdfe186

View file

@ -75,6 +75,12 @@ func (s *Store) GetCakes() ([]Cake, error) {
}
func (s *Store) updateCake(newCake Cake) error {
query := "update cake set name = ? , price = ?, category = ?, availability = ? where id = ?"
_, err := s.db.Exec(query, newCake.Name, newCake.Price, newCake.Category, newCake.Availability, newCake.ID)
if err != nil {
return err
}
i, err := s.searchCakes(newCake.ID)
if err != nil {
return err
@ -92,7 +98,19 @@ func (s *Store) SaveCake(newCake Cake) (int, error) {
if newCake.ID != 0 {
return newCake.ID, s.updateCake(newCake)
}
newCake.ID = len(s.cakes) + 1
query := "insert into cake(name, price, category, availability) values (?, ?, ?, ?) returning id;"
rows, err := s.db.Query(query, newCake.Name, newCake.Price, newCake.Category, newCake.Availability)
if err != nil {
return 0, err
}
if !rows.Next() {
return 0, errors.New("query didn't return the cake id")
}
err = rows.Scan(&newCake.ID)
if err != nil {
return 0, err
}
s.cakes = append(s.cakes, newCake)
return newCake.ID, nil
}