TestController.java 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. package com.fdkankan.fusion.controller;
  2. import cn.hutool.http.HttpUtil;
  3. import com.alibaba.fastjson.JSONObject;
  4. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  5. import com.deepoove.poi.XWPFTemplate;
  6. import com.deepoove.poi.data.*;
  7. import com.deepoove.poi.data.style.Style;
  8. import com.deepoove.poi.util.PoitlIOUtils;
  9. import com.dtflys.forest.annotation.Post;
  10. import com.fdkankan.fusion.common.ResultCode;
  11. import com.fdkankan.fusion.common.ResultData;
  12. import com.fdkankan.fusion.common.util.ShellUtil;
  13. import com.fdkankan.fusion.entity.*;
  14. import com.fdkankan.fusion.exception.BusinessException;
  15. import com.fdkankan.fusion.response.CaseInquestVo;
  16. import com.fdkankan.fusion.response.WitnessVo;
  17. import com.fdkankan.fusion.service.*;
  18. import com.fdkankan.redis.util.RedisUtil;
  19. import org.apache.commons.lang3.StringUtils;
  20. import org.apache.ibatis.annotations.Case;
  21. import org.springframework.beans.BeanUtils;
  22. import org.springframework.beans.factory.annotation.Autowired;
  23. import org.springframework.web.bind.annotation.*;
  24. import sun.java2d.pipe.SpanIterator;
  25. import javax.servlet.http.HttpServletRequest;
  26. import javax.servlet.http.HttpServletResponse;
  27. import javax.xml.transform.Result;
  28. import java.io.BufferedOutputStream;
  29. import java.io.IOException;
  30. import java.io.InputStream;
  31. import java.io.OutputStream;
  32. import java.util.ArrayList;
  33. import java.util.List;
  34. import java.util.concurrent.ThreadPoolExecutor;
  35. @RestController
  36. @RequestMapping("/test")
  37. public class TestController {
  38. @Autowired
  39. RedisUtil redisUtil;
  40. @Autowired
  41. IFusionMeterService meterService;
  42. @Autowired
  43. ITmProjectService tmProjectService;
  44. @Autowired
  45. ITmUserService tmUserService;
  46. @Autowired
  47. ICaseService caseService;
  48. @Autowired
  49. ICaseNumService caseNumService;
  50. @RequestMapping("/projectAddCase")
  51. public ResultData projectAddCase(){
  52. List<TmProject> list = tmProjectService.list();
  53. for (TmProject project : list) {
  54. LambdaQueryWrapper<CaseEntity> wrapper = new LambdaQueryWrapper<>();
  55. wrapper.eq(CaseEntity::getTmProjectId,project.getId());
  56. CaseEntity caseEntity = caseService.getOne(wrapper);
  57. if(caseEntity == null){
  58. caseEntity = new CaseEntity();
  59. caseEntity.setCaseTitle(project.getProjectName());
  60. caseEntity.setType(1);
  61. caseEntity.setTmProjectId(project.getId());
  62. caseService.save(caseEntity);
  63. }
  64. if(StringUtils.isNotBlank(project.getSceneNum())){
  65. CaseNumEntity caseNumEntity = new CaseNumEntity();
  66. caseNumEntity.setCaseId(caseEntity.getCaseId());
  67. caseNumEntity.setNumType(null);
  68. caseNumEntity.setNum(project.getSceneNum());
  69. if(project.getSceneNum().contains("KK-")){
  70. caseNumEntity.setNumType(0);
  71. }
  72. if(project.getSceneNum().contains("KJ-")){
  73. caseNumEntity.setNumType(1);
  74. }
  75. caseNumService.save(caseNumEntity);
  76. }
  77. }
  78. return ResultData.ok();
  79. }
  80. @Autowired
  81. IModelService modelService;
  82. @GetMapping("/updateModelDeptId")
  83. public ResultData updateModelDeptId(){
  84. List<Model> list = modelService.list();
  85. for (Model model : list) {
  86. if(model.getType() == 3){
  87. TmUser byUserName = tmUserService.getByUserName(model.getUserName());
  88. model.setDeptId(byUserName.getDeptId());
  89. modelService.updateById(model);
  90. }
  91. }
  92. return ResultData.ok();
  93. }
  94. @GetMapping("/updateCaseDeptId")
  95. public ResultData updateCaseDeptId(){
  96. List<CaseEntity> list = caseService.list();
  97. for (CaseEntity casee : list) {
  98. if(casee.getType() == 0){
  99. TmUser byUserName = tmUserService.getByUserName(casee.getUserName());
  100. casee.setDeptId(byUserName.getDeptId());
  101. caseService.updateById(casee);
  102. }
  103. }
  104. return ResultData.ok();
  105. }
  106. }