package com.zy.crm.manager.controller; 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.domain.KeyValueVo; import com.zy.crm.common.web.BaseController; import com.zy.crm.manager.entity.Plan; import com.zy.crm.manager.entity.Pri; import com.zy.crm.manager.entity.PriOnline2; import com.zy.crm.manager.service.PlanService; import com.zy.crm.manager.service.PriOnline2Service; import com.zy.crm.manager.service.PriService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.core.io.ClassPathResource; import org.springframework.util.ClassUtils; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.*; import java.text.SimpleDateFormat; import java.util.*; @RestController public class PriOnline2Controller extends BaseController { @Autowired private PriOnline2Service priOnline2Service; @Autowired private PriService priService; @Autowired private PlanService planService; @RequestMapping(value = "/priOnline2/{id}/auth") @ManagerAuth public R get(@PathVariable("id") String id) { return R.ok(priOnline2Service.selectById(String.valueOf(id))); } @RequestMapping(value = "/priOnline2/viewCheck/{id}/auth") @ManagerAuth(memo = "查看询价") public R viewCheck(@PathVariable("id") String id) { PriOnline2 priOnline = priOnline2Service.selectById(String.valueOf(id)); if (Cools.isEmpty(priOnline.getCheckData())) { return R.error("请先上传询价"); } return R.ok(priOnline); } @RequestMapping(value = "/priOnline2/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(required = false)String condition, @RequestParam Map param){ EntityWrapper wrapper = new EntityWrapper<>(); wrapper.setSqlSelect("id,title,create_time as createTime,filepath,item_id as itemId,order_num as orderNum,template_name as templateName,user_id as userId,dept_id as deptId,status,update_time as updateTime,check_data as checkData,update_user_id as updateUserId,member_id as memberId"); wrapper.in("member_id", getUserRoleBelongsToUserId("allopen")); excludeTrash(param); convert(param, wrapper); allLike(PriOnline2.class, param.keySet(), wrapper, condition); if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));} return R.ok(priOnline2Service.selectPage(new Page<>(curr, limit), wrapper)); } private void convert(Map map, EntityWrapper wrapper){ Long deptId = getDeptId(); boolean signUserId = false; boolean signDeptId = false; for (Map.Entry entry : map.entrySet()){ if (entry.getKey().equals("dept_id")){ signDeptId = true; } } for (Map.Entry 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 if (entry.getKey().equals("dept_id")){ if (!val.equals("19")){ wrapper.eq(entry.getKey(), val); } } else if (entry.getKey().equals("user_id") && !signDeptId){ signUserId = true; wrapper.eq(entry.getKey(), val); } else { wrapper.like(entry.getKey(), val); } } if (!signUserId && !signDeptId){ wrapper.eq("user_id", getUserId()); } } @RequestMapping(value = "/priOnline2/add/auth") @ManagerAuth(memo = "添加核价管理") public R add(@RequestBody Map map) { PriOnline2 priOnline2 = new PriOnline2(); priOnline2.setCreateTime(new Date()); priOnline2.setTitle(map.get("title").toString()); priOnline2.setTemplateName(map.get("title").toString()); priOnline2.setSheetData(map.get("sheetData").toString()); priOnline2.setItemId(Long.parseLong(map.get("itemId").toString())); SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmmss"); priOnline2.setOrderNum(format.format(new Date())); //创建人员 priOnline2.setUserId(getUserId()); //创建人员部门 priOnline2.setDeptId(getDeptId()); //更新时间 priOnline2.setUpdateTime(new Date()); //更新人员 priOnline2.setUpdateUserId(getUserId()); //状态,未完成 priOnline2.setStatus(0); //业务员 // Item item = itemService.selectById(priOnline.getItemId()); Plan plan = planService.selectById(priOnline2.getItemId()); priOnline2.setMemberId(plan.getUserId()); //设置项目流程 plan.setStep(2); planService.updateById(plan); priOnline2Service.insert(priOnline2); return R.ok(); } @RequestMapping(value = "/priOnline2/addOther/auth") @ManagerAuth(memo = "另存核价管理") public R addOther(@RequestBody Map map) { PriOnline2 online = priOnline2Service.selectById(Long.parseLong(map.get("id").toString())); PriOnline2 priOnline2 = new PriOnline2(); priOnline2.setCreateTime(new Date()); priOnline2.setTitle(map.get("title").toString()); priOnline2.setTemplateName(online.getTemplateName()); priOnline2.setSheetData(map.get("sheetData").toString()); priOnline2.setItemId(online.getItemId()); SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmmss"); priOnline2.setOrderNum(format.format(new Date())); //创建人员 priOnline2.setUserId(getUserId()); //更新时间 priOnline2.setUpdateTime(new Date()); //更新人员 priOnline2.setUpdateUserId(getUserId()); //状态,未完成 priOnline2.setStatus(0); priOnline2.setDeptId(getDeptId()); //业务员 // Item item = itemService.selectById(priOnline.getItemId()); Plan plan = planService.selectById(priOnline2.getItemId()); priOnline2.setMemberId(plan.getUserId()); priOnline2Service.insert(priOnline2); return R.ok(); } @RequestMapping(value = "/priOnline2/update/auth") @ManagerAuth(memo = "更新核价管理") public R update(@RequestBody Map map){ PriOnline2 priOnline2 = priOnline2Service.selectById(Long.parseLong(map.get("id").toString())); if (priOnline2.getStatus() == 1) { return R.error("核价已完成,禁止保存"); } priOnline2.setTitle(map.get("title").toString()); priOnline2.setSheetData(map.get("sheetData").toString()); //更新人员 priOnline2.setUpdateUserId(getUserId()); //更新时间 priOnline2.setUpdateTime(new Date()); priOnline2Service.updateById(priOnline2); return R.ok(); } //更新状态 @RequestMapping(value = "/priOnline2/updateForm/auth") @ManagerAuth(memo = "更新核价管理状态") public R updateForm(Long id,Integer status,String title,String templateName){ PriOnline2 priOnline2 = priOnline2Service.selectById(id); priOnline2.setStatus(status); priOnline2.setUpdateTime(new Date()); priOnline2.setTitle(title); priOnline2.setTemplateName(templateName); priOnline2Service.updateById(priOnline2); return R.ok(); } @RequestMapping(value = "/priOnline2/uploadCheck/auth") @ManagerAuth(memo = "上传询价文件") public R uploadCheck(@RequestParam("id") Integer id, @RequestParam("checkData") String checkData, @RequestParam("file") MultipartFile[] files){ PriOnline2 priOnline2 = priOnline2Service.selectById(id); if (priOnline2.getStatus() == 1) { return R.error("核价已完成,禁止上传"); } priOnline2.setCheckData(checkData); MultipartFile file = files[0]; SimpleDateFormat format = new SimpleDateFormat("yyyyMMdd_HHmmss"); String path = ClassUtils.getDefaultClassLoader().getResource("excel/uploadCheckData").getPath(); //文件后缀名 String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")); //上传文件名 String filename = format.format(new Date()) + suffix; //最终文件路径 String filepath = path + "/" + filename; //服务器端保存的文件对象 File serverFile = new File(filepath); if(!serverFile.exists()) { try { //创建文件 serverFile.createNewFile(); //将上传的文件写入到服务器端文件内 file.transferTo(serverFile); } catch (IOException e) { e.printStackTrace(); } } //保存文件名 priOnline2.setCheckDataFile(filename); priOnline2Service.updateById(priOnline2); return R.ok(); } @GetMapping("/priOnline2/checkDataDownload/{filename}") public void download(@PathVariable String filename, HttpServletResponse response) { try { ClassPathResource pathResource = new ClassPathResource("excel/uploadCheckData/" + filename); File file = pathResource.getFile(); InputStream inputStream = pathResource.getInputStream(); //输出文件 InputStream fis = new BufferedInputStream(inputStream); byte[] buffer = new byte[fis.available()]; fis.read(buffer); fis.close(); response.reset(); //获取文件的名字再浏览器下载页面 String name = file.getName(); response.addHeader("Content-Disposition", "attachment;filename=" + new String(name.getBytes(), "iso-8859-1")); response.addHeader("Content-Length", "" + file.length()); OutputStream out = new BufferedOutputStream(response.getOutputStream()); response.setContentType("application/octet-stream"); out.write(buffer); out.flush(); out.close(); } catch (Exception e) { e.printStackTrace(); } } @RequestMapping(value = "/priOnline2/delete/auth") @ManagerAuth(memo = "删除核价管理") public R delete(Long[] ids){ if (Cools.isEmpty(ids)){ return R.error(); } priOnline2Service.deleteBatchIds(Arrays.asList(ids)); return R.ok(); } @RequestMapping(value = "/priOnline2/export/auth") @ManagerAuth public R export(@RequestBody JSONObject param){ EntityWrapper wrapper = new EntityWrapper<>(); List fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class); Map map = excludeTrash(param.getJSONObject("priOnline")); convert(map, wrapper); List list = priOnline2Service.selectList(wrapper); return R.ok(exportSupport(list, fields)); } @RequestMapping(value = "/priOnline2Query/auth") @ManagerAuth public R query(String condition) { EntityWrapper wrapper = new EntityWrapper<>(); wrapper.like("order_num", condition); wrapper.in("member_id", getUserRoleBelongsToUserId("allopen")); Page page = priOnline2Service.selectPage(new Page<>(0, 10), wrapper); List> result = new ArrayList<>(); for (PriOnline2 priOnline2 : page.getRecords()){ Map map = new HashMap<>(); map.put("id", priOnline2.getId()); map.put("value", priOnline2.getOrderNum() + "/" + priOnline2.getPlanId$() + "/" + priOnline2.getMemberId$()); result.add(map); } return R.ok(result); } @RequestMapping(value = "/priOnline2/check/column/auth") @ManagerAuth public R query(@RequestBody JSONObject param) { Wrapper wrapper = new EntityWrapper().eq(humpToLine(String.valueOf(param.get("key"))), param.get("val")); if (null != priOnline2Service.selectOne(wrapper)){ return R.parse(BaseRes.REPEAT).add(getComment(PriOnline2.class, String.valueOf(param.get("key")))); } return R.ok(); } @RequestMapping("/priOnline2/all/get/kv") @ManagerAuth public R getDataKV(@RequestParam(required = false) String condition) { List vos = new ArrayList<>(); Wrapper wrapper = new EntityWrapper().andNew().like("id", condition).orderBy("create_time", false); priOnline2Service.selectPage(new Page<>(1, 30), wrapper).getRecords().forEach(item -> vos.add(new KeyValueVo(String.valueOf(item.getId()), item.getId()))); return R.ok().add(vos); } }