package com.jsh.erp.controller; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.jsh.erp.datasource.entities.Depot; import com.jsh.erp.service.depot.DepotService; import com.jsh.erp.service.userBusiness.UserBusinessService; import com.jsh.erp.utils.BaseResponseInfo; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.dao.DataAccessException; import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; import javax.servlet.http.HttpServletRequest; import java.io.IOException; import java.util.List; @RestController @RequestMapping(value = "/depot") public class DepotController { private Logger logger = LoggerFactory.getLogger(DepotController.class); @Resource private DepotService depotService; @Resource private UserBusinessService userBusinessService; @GetMapping(value = "/getAllList") public BaseResponseInfo getAllList(HttpServletRequest request) { BaseResponseInfo res = new BaseResponseInfo(); try { List depotList = depotService.getAllList(); res.code = 200; res.data = depotList; } catch(Exception e){ e.printStackTrace(); res.code = 500; res.data = "获取数据失败"; } return res; } /** * 用户对应仓库显示 * @param type * @param keyId * @param request * @return */ @PostMapping(value = "/findUserDepot") public JSONArray findUserDepot(@RequestParam("UBType") String type, @RequestParam("UBKeyId") String keyId, HttpServletRequest request) { JSONArray arr = new JSONArray(); try { List dataList = depotService.findUserDepot(); //开始拼接json数据 JSONObject outer = new JSONObject(); outer.put("id", 1); outer.put("text", "仓库列表"); outer.put("state", "open"); //存放数据json数组 JSONArray dataArray = new JSONArray(); if (null != dataList) { for (Depot depot : dataList) { JSONObject item = new JSONObject(); item.put("id", depot.getId()); item.put("text", depot.getName()); //勾选判断1 Boolean flag = false; try { flag = userBusinessService.checkIsUserBusinessExist(type, keyId, "[" + depot.getId().toString() + "]"); } catch (Exception e) { logger.error(">>>>>>>>>>>>>>>>>设置用户对应的仓库:类型" + type + " KeyId为: " + keyId + " 存在异常!"); } if (flag == true) { item.put("checked", true); } //结束 dataArray.add(item); } } outer.put("children", dataArray); arr.add(outer); } catch (Exception e) { e.printStackTrace(); } return arr; } @GetMapping(value = "/findDepotByUserId") public JSONArray findDepotByUserId(@RequestParam("UBType") String type, @RequestParam("UBKeyId") String keyId, HttpServletRequest request) { JSONArray arr = new JSONArray(); try { List dataList = depotService.findUserDepot(); //开始拼接json数据 if (null != dataList) { for (Depot depot : dataList) { JSONObject item = new JSONObject(); //勾选判断1 Boolean flag = false; try { flag = userBusinessService.checkIsUserBusinessExist(type, keyId, "[" + depot.getId().toString() + "]"); } catch (DataAccessException e) { logger.error(">>>>>>>>>>>>>>>>>查询用户对应的仓库:类型" + type + " KeyId为: " + keyId + " 存在异常!"); } if (flag == true) { item.put("id", depot.getId()); item.put("depotName", depot.getName()); arr.add(item); } } } } catch (Exception e) { e.printStackTrace(); } return arr; } /** * 查找礼品卡-虚拟仓库 * @param type * @param request * @return */ @PostMapping(value = "/findGiftByType") public JSONArray findGiftByType(@RequestParam("type") Integer type, HttpServletRequest request) { JSONArray arr = new JSONArray(); try { List dataList = depotService.findGiftByType(type); //存放数据json数组 if (null != dataList) { for (Depot depot : dataList) { JSONObject item = new JSONObject(); item.put("id", depot.getId()); //仓库名称 item.put("name", depot.getName()); arr.add(item); } } } catch (Exception e) { logger.error(">>>>>>>>>查找仓库信息异常", e); } return arr; } }