From 76561071d3441cd8a551772d0f43d93c9f9a1e6a Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期二, 29 十月 2024 18:08:25 +0800 Subject: [PATCH] 修改 --- ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TDataGeneratorServiceImpl.java | 23 ++++++++++++++++------- 1 files changed, 16 insertions(+), 7 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 72e12d1..1d09a6c 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 @@ -49,6 +49,8 @@ @Autowired private TFoundationConfigService foundationConfigService; @Autowired + private TFoundationPersonService foundationPersonService; + @Autowired private TGoodsService goodsService; @Autowired private TOrderMealService orderMealService; @@ -92,10 +94,10 @@ dataGenerator.setOtherPay(dto.getOtherProportion()); this.save(dataGenerator); long start = System.currentTimeMillis(); - // 查询所有的人数用餐标准 - List<TFoundationConfigVO> foundationConfigs = foundationConfigService.getList(dto.getShopId()); + TShop shop = shopService.getById(dto.getShopId()); + List<TFoundationConfigVO> foundationConfigs = foundationConfigService.getList(shop.getMealType()); if(CollectionUtils.isEmpty(foundationConfigs)){ - throw new ServiceException("请先添加该店铺用餐标准"); + throw new ServiceException("请先添加中餐或火锅用餐标准"); } // 查询所有桌子 List<TBoard> boards = boardService.list(Wrappers.lambdaQuery(TBoard.class) @@ -118,6 +120,9 @@ throw new ServiceException("该桌台信息不存在"); } Integer orderCount = orderMealGeneratorCountDTO.getOrderCount(); + if(orderCount == 0){ + continue; + } // 查询当前桌的用餐人数 for (int i = 1; i <= orderCount; i++) { int random = getRandom(board.getMinPerson(), board.getMaxPerson()); @@ -137,15 +142,19 @@ // 每种菜品分类需要几个菜的生成 foundationConfigVOS.forEach(e -> { int random1 = getRandom(e.getMinCount(), e.getMaxCount()); - List<TGoods> typeGoods = goods.stream().filter(m -> e.getTypeId().equals(m.getTypeId())).collect(Collectors.toList()); - typeGoods = randomSelection(typeGoods, random1); + List<TGoods> typeGoodsList = goods.stream().filter(m -> e.getTypeId().equals(m.getTypeId())).collect(Collectors.toList()); + List<TGoods> typeGoods = randomSelection(typeGoodsList, random1); + if(typeGoods.size() != random1){ + List<TGoods> goods1 = randomSelection(typeGoodsList, random1 - typeGoods.size()); + typeGoods.addAll(goods1); + } for (TGoods typeGood : typeGoods) { int count = 0; for (TOrderMealGoods good : orderMealGoods) { if (good.getGoodsNum().equals(typeGood.getGoodsNum())) { count+=1; good.setGoodsCount(good.getGoodsCount() + 1); - good.setGoodsSalePrice(good.getGoodsSalePrice().add(typeGood.getSalePrice())); +// good.setGoodsSalePrice(good.getGoodsSalePrice().add(typeGood.getSalePrice())); break; // 找到后直接跳出循环 } } @@ -474,7 +483,7 @@ return; } // 删除上一次的数据生成 - if(Objects.nonNull(orderStockLast)){ + if(Objects.nonNull(orderStockLast) && Objects.nonNull(stockDataSet) && stockDataSets.size() > 1){ orderStockService.removeById(orderStockLast); } orderSaleService.saveBatch(orderSales); -- Gitblit v1.7.1