bronku/cake-order-tracker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

extracted UpdateOrderContents

Browse source
This commit is contained in:
bronku 2025-03-31 07:42:31 +02:00
parent 3136739f4e
commit 5661ac4594
2 changed files with 18 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

31
internal/store/order.go
View file

@ -99,6 +99,23 @@ func (s *Store) GetTopOrders(from, to time.Time) ([]models.Order, error) {
return s.getOrdersFromQuery(query, start, end)
}
func (s *Store) UpdateOrderContents(tx *sql.Tx, newOrder models.Order) error {
query := "delete from ordered_cake where customer_order = ?;"
_, err := tx.Exec(query, newOrder.ID)
if err != nil {
return err
}
query = "insert into ordered_cake(customer_order, cake, amount) values (?,?,?);"
for _, e := range newOrder.Cakes {
_, err := tx.Exec(query, newOrder.ID, e.ID, e.Amount)
if err != nil {
return err
}
}
return nil
}
func (s *Store) SaveOrder(newOrder models.Order) (int, error) {
query := "insert into customer_order(name, surname, phone, location, order_date, delivery_date, status, paid) values (?, ?, ?, ?, ?, ?, ?, ?) returning id;"
if newOrder.ID != 0 {
@ -129,24 +146,12 @@ func (s *Store) SaveOrder(newOrder models.Order) (int, error) {
return newOrder.ID, err
}
// remove all ordered_cakes associated with this order before inserting
query = "delete from ordered_cake where customer_order = ?;"
_, err = tx.Exec(query, newOrder.ID)
err = s.UpdateOrderContents(tx, newOrder)
if err != nil {
_ = tx.Rollback()
return newOrder.ID, err
}
// add all ordered_cakes for this order
query = "insert into ordered_cake(customer_order, cake, amount) values (?,?,?);"
for _, e := range newOrder.Cakes {
_, err := tx.Exec(query, newOrder.ID, e.ID, e.Amount)
if err != nil {
_ = tx.Rollback()
return newOrder.ID, err
}
}
err = tx.Commit()
return newOrder.ID, err
}