#

Junjie

2024-09-12 fb4e7fd8575a97467a972028a91484f71985656c

 zy-asrs-wms/src/main/java/com/zy/asrs/wms/asrs/controller/ShelvesRuleController.java

@@ -5,6 +5,9 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.zy.asrs.framework.common.Cools;
import com.zy.asrs.framework.common.R;
import com.zy.asrs.framework.exception.CoolException;
import com.zy.asrs.wms.asrs.entity.ShelvesRuleDetl;
import com.zy.asrs.wms.asrs.service.ShelvesRuleDetlService;
import com.zy.asrs.wms.common.annotation.OperationLog;
import com.zy.asrs.wms.common.domain.BaseParam;
import com.zy.asrs.wms.common.domain.KeyValVo;
@@ -15,6 +18,7 @@
import com.zy.asrs.wms.utils.ExcelUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
@@ -26,6 +30,8 @@

    @Autowired
    private ShelvesRuleService shelvesRuleService;
    @Autowired
    private ShelvesRuleDetlService shelvesRuleDetlService;

    @PreAuthorize("hasAuthority('asrs:shelvesRule:list')")
    @PostMapping("/shelvesRule/page")
@@ -55,10 +61,21 @@
            return R.error("缺少参数");
        }
        ShelvesRule shelvesRule = new ShelvesRule();
        shelvesRule.setName(param.get("name").toString());
        shelvesRule.setRuleType(Integer.parseInt(param.get("ruleType").toString()));
        shelvesRule.setLane(JSON.toJSONString(param.get("laneList")));
        if (!shelvesRuleService.save(shelvesRule)) {
            return R.error("添加失败");
            throw new CoolException("添加失败");
        }

        List<Integer> detls = JSON.parseArray(JSON.toJSONString(param.get("detls")), Integer.class);
        for (Integer detlType : detls) {
            ShelvesRuleDetl shelvesRuleDetl = new ShelvesRuleDetl();
            shelvesRuleDetl.setRuleId(shelvesRule.getId());
            shelvesRuleDetl.setDetlType(detlType);
            if (!shelvesRuleDetlService.save(shelvesRuleDetl)) {
                throw new CoolException("添加规则明细失败");
            }
        }
        return R.ok("添加成功");
    }
@@ -66,6 +83,7 @@
    @PreAuthorize("hasAuthority('asrs:shelvesRule:update')")
    @OperationLog("修改上架规则")
    @PostMapping("/shelvesRule/update")
    @Transactional
    public R update(@RequestBody HashMap<String, Object> param) {
        if(!param.containsKey("ruleType") || !param.containsKey("laneList") || !param.containsKey("id")) {
            return R.error("缺少参数");
@@ -76,11 +94,25 @@
            return R.error("上架规则不存在");
        }

        shelvesRule.setName(param.get("name").toString());
        shelvesRule.setRuleType(Integer.parseInt(param.get("ruleType").toString()));
        shelvesRule.setLane(JSON.toJSONString(param.get("laneList")));
        shelvesRule.setStatus(Integer.parseInt(param.get("status").toString()));
        if (!shelvesRuleService.updateById(shelvesRule)) {
            return R.error("修改失败");
            throw new CoolException("修改失败");
        }

        shelvesRuleDetlService.remove(new LambdaQueryWrapper<ShelvesRuleDetl>().eq(ShelvesRuleDetl::getRuleId, shelvesRule.getId()));
        List<Integer> detls = JSON.parseArray(JSON.toJSONString(param.get("detls")), Integer.class);
        for (Integer detlType : detls) {
            ShelvesRuleDetl shelvesRuleDetl = new ShelvesRuleDetl();
            shelvesRuleDetl.setRuleId(shelvesRule.getId());
            shelvesRuleDetl.setDetlType(detlType);
            if (!shelvesRuleDetlService.save(shelvesRuleDetl)) {
                throw new CoolException("添加规则明细失败");
            }
        }

        return R.ok("修改成功");
    }