RDServiceImpl.java 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
  1. package com.goafanti.RD.service.impl;
  2. import com.goafanti.RD.bo.*;
  3. import com.goafanti.RD.service.RDService;
  4. import com.goafanti.common.bo.Result;
  5. import com.goafanti.common.constant.AFTConstants;
  6. import com.goafanti.common.dao.RdDetailsMapper;
  7. import com.goafanti.common.model.RdDetails;
  8. import com.goafanti.common.utils.DateUtils;
  9. import com.goafanti.common.utils.excel.FileUtils;
  10. import com.goafanti.common.utils.excel.NewExcelUtil;
  11. import com.goafanti.common.utils.pdf.PDFUtils;
  12. import com.goafanti.common.utils.word.WordUtils;
  13. import com.goafanti.core.mybatis.BaseMybatisDao;
  14. import com.goafanti.core.mybatis.page.Pagination;
  15. import com.goafanti.core.shiro.token.TokenManager;
  16. import com.itextpdf.text.*;
  17. import com.itextpdf.text.pdf.*;
  18. import com.lowagie.text.rtf.RtfWriter2;
  19. import org.springframework.beans.BeanUtils;
  20. import org.springframework.beans.factory.annotation.Autowired;
  21. import org.springframework.beans.factory.annotation.Value;
  22. import org.springframework.http.MediaType;
  23. import org.springframework.stereotype.Service;
  24. import org.springframework.web.multipart.MultipartFile;
  25. import javax.servlet.http.HttpServletResponse;
  26. import java.io.*;
  27. import java.util.*;
  28. import java.util.List;
  29. @Service
  30. public class RDServiceImpl extends BaseMybatisDao<RdDetailsMapper> implements RDService {
  31. @Autowired
  32. private RdDetailsMapper rdDetailsMapper;
  33. @Value(value = "${upload.path}")
  34. private final String uploadPath = null;
  35. @Override
  36. public int add(RdDetails in) {
  37. in.setAid(TokenManager.getAdminId());
  38. in.setStatus(0);
  39. return rdDetailsMapper.insertSelective(in);
  40. }
  41. @Override
  42. public int update(InputRdDetails in) {
  43. return rdDetailsMapper.updateByPrimaryKeySelective(in);
  44. }
  45. @Override
  46. public int updateBrowseOrDownload(Long id, Integer type) {
  47. return rdDetailsMapper.updateBrowseOrDownload(id,type);
  48. }
  49. @Override
  50. public boolean checkRdNo(String rdNo, Long id) {
  51. int i = rdDetailsMapper.checkRdNo(rdNo,id);
  52. if (i>0)return true;
  53. return false;
  54. }
  55. @Override
  56. public OutRdDetails details(Long id) {
  57. return rdDetailsMapper.details(id);
  58. }
  59. @Override
  60. public Pagination<OutExcelRdDetails> list(InputRdList in) {
  61. Map<String, Object> param = getParam(in);
  62. return (Pagination<OutExcelRdDetails>) findPage("RdDetailsList","RdDetailsCount",param,in.getPageNo(), in.getPageSize());
  63. }
  64. private Map<String, Object> getParam(InputRdList in) {
  65. Map<String,Object> param=new HashMap<>();
  66. param.put("RdStart", in.getRdStart());
  67. param.put("RdNo", in.getRdNo());
  68. param.put("RdEnd", in.getRdEnd());
  69. param.put("RdName", in.getRdName());
  70. if (TokenManager.hasRole(AFTConstants.SUPERADMIN)||TokenManager.hasRole(AFTConstants.APPROVAL_DECISION_AUDITOR)||TokenManager.hasRole(AFTConstants.CED)){
  71. }else {
  72. param.put("aid", TokenManager.getAdminId());
  73. }
  74. return param;
  75. }
  76. @Override
  77. public Result exportList(InputRdList in, String uploadPath) {
  78. Map<String, Object> param = getParam(in);
  79. List<OutExcelRdDetails> list = (List<OutExcelRdDetails>) findList("RdDetailsList", param, 1, 9999);
  80. NewExcelUtil<OutExcelRdDetails> newExcelUtil=new NewExcelUtil<>(OutExcelRdDetails.class);
  81. return newExcelUtil.exportExcel(list,"研发活动信息",uploadPath);
  82. }
  83. @Override
  84. public Object batchList(MultipartFile file) {
  85. NewExcelUtil<InputExcelRdDetails> newExcelUtil=new NewExcelUtil<>(InputExcelRdDetails.class);
  86. List<InputExcelRdDetails> list = null;
  87. try {
  88. list = newExcelUtil.importExcel(file.getInputStream(),2);
  89. } catch (Exception e) {
  90. e.printStackTrace();
  91. }
  92. //新增
  93. return insertBatchList(list);
  94. }
  95. private Map<String,Object> insertBatchList(List<InputExcelRdDetails> list) {
  96. String aid=TokenManager.getAdminId();
  97. Date date =new Date();
  98. Map<String,Object> map=new HashMap<>();
  99. List<InputExcelRdDetails> relust= new ArrayList<>();
  100. List<String>errors=new ArrayList<>();
  101. for (InputExcelRdDetails in : list) {
  102. boolean falg = checkRdNo(in.getRdNo(),null);
  103. if (falg){
  104. errors.add(in.getRdNo());
  105. }else {
  106. relust.add(in);
  107. }
  108. }
  109. for (InputExcelRdDetails in : relust) {
  110. in.setStatus(1);
  111. in.setAid(aid);
  112. in.setCreateTime(date);
  113. }
  114. if (relust.size()>0)insertBatch(relust);
  115. map.put("OK",relust.size());
  116. map.put("ERROR",errors);
  117. return map;
  118. }
  119. private void insertBatch(List<InputExcelRdDetails> relust) {
  120. List<InputExcelRdDetails> list=new ArrayList<>();
  121. for (int i = 0; i < relust.size(); i++) {
  122. list.add(relust.get(i));
  123. if (50 == list.size() || i == relust.size() - 1) {
  124. if (list.size() > 0) rdDetailsMapper.insertBatch(list);
  125. list.clear();
  126. }
  127. }
  128. }
  129. @Override
  130. public void pushDownloadDetailsWord(Long id, HttpServletResponse response) {
  131. OutRdDetails bo = rdDetailsMapper.details(id);
  132. // rdDetailsMapper.updateBrowseOrDownload(id, 1);
  133. OutWordRdDetails data =new OutWordRdDetails();
  134. BeanUtils.copyProperties(bo,data);
  135. if (bo.getRdStartStr()!=null&&bo.getRdEndStr()!=null){
  136. data.setStartEndTime(bo.getRdStartStr()+"~"+bo.getRdEndStr());
  137. }else {
  138. data.setStartEndTime("");
  139. }
  140. String tableName = data.getRdName();
  141. String realFileName = uploadPath+"/tmp/"+new Date().getTime() + ".doc";
  142. String attName = data.getRdName()+new Date().getTime() + ".doc";
  143. response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
  144. WordUtils wordUtils=new WordUtils();
  145. wordUtils.pushWord(data,response,tableName,attName,realFileName);
  146. }
  147. @Override
  148. public void pushDownloadDetailsPDF(Long id, HttpServletResponse response) {
  149. OutRdDetails bo = rdDetailsMapper.details(id);
  150. // rdDetailsMapper.updateBrowseOrDownload(id, 1);
  151. OutWordRdDetails data = new OutWordRdDetails();
  152. BeanUtils.copyProperties(bo, data);
  153. if (bo.getRdStartStr()!=null&&bo.getRdEndStr()!=null){
  154. data.setStartEndTime(bo.getRdStartStr()+"~"+bo.getRdEndStr());
  155. }else {
  156. data.setStartEndTime("");
  157. }
  158. String tableName = data.getRdName();
  159. String realFileName = uploadPath+"/tmp/"+new Date().getTime() + ".pdf";
  160. String attName = data.getRdName()+new Date().getTime() + ".pdf";
  161. response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
  162. PDFUtils pdfUtils=new PDFUtils();
  163. pdfUtils.pushPDF(data,response,tableName,attName,realFileName);
  164. }
  165. }