| package com.zy.asrs.wms.asrs.controller; | 
|   | 
| import com.alibaba.fastjson.JSON; | 
| import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; | 
| 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; | 
| import com.zy.asrs.wms.common.domain.PageParam; | 
| import com.zy.asrs.wms.asrs.entity.ShelvesRule; | 
| import com.zy.asrs.wms.asrs.service.ShelvesRuleService; | 
| import com.zy.asrs.wms.system.controller.BaseController; | 
| 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; | 
| import java.util.*; | 
|   | 
| @RestController | 
| @RequestMapping("/api") | 
| public class ShelvesRuleController extends BaseController { | 
|   | 
|     @Autowired | 
|     private ShelvesRuleService shelvesRuleService; | 
|     @Autowired | 
|     private ShelvesRuleDetlService shelvesRuleDetlService; | 
|   | 
|     @PreAuthorize("hasAuthority('asrs:shelvesRule:list')") | 
|     @PostMapping("/shelvesRule/page") | 
|     public R page(@RequestBody Map<String, Object> map) { | 
|         BaseParam baseParam = buildParam(map, BaseParam.class); | 
|         PageParam<ShelvesRule, BaseParam> pageParam = new PageParam<>(baseParam, ShelvesRule.class); | 
|         return R.ok().add(shelvesRuleService.page(pageParam, pageParam.buildWrapper(true))); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('asrs:shelvesRule:list')") | 
|     @PostMapping("/shelvesRule/list") | 
|     public R list(@RequestBody Map<String, Object> map) { | 
|         return R.ok().add(shelvesRuleService.list()); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('asrs:shelvesRule:list')") | 
|     @GetMapping("/shelvesRule/{id}") | 
|     public R get(@PathVariable("id") Long id) { | 
|         return R.ok().add(shelvesRuleService.getById(id)); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('asrs:shelvesRule:save')") | 
|     @OperationLog("添加上架规则") | 
|     @PostMapping("/shelvesRule/save") | 
|     public R save(@RequestBody HashMap<String, Object> param) { | 
|         if(!param.containsKey("ruleType") || !param.containsKey("laneList")) { | 
|             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)) { | 
|             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("添加成功"); | 
|     } | 
|   | 
|     @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("缺少参数"); | 
|         } | 
|   | 
|         ShelvesRule shelvesRule = shelvesRuleService.getById(param.get("id").toString()); | 
|         if (shelvesRule == null) { | 
|             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)) { | 
|             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("修改成功"); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('asrs:shelvesRule:remove')") | 
|     @OperationLog("删除上架规则") | 
|     @PostMapping("/shelvesRule/remove/{ids}") | 
|     public R remove(@PathVariable Long[] ids) { | 
|         if (!shelvesRuleService.removeByIds(Arrays.asList(ids))) { | 
|             return R.error("删除失败"); | 
|         } | 
|         return R.ok("删除成功"); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('asrs:shelvesRule:list')") | 
|     @PostMapping("/shelvesRule/query") | 
|     public R query(@RequestParam(required = false) String condition) { | 
|         List<KeyValVo> vos = new ArrayList<>(); | 
|         LambdaQueryWrapper<ShelvesRule> wrapper = new LambdaQueryWrapper<>(); | 
|         if (!Cools.isEmpty(condition)) { | 
|             wrapper.like(ShelvesRule::getId, condition); | 
|         } | 
|         shelvesRuleService.page(new Page<>(1, 30), wrapper).getRecords().forEach( | 
|                 item -> vos.add(new KeyValVo(item.getId(), item.getId())) | 
|         ); | 
|         return R.ok().add(vos); | 
|     } | 
|   | 
|     @PreAuthorize("hasAuthority('asrs:shelvesRule:list')") | 
|     @PostMapping("/shelvesRule/export") | 
|     public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception { | 
|         ExcelUtil.build(ExcelUtil.create(shelvesRuleService.list(), ShelvesRule.class), response); | 
|     } | 
|   | 
| } |