|
@@ -2,6 +2,7 @@ package com.ym.mec.web.controller;
|
|
|
|
|
|
import com.ym.mec.biz.dal.dao.QuestionnaireQuestionDao;
|
|
|
import com.ym.mec.biz.dal.entity.QuestionnaireQuestion;
|
|
|
+import com.ym.mec.biz.dal.enums.QuestionnaireActiveTypeEnum;
|
|
|
import com.ym.mec.biz.dal.page.QuestionnaireUserResultQueryInfo;
|
|
|
import com.ym.mec.biz.service.QuestionnaireUserResultService;
|
|
|
import com.ym.mec.common.controller.BaseController;
|
|
@@ -16,6 +17,7 @@ import org.springframework.security.access.prepost.PreAuthorize;
|
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
+
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
import java.io.IOException;
|
|
|
import java.io.OutputStream;
|
|
@@ -42,7 +44,7 @@ public class QuestionnaireUserResultController extends BaseController {
|
|
|
@ApiOperation(value = "导出")
|
|
|
@RequestMapping("/export")
|
|
|
@PreAuthorize("@pcs.hasPermissions('questionnaireUserResult/export')")
|
|
|
- public void questionnaireUserResultExport(Integer cooperationId, HttpServletResponse response) throws Exception {
|
|
|
+ public void questionnaireUserResultExport(QuestionnaireActiveTypeEnum activeType, Integer cooperationId, HttpServletResponse response) throws Exception {
|
|
|
List<QuestionnaireQuestion> questionList = questionnaireQuestionDao.queryQuestionList(cooperationId);
|
|
|
questionList.removeAll(Collections.singleton(null));
|
|
|
if(questionList != null && questionList.size() > 0){
|
|
@@ -54,7 +56,57 @@ public class QuestionnaireUserResultController extends BaseController {
|
|
|
List<String> collect = questionList.stream().map(e -> e.getContent()).collect(Collectors.toList());
|
|
|
column.addAll(collect);
|
|
|
String[] header = column.toArray(new String[column.size()]);
|
|
|
- Map<String,Object> resultMap = questionnaireUserResultService.questionnaireUserResultExport(cooperationId,questionList);
|
|
|
+ Map<String,Object> resultMap = questionnaireUserResultService.questionnaireUserResultExport(activeType.getCode(),cooperationId,questionList);
|
|
|
+ List<Map<String,Object>> jsonObjectList = (List<Map<String,Object>>)resultMap.get("jsonObjectList");
|
|
|
+ if(jsonObjectList.size() == 0){
|
|
|
+ throw new BizException("暂无可导出数据");
|
|
|
+ }
|
|
|
+ OutputStream outputStream = response.getOutputStream();
|
|
|
+ HSSFWorkbook workbook = null;
|
|
|
+ try {
|
|
|
+ Set<String> body1 = (Set<String>)resultMap.get("body");
|
|
|
+ String[] body = body1.toArray(new String[body1.size()]);
|
|
|
+ workbook = POIUtil.exportExcel(header, body, jsonObjectList);
|
|
|
+ response.setContentType("application/octet-stream");
|
|
|
+ response.setHeader("Content-Disposition", "attachment;filename=replacement-" + DateUtil.getDate(new Date()) + ".xls");
|
|
|
+ response.flushBuffer();
|
|
|
+ outputStream = response.getOutputStream();
|
|
|
+ workbook.write(outputStream);
|
|
|
+ outputStream.flush();
|
|
|
+ workbook.close();
|
|
|
+ } catch (Exception e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ } finally {
|
|
|
+ if (outputStream != null) {
|
|
|
+ try {
|
|
|
+ workbook.close();
|
|
|
+ outputStream.close();
|
|
|
+ } catch (IOException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }else {
|
|
|
+ throw new BizException("暂无可导出数据");
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ @ApiOperation(value = "导出")
|
|
|
+ @RequestMapping("/export")
|
|
|
+ @PreAuthorize("@pcs.hasPermissions('questionnaireUserResult/export1')")
|
|
|
+ public void questionnaireUserResultExport1(QuestionnaireActiveTypeEnum activeType, Integer cooperationId, HttpServletResponse response) throws Exception {
|
|
|
+ List<QuestionnaireQuestion> questionList = questionnaireQuestionDao.findByActiveType(activeType.getCode());
|
|
|
+ questionList.removeAll(Collections.singleton(null));
|
|
|
+ if(questionList != null && questionList.size() > 0){
|
|
|
+ List<String> column = new ArrayList<>();
|
|
|
+ column.add("活动编号");
|
|
|
+ column.add("学员编号");
|
|
|
+ column.add("学员姓名");
|
|
|
+ column.add("手机号");
|
|
|
+ List<String> collect = questionList.stream().map(e -> e.getContent()).collect(Collectors.toList());
|
|
|
+ column.addAll(collect);
|
|
|
+ String[] header = column.toArray(new String[column.size()]);
|
|
|
+ Map<String,Object> resultMap = questionnaireUserResultService.questionnaireUserResultExport(activeType.getCode(),cooperationId,questionList);
|
|
|
List<Map<String,Object>> jsonObjectList = (List<Map<String,Object>>)resultMap.get("jsonObjectList");
|
|
|
if(jsonObjectList.size() == 0){
|
|
|
throw new BizException("暂无可导出数据");
|