fix(db/dao): removal of product from operation

1 files changed, 5 insertions, 3 deletions

diff --git a/src/main/java/ru/mrfoxygmfr/warehouse_accounting/db/models/Operation.java b/src/main/java/ru/mrfoxygmfr/warehouse_accounting/db/models/Operation.java
index 31ec626..607a08a 100644
--- a/src/main/java/ru/mrfoxygmfr/warehouse_accounting/db/models/Operation.java
+++ b/src/main/java/ru/mrfoxygmfr/warehouse_accounting/db/models/Operation.java
@@ -7,6 +7,7 @@ import org.hibernate.annotations.UpdateTimestamp;
 
 import java.sql.Timestamp;
 import java.util.HashSet;
+import java.util.Optional;
 import java.util.Set;
 import java.util.stream.Collectors;
 
@@ -69,9 +70,10 @@ public class Operation {
             products = new HashSet<>();
         }
 
-        products = products.parallelStream()
-                .filter((p) -> p.getProduct() != product.getProduct())
-                .collect(Collectors.toSet());
+        products.parallelStream()
+                .filter((p) -> p.getProduct().getId() == product.getProduct().getId())
+                .findAny()
+                .ifPresent(operationProduct -> products.remove(operationProduct));
 
         if (product.getAmount() == 0) {
             return;