From bfdb3faf4f27df01718f58ac8c4ec0bcc092e7b6 Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期六, 30 十一月 2024 18:50:47 +0800 Subject: [PATCH] 生成数据明细导出修改 --- ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java | 28 +++++++++++++++++----------- 1 files changed, 17 insertions(+), 11 deletions(-) diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java index b5e28f5..d44f85b 100644 --- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TGoodsTypeController.java @@ -47,13 +47,14 @@ @ApiOperation( value = "查询商品分类列表") @PostMapping(value = "/list") public AjaxResult<List<TGoodsType>> list() { -// Long shopId = tokenService.getLoginUser().getObjectId(); -// Integer roleType = tokenService.getLoginUser().getRoleType(); -// LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>(); -// if(roleType != 1){ -// wrapper.eq(TGoodsType::getShopId, shopId); -// } - return AjaxResult.success(goodsTypeService.list()); + Long shopId = tokenService.getLoginUser().getObjectId(); + Integer roleType = tokenService.getLoginUser().getRoleType(); + TShop shop = shopService.getById(shopId); + LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>(); + if(roleType != 1){ + wrapper.eq(TGoodsType::getMealType, shop.getMealType()); + } + return AjaxResult.success(goodsTypeService.list(wrapper)); } /** @@ -61,7 +62,7 @@ */ @ApiOperation( value = "查询商品分类列表-后台使用") @GetMapping(value = "/getList") - public AjaxResult<List<TGoodsType>> getList(@RequestParam("mealType") Integer mealType) { + public AjaxResult<List<TGoodsType>> getList(@RequestParam(value = "mealType",required = false) Integer mealType) { // 获取餐饮分类店铺id // TShop one = shopService.getOne(Wrappers.lambdaQuery(TShop.class) // .eq(TShop::getMealType, mealType) @@ -70,9 +71,11 @@ // String str = mealType == 1 ? "中餐" : "火锅"; // return AjaxResult.error("无"+str+"类型店铺"); // } -// LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>(); -// wrapper.eq(TGoodsType::getShopId, one.getId()); - return AjaxResult.success(goodsTypeService.list()); + LambdaQueryWrapper<TGoodsType> wrapper = new LambdaQueryWrapper<>(); + if(Objects.nonNull(mealType)){ + wrapper.eq(TGoodsType::getMealType, mealType); + } + return AjaxResult.success(goodsTypeService.list(wrapper)); } /** @@ -81,6 +84,9 @@ @ApiOperation( value = "添加商品分类") @PostMapping(value = "/add") public AjaxResult<Boolean> add(@RequestBody TGoodsType dto) { + Long shopId = tokenService.getLoginUser().getObjectId(); + TShop shop = shopService.getById(shopId); + dto.setMealType(shop.getMealType()); dto.setShopId(tokenService.getLoginUser().getObjectId()); return AjaxResult.success(goodsTypeService.save(dto)); } -- Gitblit v1.7.1