| package com.zy.asrs.controller; | 
|   | 
| import com.alibaba.excel.EasyExcel; | 
| import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy; | 
| import com.alibaba.fastjson.JSONArray; | 
| import com.alibaba.fastjson.JSONObject; | 
| import com.baomidou.mybatisplus.mapper.EntityWrapper; | 
| import com.baomidou.mybatisplus.mapper.Wrapper; | 
| import com.baomidou.mybatisplus.plugins.Page; | 
| import com.core.annotations.ManagerAuth; | 
| import com.core.common.BaseRes; | 
| import com.core.common.Cools; | 
| import com.core.common.DateUtils; | 
| import com.core.common.R; | 
| import com.core.exception.CoolException; | 
| import com.zy.asrs.entity.ManLocDetl; | 
| import com.zy.asrs.entity.Node; | 
| import com.zy.asrs.entity.param.InitPakoutParam; | 
| import com.zy.asrs.entity.param.PakinParam; | 
| import com.zy.asrs.mapper.ManLocDetlMapper; | 
| import com.zy.asrs.service.NodeService; | 
| import com.zy.common.entity.NodeExcel; | 
| import com.zy.common.entity.NodeExcelListener; | 
| import com.zy.common.utils.ListUtils; | 
| import com.zy.common.utils.NodeUtils; | 
| import com.zy.common.utils.TreeUtils; | 
| import com.zy.common.web.BaseController; | 
| import org.springframework.beans.factory.annotation.Autowired; | 
| import org.springframework.transaction.annotation.Transactional; | 
| import org.springframework.web.bind.annotation.*; | 
| import org.springframework.web.multipart.MultipartFile; | 
|   | 
| import javax.servlet.http.HttpServletResponse; | 
| import java.io.IOException; | 
| import java.net.URLEncoder; | 
| import java.util.*; | 
|   | 
| import static jdk.nashorn.api.scripting.ScriptUtils.convert; | 
|   | 
| @RestController | 
| public class NodeController extends BaseController { | 
|   | 
|     @Autowired | 
|     private NodeService nodeService; | 
|     @Autowired | 
|     private TreeUtils treeUtils; | 
|     @Autowired | 
|     private ManLocDetlMapper manLocDetlMapper; | 
|   | 
|     @RequestMapping(value = "/node/{id}/auth") | 
|     @ManagerAuth | 
|     public R get(@PathVariable("id") String id) { | 
|         return R.ok(nodeService.selectById(String.valueOf(id))); | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/list/auth") | 
|     @ManagerAuth | 
|     public R list(@RequestParam(defaultValue = "1")Integer curr, | 
|                   @RequestParam(defaultValue = "10")Integer limit, | 
|                   @RequestParam(required = false)String orderByField, | 
|                   @RequestParam(required = false)String orderByType, | 
|                   @RequestParam Map<String, Object> param){ | 
|         EntityWrapper<Node> wrapper = new EntityWrapper<>(); | 
|         excludeTrash(param); | 
|         convert(param, wrapper); | 
|         hostEq(wrapper); | 
|         if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));} | 
|         return R.ok(nodeService.selectPage(new Page<>(curr, limit), wrapper)); | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/list/tree/auth") | 
|     @ManagerAuth | 
|     public R listTree(@RequestParam(defaultValue = "1")Integer curr, | 
|                       @RequestParam(defaultValue = "10")Integer limit, | 
|                       @RequestParam(required = false)String orderByField, | 
|                       @RequestParam(required = false)String orderByType, | 
|                       @RequestParam Map<String, Object> param){ | 
|         EntityWrapper<Node> wrapper = new EntityWrapper<>(); | 
|         excludeTrash(param); | 
|         convert(param, wrapper); | 
|         hostEq(wrapper); | 
|         if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));} | 
|         return R.parse("0-操作成功").add(nodeService.selectList(wrapper)); | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/tree/auth") | 
|     @ManagerAuth | 
|     public R tree(@RequestParam(defaultValue = "1")Integer curr, | 
|                   @RequestParam(defaultValue = "10")Integer limit, | 
|                   @RequestParam(required = false)String orderByField, | 
|                   @RequestParam(required = false)String orderByType, | 
|                   @RequestParam Map<String, Object> param){ | 
|         EntityWrapper<Node> wrapper = new EntityWrapper<>(); | 
|         excludeTrash(param); | 
|         convert(param, wrapper); | 
|         hostEq(wrapper); | 
|         if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));} | 
|         return R.parse("0-操作成功").add(nodeService.selectList(wrapper)); | 
|     } | 
|   | 
|     private void convert(Map<String, Object> map, EntityWrapper wrapper){ | 
|         for (Map.Entry<String, Object> entry : map.entrySet()){ | 
|             String val = String.valueOf(entry.getValue()); | 
|             if (val.contains(RANGE_TIME_LINK)){ | 
|                 String[] dates = val.split(RANGE_TIME_LINK); | 
|                 wrapper.ge(entry.getKey(), DateUtils.convert(dates[0])); | 
|                 wrapper.le(entry.getKey(), DateUtils.convert(dates[1])); | 
|             } else { | 
|                 wrapper.like(entry.getKey(), val); | 
|             } | 
|         } | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/add/auth") | 
|     @ManagerAuth | 
|     public R add(Node node) { | 
|         Node node1 = nodeService.selectByUuid(node.getName()); | 
|         if (!Cools.isEmpty(node1)) { | 
|             throw new CoolException("该编号/名称已存在"); | 
|         } | 
|         if (node.getType() != 1) { | 
|             Node parentNode = nodeService.selectById(node.getParentId()); | 
|             if (parentNode == null || parentNode.getStatus() == 0) { | 
|                 return R.error(node.getType()==2?"所属仓库不存在":"所属库区不存在"); | 
|             } | 
|             node.setParentName(parentNode.getName()); | 
|             node.setLevel(parentNode.getLevel() + 1); | 
|         } | 
|         // path | 
|         NodeUtils nodeUtils = new NodeUtils(); | 
|         nodeUtils.executePath(node); | 
|         node.setHostId(getHostId()); | 
|         node.setUuid(node.getName().toString()); | 
|         node.setPath(nodeUtils.path.toString()); | 
|         node.setNamePath(nodeUtils.pathName.toString()); | 
|         node.setLevel(node.getType()); | 
|         node.setCreateBy(getUserId()); | 
|         node.setCreateTime(new Date()); | 
|         node.setUpdateBy(getUserId()); | 
|         node.setUpdateTime(new Date()); | 
|         node.setStatus(1); | 
|         nodeService.insert(node); | 
|         return R.ok(); | 
|     } | 
|   | 
|   | 
|   | 
|     @RequestMapping(value = "/node/update/auth") | 
|     @ManagerAuth | 
|     public R update(Node node){ | 
|         Node node1 = nodeService.selectByUuid(node.getName()); | 
|         if (!Cools.isEmpty(node1)) { | 
|             throw new CoolException("该编号/名称已存在"); | 
|         } | 
|         if (Cools.isEmpty(node) || null==node.getId()){ | 
|             return R.error(); | 
|         } | 
|         if (node.getType() != 1) { | 
|             Node parentNode = nodeService.selectById(node.getParentId()); | 
|             if (parentNode == null || parentNode.getStatus() == 0) { | 
|                 return R.error(node.getType()==2?"所属仓库不存在":"所属库区不存在"); | 
|             } | 
|             node.setParentName(parentNode.getName()); | 
|             node.setLevel(parentNode.getLevel() + 1); | 
|         } | 
|         // path | 
|         NodeUtils nodeUtils = new NodeUtils(); | 
|         nodeUtils.executePath(node); | 
|         node.setPath(nodeUtils.path.toString()); | 
|         node.setNamePath(nodeUtils.pathName.toString()); | 
|         node.setUuid(node.getName()); | 
|   | 
|         manLocDetlMapper.updateLocNo0(node.getId(),node.getName()); | 
|         node.setUpdateBy(getUserId()); | 
|         node.setUpdateTime(new Date()); | 
|         nodeService.updateById(node); | 
|         return R.ok(); | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/delete/auth") | 
|     @ManagerAuth | 
|     public R delete(@RequestParam String param){ | 
|         List<Node> list = JSONArray.parseArray(param, Node.class); | 
|         if (Cools.isEmpty(list)){ | 
|             return R.error(); | 
|         } | 
|         for (Node entity : list){ | 
|             nodeService.delete(new EntityWrapper<>(entity)); | 
|         } | 
|         return R.ok(); | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/delete0/auth") | 
|     @ManagerAuth | 
|     public R delete(@RequestParam(value="ids[]") Long[] ids){ | 
|         for (Long id : ids){ | 
|             nodeService.deleteById(id); | 
|         } | 
|         return R.ok(); | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/export/auth") | 
|     @ManagerAuth | 
|     public R export(@RequestBody JSONObject param){ | 
|         EntityWrapper<Node> wrapper = new EntityWrapper<>(); | 
|         List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class); | 
|         Map<String, Object> map = excludeTrash(param.getJSONObject("node")); | 
|         convert(map, wrapper); | 
|         List<Node> list = nodeService.selectList(wrapper); | 
|         return R.ok(exportSupport(list, fields)); | 
|     } | 
|   | 
|     @RequestMapping(value = "/nodeQuery/auth") | 
|     @ManagerAuth | 
|     public R query(String condition) { | 
|         EntityWrapper<Node> wrapper = new EntityWrapper<>(); | 
|         wrapper.like("name", condition).or().like("uuid", condition).eq("type", 3); | 
|         hostEq(wrapper); | 
|         Page<Node> page = nodeService.selectPage(new Page<>(0, 10), wrapper); | 
|         List<Map<String, Object>> result = new ArrayList<>(); | 
|         for (Node node : page.getRecords()){ | 
|             Map<String, Object> map = new HashMap<>(); | 
|             map.put("id", node.getId()); | 
|             map.put("value", node.getUuid() + "(" +node.getName()+ ")"); | 
|             result.add(map); | 
|         } | 
|         return R.ok(result); | 
|     } | 
|   | 
|     @RequestMapping(value = "/node/check/column/auth") | 
|     @ManagerAuth | 
|     public R query(@RequestBody JSONObject param) { | 
|         Wrapper<Node> wrapper = new EntityWrapper<Node>().eq(humpToLine(String.valueOf(param.get("key"))), param.get("val")); | 
|         if (null != nodeService.selectOne(wrapper)){ | 
|             return R.parse(BaseRes.REPEAT).add(getComment(Node.class, String.valueOf(param.get("key")))); | 
|         } | 
|         return R.ok(); | 
|     } | 
|   | 
|     @PostMapping(value = "/parent/node/group") | 
|     @ManagerAuth | 
|     public R getParentNodeGroup(@RequestParam(required = false) String condition, | 
|                                 @RequestParam Integer type) { | 
|         EntityWrapper<Node> wrapper = new EntityWrapper<>(); | 
|         wrapper.eq("type", type).eq("status", 1); | 
|         hostEq(wrapper); | 
|         List<Node> nodes = nodeService.selectList(wrapper); | 
|   | 
|         List<Map<String, Object>> result = new ArrayList<>(); | 
|         for (Node node : nodes) { | 
|             Map<String, Object> map = new HashMap<>(); | 
|             map.put("key", node.getId()); | 
|             map.put("val", node.getName()); | 
|             result.add(map); | 
|         } | 
|         return R.ok().add(result); | 
|     } | 
|   | 
|     @PostMapping(value = "/node/tree/auth") | 
|     @ManagerAuth | 
|     public R tree(@RequestParam(required = false, defaultValue = "") String condition) throws IOException, ClassNotFoundException { | 
|         ArrayList<Map> tree = treeUtils.getNodeTree(String.valueOf(getOriginNode().getId()), getHostId()); | 
|         // 深拷贝 | 
|         List<Map> result = ListUtils.deepCopy(tree); | 
|         if (!Cools.isEmpty(condition)) { | 
|             treeUtils.remove(condition, result); | 
|             treeUtils.remove(condition, result); | 
|         } | 
|         return R.ok(result); | 
|     } | 
|   | 
|     /*************************************** 数据相关 ***********************************************/ | 
|   | 
|     /** | 
|      * excel导入模板下载 | 
|      */ | 
|     @RequestMapping(value = "/node/excel/import/mould") | 
|     public void nodeExcelImportMould(HttpServletResponse response) throws IOException { | 
|         List<NodeExcel> excels = new ArrayList<>(); | 
|         response.setContentType("application/vnd.ms-excel"); | 
|         response.setCharacterEncoding("utf-8"); | 
|         String fileName = URLEncoder.encode("货位档案Excel导入模板", "UTF-8"); | 
|         response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx"); | 
|         EasyExcel.write(response.getOutputStream(), NodeExcel.class) | 
|                 .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) | 
|                 .sheet("sheet1") | 
|                 .doWrite(excels); | 
|     } | 
|   | 
|     // excel导入 | 
|     @PostMapping(value = "/node/excel/import/auth") | 
|     @ManagerAuth(memo = "货位档案数据导入") | 
|     @Transactional | 
|     public R nodeExcelImport(MultipartFile file) throws IOException { | 
|         NodeExcelListener listener = new NodeExcelListener(getUserId(), getHostId()); | 
|         EasyExcel.read(file.getInputStream(), NodeExcel.class, listener).sheet().doRead(); | 
|         return R.ok("成功同步"+listener.getTotal()+"个货位"); | 
|     } | 
|   | 
|     /*************************************** 入库出相关 ***********************************************/ | 
|   | 
|     @RequestMapping("/work/stock/pakin") | 
|     @ManagerAuth(memo = "入库") | 
|     public R stockPakin(@RequestBody PakinParam number) { | 
|         return nodeService.stockPakin(number, getUserId(), getHostId()); | 
|     } | 
|   | 
|     @RequestMapping("/work/stock/pakout") | 
|     @ManagerAuth(memo = "出库") | 
|     public R initPakout(@RequestBody List<InitPakoutParam> params) { | 
|         return nodeService.initPakout(params, getUserId(), getHostId()); | 
|     } | 
|   | 
|     @PostMapping(value = "/work/empty/stock") | 
|     public R getGroupEmptyStock(@RequestParam(required = false) String sourceLocNo) { | 
|         List<Node> allEmptys = nodeService.selectList(new EntityWrapper<Node>() | 
|                 .eq("type","3")); | 
|         return R.ok().add(allEmptys); | 
|     } | 
|     @RequestMapping("/work/move/start") | 
|     @ManagerAuth(memo = "库位移转") | 
|     public R locMoveStart(@RequestParam String sourceLocNo, | 
|                           @RequestParam String targetLocNo) { | 
|         nodeService.locMove(sourceLocNo, targetLocNo, getUserId()); | 
|         return R.ok("移库启动成功"); | 
|     } | 
|     @RequestMapping(value = "/node/select/{id}/auth") | 
|     @ManagerAuth | 
|     public R getById(@PathVariable("id") String id) { | 
|         Node node = nodeService.selectOne(new EntityWrapper<Node>() | 
|                 .eq("uuid",id)); | 
|         return R.ok(node); | 
|     } | 
|   | 
| } |