From 20cd42287efb95254ea788ce2b6d0c559e378e04 Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期四, 19 九月 2024 14:36:21 +0800 Subject: [PATCH] 修改接口 --- ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TDataGeneratorServiceImpl.java | 27 +++++++++++++++++++-------- 1 files changed, 19 insertions(+), 8 deletions(-) diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TDataGeneratorServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TDataGeneratorServiceImpl.java index 19c1960..d2561a5 100644 --- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TDataGeneratorServiceImpl.java +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TDataGeneratorServiceImpl.java @@ -90,11 +90,19 @@ this.save(dataGenerator); long start = System.currentTimeMillis(); // 查询所有的人数用餐标准 - List<TFoundationConfigVO> foundationConfigs = foundationConfigService.getList(); + List<TFoundationConfigVO> foundationConfigs = foundationConfigService.getList(dto.getShopId()); // 查询所有桌子 - List<TBoard> boards = boardService.list(); + List<TBoard> boards = boardService.list(Wrappers.lambdaQuery(TBoard.class) + .eq(TBoard::getShopId,dto.getShopId())); + if(CollectionUtils.isEmpty(boards)){ + throw new ServiceException("请先添加桌台信息"); + } // 查询所有菜品 - List<TGoods> goods = goodsService.list(); + List<TGoods> goods = goodsService.list(Wrappers.lambdaQuery(TGoods.class) + .eq(TGoods::getShopId,dto.getShopId())); + if(CollectionUtils.isEmpty(goods)){ + throw new ServiceException("请先添加菜品信息"); + } // 循环待生成订单列表,添加到集合中,批量插入 List<OrderMealGeneratorCountDTO> orderMealGeneratorCountDTOS = dto.getOrderMealGeneratorCountDTOS(); List<TOrderMeal> orderMeals = new ArrayList<>(); @@ -141,14 +149,16 @@ } } }); - orderMeal.setMealOrderGoods(orderMealGoods); - orderMeal.setOrderMoney(orderMealGoods.stream().map(TOrderMealGoods::getGoodsSalePrice).reduce(BigDecimal::add).get()); - orderMeal.setPayMoney(orderMealGoods.stream().map(TOrderMealGoods::getGoodsSalePrice).reduce(BigDecimal::add).get()); - orderMeals.add(orderMeal); + if(!CollectionUtils.isEmpty(orderMealGoods)){ + orderMeal.setMealOrderGoods(orderMealGoods); + orderMeal.setOrderMoney(orderMealGoods.stream().map(TOrderMealGoods::getGoodsSalePrice).reduce(BigDecimal::add).get()); + orderMeal.setPayMoney(orderMealGoods.stream().map(TOrderMealGoods::getGoodsSalePrice).reduce(BigDecimal::add).get()); + orderMeals.add(orderMeal); + } } } BigDecimal sumMoney = orderMeals.stream().map(TOrderMeal::getPayMoney).reduce(BigDecimal::add).get(); - if(sumMoney.compareTo(dto.getMinMoney()) <= 0 || sumMoney.compareTo(dto.getMaxMoney()) >= 0){ + if(sumMoney.compareTo(dto.getMinMoney()) >= 0 && sumMoney.compareTo(dto.getMaxMoney()) <= 0){ int weiXin = getRandomPayType(orderMeals.size(), dto.getWeiXinProportion()); int ali = getRandomPayType(orderMeals.size(), dto.getAliProportion()); int card = getRandomPayType(orderMeals.size(), dto.getCardProportion()); @@ -222,6 +232,7 @@ long end = System.currentTimeMillis() - start; System.err.println("相差时间========="+end); }else { + this.removeById(dataGenerator); throw new ServiceException("数据生成不在营业额范围内"); } -- Gitblit v1.7.1