diff --git a/src/main/java/com/zmzm/finance/common/controller/ImportController.java b/src/main/java/com/zmzm/finance/common/controller/ImportController.java index b2f6d52..ff3a471 100644 --- a/src/main/java/com/zmzm/finance/common/controller/ImportController.java +++ b/src/main/java/com/zmzm/finance/common/controller/ImportController.java @@ -2,9 +2,11 @@ package com.zmzm.finance.common.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.zmzm.finance.common.entity.Import; +import com.zmzm.finance.common.entity.Orders; import com.zmzm.finance.common.entity.Payment; import com.zmzm.finance.common.entity.PaymentToOrder; import com.zmzm.finance.common.service.IImportService; +import com.zmzm.finance.common.service.IOrdersService; import com.zmzm.finance.common.service.IPaymentService; import com.zmzm.finance.common.service.IPaymentToOrderService; import com.zmzm.finance.util.ExcelUtil; @@ -42,6 +44,8 @@ public class ImportController { private IPaymentService paymentService; @Autowired private IPaymentToOrderService paymentToOrderService; + @Autowired + private IOrdersService ordersService; //导入文件列表 @RequestMapping("/getImportList") @@ -78,8 +82,15 @@ public class ImportController { if (!oldPaymentIds.isEmpty()) { List ptos = paymentToOrderService.list(new LambdaQueryWrapper() .in(PaymentToOrder::getPaymentId,oldPaymentIds)); + List os = new ArrayList<>(); + for (PaymentToOrder pto:ptos) { + Orders o = ordersService.getById(pto.getOrderId()); + o.setUseFlag(0); + os.add(o); + } paymentService.removeBatchByIds(oldPaymentIds); paymentToOrderService.removeByIds(ptos); + ordersService.updateBatchById(os); } }else { imp = new Import();