update

pkg/cart/mutation_change_quantity.go

@@ -1,7 +1,9 @@
 package cart
 
 import (
+	"context"
 	"fmt"
+	"log"
 
 	messages "git.k6n.net/go-cart-actor/pkg/messages"
 )
@@ -30,6 +32,7 @@ func (c *CartMutationContext) ChangeQuantity(g *CartGrain, m *messages.ChangeQua
 	if m == nil {
 		return fmt.Errorf("ChangeQuantity: nil payload")
 	}
+	ctx := context.Background()
 
 	foundIndex := -1
 	for i, it := range g.Items {
@@ -44,6 +47,11 @@ func (c *CartMutationContext) ChangeQuantity(g *CartGrain, m *messages.ChangeQua
 
 	if m.Quantity <= 0 {
 		// Remove the item
+		itemToRemove := g.Items[foundIndex]
+		err := c.ReleaseItem(ctx, g.Id, itemToRemove.Sku, itemToRemove.StoreId)
+		if err != nil {
+			log.Printf("unable to release reservation for %s in location: %v", itemToRemove.Sku, itemToRemove.StoreId)
+		}
 		g.Items = append(g.Items[:foundIndex], g.Items[foundIndex+1:]...)
 		g.UpdateTotals()
 		return nil
@@ -53,10 +61,19 @@ func (c *CartMutationContext) ChangeQuantity(g *CartGrain, m *messages.ChangeQua
 		return fmt.Errorf("ChangeQuantity: item id %d not found", m.Id)
 	}
 	if item.ReservationEndTime != nil {
-		return fmt.Errorf("ChangeQuantity: cannot change quantity of reserved item id %d", m.Id)
-	} else {
-		item.Quantity = int(m.Quantity)
-		g.UpdateTotals()
+		err := c.ReleaseItem(ctx, g.Id, item.Sku, item.StoreId)
+		if err != nil {
+			log.Printf("unable to release reservation for %s in location: %v", item.Sku, item.StoreId)
+		}
+
 	}
+	endTime, err := c.ReserveItem(ctx, g.Id, item.Sku, item.StoreId, uint16(m.Quantity))
+	if err != nil {
+		return err
+	}
+	item.ReservationEndTime = endTime
+	item.Quantity = uint16(m.Quantity)
+	g.UpdateTotals()
+
 	return nil
 }