Ejemplo n.º 1
0
 /**
  * 执行批量数据导入的操作,如果勾选了isRewrite,则会先清除表中原有的数据,然后再重新插入,<br>
  * 若没有勾选,则直接执行插入操作,与原有的数据共同保存在数据库中。
  *
  * @param sheetContentDTO
  * @param br
  * @param attach
  * @param req
  * @param model
  * @return
  * @throws IOException
  */
 @RequestMapping(value = "/import/{cid}", method = RequestMethod.POST)
 public String importDatas(
     @Valid SheetContentDTO sheetContentDTO,
     BindingResult br,
     @RequestParam("attachs") MultipartFile attach,
     HttpServletRequest req,
     Model model) {
   model.addAttribute("casename", sheetContentDTO.getCaseName());
   model.addAttribute("caseid", sheetContentDTO.getCaseId());
   if (br.hasErrors()) return "data/importData";
   ExcelUtil excelUtil = new ExcelUtil();
   String realpath = req.getSession().getServletContext().getRealPath("/resources/upload");
   String errorInfo = null;
   if (attach.isEmpty()) errorInfo = "文件不能为空";
   if (!attach.getOriginalFilename().endsWith(".xlsx")
       && !attach.getOriginalFilename().endsWith(".xls")) errorInfo = "请选择[xlsx、xls]格式的数据文件";
   if (errorInfo != null) {
     model.addAttribute("errorInfo", errorInfo);
     return "data/importData";
   }
   String filePath =
       realpath
           + "/"
           + req.getSession().getId()
           + "."
           + Files.getFileExtension(attach.getOriginalFilename());
   ;
   File f = new File(filePath);
   SheetContentDTO sheetContent = null;
   try {
     FileUtils.copyInputStreamToFile(attach.getInputStream(), f);
     excelUtil.setImportExcelPath(filePath);
     sheetContent = excelUtil.getExcelContent(sheetContentDTO.getSheetName());
   } catch (Exception e) {
     errorInfo =
         "发生异常,请检查数据表的数据格式和指定的Sheet页名称是否正确," + e.getClass().getSimpleName() + "," + e.getMessage();
     model.addAttribute("errorInfo", errorInfo);
     f.delete();
     return "data/importData";
   }
   CaseBean caseBean = null;
   if (sheetContent.getCaseId() != null)
     caseBean = caseService.getCaseBean(Integer.parseInt(sheetContent.getCaseId()));
   else if (sheetContent.getCaseName() != null)
     caseBean = caseService.getCaseBeanByName(sheetContent.getCaseName());
   if (!sheetContentDTO.getCaseId().equals(caseBean.getId() + "")) {
     errorInfo = "当前用例是:[ " + sheetContentDTO.getCaseName() + " ] ,数据表中的数据不属于当前用例";
     model.addAttribute("errorInfo", errorInfo);
     f.delete();
     return "data/importData";
   }
   if (sheetContentDTO.getIsRewrite()) {
     inBatchDataService.deleteInBatchByCaseId(caseBean.getId());
     for (String[] params : sheetContent.getParameters()) {
       InterfaceBatchDataBean tempBatchBean =
           inBatchDataService.addInBatch(new InterfaceBatchDataBean(caseBean, params[0]));
       for (int i = 1; i < sheetContent.getParamNames().length; i++) {
         inDataBeanService.addDataBean(
             new InterfaceDataBean(tempBatchBean, sheetContent.getParamNames()[i], params[i]));
       }
     }
   } else {
     for (String[] params : sheetContent.getParameters()) {
       InterfaceBatchDataBean tempBatchBean =
           inBatchDataService.addInBatch(new InterfaceBatchDataBean(caseBean, params[0]));
       for (int i = 1; i < sheetContent.getParamNames().length; i++) {
         inDataBeanService.addDataBean(
             new InterfaceDataBean(tempBatchBean, sheetContent.getParamNames()[i], params[i]));
       }
     }
   }
   f.delete();
   return "redirect:/data/INTERFACE_CASE/list/" + caseBean.getId();
 }
Ejemplo n.º 2
0
 @RequestMapping(
     value = "/export/{id}",
     method = RequestMethod.POST,
     produces = "text/plain;charset=UTF-8")
 public @ResponseBody String exportData(
     @PathVariable Integer id, HttpServletRequest req, Model model) {
   CaseBean caseBean = caseService.getCaseBean(id);
   String fileName = GetNow.getCurrentTime("yyyyMMddHHmmss") + "_" + caseBean.getCaseName();
   String filePath =
       req.getSession().getServletContext().getRealPath("/resources/upload/")
           + "/"
           + fileName
           + ".xlsx";
   SheetContentDTO sheetDTO = new SheetContentDTO();
   ExcelUtil excelUtil = new ExcelUtil();
   List<String[]> parameters = new ArrayList<String[]>();
   List<InterfaceBatchDataBean> ibatchDataList = inBatchDataService.getInBatchList(id);
   sheetDTO.setCaseName(caseBean.getCaseName());
   sheetDTO.setCaseId(id + "");
   String[] paramName = null;
   for (InterfaceBatchDataBean batchData : ibatchDataList) {
     List<InterfaceDataBean> idataBeanList = inDataBeanService.getDataBeans(batchData.getId());
     String[] paramContent = new String[idataBeanList.size() + 1];
     paramName = new String[idataBeanList.size() + 1];
     paramName[0] = "期望值\\参数名";
     paramContent[0] = batchData.getExpectData();
     for (int i = 0; i < idataBeanList.size(); i++) {
       paramName[i + 1] = idataBeanList.get(i).getDataName();
       paramContent[i + 1] = idataBeanList.get(i).getDataContent();
     }
     parameters.add(paramContent);
   }
   sheetDTO.setParamNames(paramName);
   sheetDTO.setParameters(parameters);
   excelUtil.setExportExcelPath(filePath);
   try {
     excelUtil.exportExcel(sheetDTO);
     String fileUrl =
         "http://"
             + req.getServerName()
             + ":"
             + req.getServerPort()
             + "/phoenix_web/resources/upload/"
             + fileName
             + ".xlsx";
     return JSON.toJSONString(
         new AjaxObj(
             1,
             "导出接口用例[ "
                 + caseBean.getCaseName()
                 + " ]的数据成功!<br>点击下载:<a href='"
                 + fileUrl
                 + "'>"
                 + fileUrl
                 + "</a>",
             fileName));
   } catch (Exception e) {
     return JSON.toJSONString(
         new AjaxObj(0, "导出接口用例[ " + caseBean.getCaseName() + " ]的数据失败!<br>" + e.getCause()));
   }
 }