楚江ERP：补货单生成补货单号

@@ -255,4 +255,19 @@ public class ReplenishmentOrderController extends DefaultBaseController {
 
         return InvokeResultBuilder.success();
     }
+
+    @ApiOperation("回去补回单编号")
+    @PostMapping("/getCode")
+    public InvokeResult<String> getCode(@NotBlank(message = "订货单ID不可为空！") String purchaseOrderId) {
+        PurchaseOrderInfo purchaseOrderInfo = purchaseOrderInfoService.findById(purchaseOrderId);
+        if (purchaseOrderInfo == null) {
+            throw new DefaultClientException("订货单不存在！");
+        }
+
+        Wrapper<ReplenishmentOrder> wrapper = Wrappers.lambdaQuery(ReplenishmentOrder.class)
+                .eq(ReplenishmentOrder::getPurchaseOrderId, purchaseOrderId);
+        List<ReplenishmentOrder> replenishmentOrderList = replenishmentOrderService.list(wrapper);
+        int size = CollectionUtils.isEmpty(replenishmentOrderList) ? 1 : replenishmentOrderList.size() + 1;
+        return InvokeResultBuilder.success(purchaseOrderInfo.getOrderNo() + "-P" + size);
+    }
 }