[MODORDERS-1210] Store updated pieces

folio-org · Nov 27, 2024 · bdb3693 · bdb3693
1 parent 3e09215
commit bdb3693
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/main/java/org/folio/service/pieces/flows/update/PieceUpdateFlowManager.java b/src/main/java/org/folio/service/pieces/flows/update/PieceUpdateFlowManager.java
@@ -103,7 +103,7 @@ public Future<Void> updatePiecesStatuses(List<String> pieceIds, PieceBatchStatus
         .map(entry -> new PieceBatchStatusUpdateHolder(newStatus, entry.getValue(), entry.getKey()))
         .map(holder -> basePieceFlowHolderBuilder.updateHolderWithOrderInformation(holder, requestContext)
           .compose(v -> updatePoLine(holder, requestContext))
-          .map(v -> updatePiecesStatusesByPoLine(holder, requestContext)))
+          .compose(v -> updatePiecesStatusesByPoLine(holder, requestContext)))
         .toList())
       .compose(HelperUtils::collectResultsOnSuccess)
       .mapEmpty();
@@ -144,12 +144,16 @@ protected <T extends BasePieceFlowHolder> Future<Void> updatePoLine(T holder, Li
       .onFailure(t -> log.error("Failed to update PO line with id: '{}' for pieceIds: {}", originPoLine.getId(), pieceIds, t));
   }
 
-  private boolean updatePiecesStatusesByPoLine(PieceBatchStatusUpdateHolder holder, RequestContext requestContext) {
+  private Future<Boolean> updatePiecesStatusesByPoLine(PieceBatchStatusUpdateHolder holder, RequestContext requestContext) {
     var isAnyPiecesUpdated = holder.getPieces().stream().anyMatch(piece -> updatePieceStatus(piece, piece.getReceivingStatus(), holder.getReceivingStatus()));
-    if (isAnyPiecesUpdated) {
-      pieceService.receiptConsistencyPiecePoLine(holder.getPoLineId(), requestContext);
-    }
-    return isAnyPiecesUpdated;
+    var updates = holder.getPieces().stream().map(piece -> pieceStorageService.updatePiece(piece, requestContext)).toList();
+    return HelperUtils.collectResultsOnSuccess(updates)
+      .map(v -> {
+        if (isAnyPiecesUpdated) {
+          pieceService.receiptConsistencyPiecePoLine(holder.getPoLineId(), requestContext);
+        }
+        return isAnyPiecesUpdated;
+      });
   }
 
   private List<Location> getPieceLocations(List<Piece> pieces, CompositePoLine poLine) {