问题与漏洞修改

WangHan

2025-04-02 a8ba678a3fe5a39da2c732014cebbb66e408e97c

 consum-base/src/main/java/com/consum/base/controller/BaseCategoryController.java

@@ -29,7 +29,6 @@
import com.walker.db.page.GenericPager;
import com.walker.infrastructure.utils.StringUtils;
import com.walker.web.ResponseValue;
import io.swagger.annotations.Api;
import org.springframework.web.multipart.MultipartFile;

/**
@@ -37,7 +36,7 @@
 * @Author 卢庆阳
 * @Date 2023/10/23
 */
@Api(value = "物品分类", tags = "物品分类")
// @Api(value = "物品分类", tags = "物品分类")
@RestController
@RequestMapping("/pc/base/category")
public class BaseCategoryController extends BaseController {
@@ -82,7 +81,15 @@
        if (category != null) {
            return ResponseValue.error("分类名称已存在");
        }

        // 判断添加的父级不能是3
        if (param2.getFatherCategoryId() != null) {
            BaseCategory baseCategory = new BaseCategory();
            baseCategory.setId(param2.getFatherCategoryId());
            BaseCategory baseCategory1 = this.baseCategoryService.get(baseCategory);
            if (baseCategory1 != null && baseCategory1.getLevels() >= 3) {
                return ResponseValue.error("分类最多支持三级");
            }
        }
        int result = this.baseCategoryService.add(param, this.getCurrentUser());
        if (result > 0) {
            return ResponseValue.success(1);
@@ -137,6 +144,17 @@
        if (currentUser == null) {
            return ResponseValue.error("登录用户信息不存在");
        }
        List<BaseCategory> list = this.baseGoodsTemplateService.getByCategoryId(baseCategory.getId());
        if (!CollectionUtils.isEmpty(list) && param2.getStates() != 1) {
            return ResponseValue.error("分类下有商品在用，不允许禁用");
        }
        // 查询分类下是否有子级分类
        BaseCategory baseCategory1 = new BaseCategory();
        baseCategory1.setFatherCategoryId(param2.getId());
        List<BaseCategory> select = this.baseCategoryService.select(baseCategory1);
        if (!CollectionUtils.isEmpty(select) && param2.getStates() != 1) {
            return ResponseValue.error("分类下有子级分类在用，不允许禁用");
        }
        int num = this.baseCategoryService.updateBaseCategory(baseCategory);
        return num > 0 ? ResponseValue.success(1) : ResponseValue.error("编辑失败！");
    }
@@ -160,10 +178,16 @@
        }
        // 根据分类id和状态查询物品模版
        List<BaseCategory> list = this.baseGoodsTemplateService.getByCategoryId(baseCategory.getId());
        if (baseCategory.getStates() == 2 && !CollectionUtils.isEmpty(list)) {
            return ResponseValue.error("分类下有正常状态的物品，不允许禁用");
        if (!CollectionUtils.isEmpty(list)) {
            return ResponseValue.error("分类下有商品在用，不允许禁用");
        }

        // 查询分类下是否有子级分类
        BaseCategory baseCategory1 = new BaseCategory();
        baseCategory1.setFatherCategoryId(param2.getId());
        List<BaseCategory> select = this.baseCategoryService.select(baseCategory1);
        if (!CollectionUtils.isEmpty(select)) {
            return ResponseValue.error("分类下有子级分类在用，不允许禁用");
        }
        int num = this.baseCategoryService.updateStatus(baseCategory);
        return num > 0 ? ResponseValue.success(1) : ResponseValue.error("修改失败！");
    }
@@ -179,9 +203,18 @@
        BaseCategory param2 = new BaseCategory();
        CommonUtil.copyProperties(baseCategory, param2);
        baseCategory = param2;

        if (baseCategory.getId() == null) {
            return ResponseValue.error("分类id为空");
        }
        // 判断分类是否有商品在用
        List<BaseCategory> byCategoryId = this.baseGoodsTemplateService.getByCategoryId(baseCategory.getId());
        if (!CollectionUtils.isEmpty(byCategoryId)) {
            return ResponseValue.error("分类下有商品在用，不允许删除");
        }
        // 查询分类下是否有子级分类
        List<BaseCategory> select = this.baseCategoryService.selectByFatherId(param2.getId());
        if (!CollectionUtils.isEmpty(select)) {
            return ResponseValue.error("分类下有子级分类在用，不允许删除");
        }
        int num = this.baseCategoryService.updateById(baseCategory, this.getCurrentUser());

@@ -217,7 +250,6 @@
        if (currentUser == null) {
            return ResponseValue.error("登录用户信息不存在");
        }

        List<BaseCategory> baseCategories = this.baseCategoryService.queryForLv3Tree();
        return ResponseValue.success(baseCategories);
    }