Administrator
2021-08-05 0dff0b23d5d37cee7b72093f135c26c657201bd6
src/main/java/org/springblade/modules/training/controller/TrainingRegistrationController.java
@@ -4,17 +4,39 @@
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.AllArgsConstructor;
import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;
import org.springblade.common.utils.arg;
import org.springblade.core.excel.util.ExcelUtil;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.mp.support.Query;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.apply.entity.Apply;
import org.springblade.modules.apply.excel.ApplyImporter;
import org.springblade.modules.apply.excel.ApplyInfoExcel;
import org.springblade.modules.exam.entity.ExamPaper;
import org.springblade.modules.exam.entity.ExamScore;
import org.springblade.modules.exam.service.ExamPaperService;
import org.springblade.modules.training.entity.TrainExam;
import org.springblade.modules.training.entity.TrainingRegistration;
import org.springblade.modules.training.excel.TrainingRegistrationImporter;
import org.springblade.modules.training.service.TrainExamService;
import org.springblade.modules.training.service.TrainingRegistrationService;
import org.springblade.modules.training.vo.TrainingRegistrationVo;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.UUID;
/**
 * @author zhongrj
@@ -27,6 +49,10 @@
public class TrainingRegistrationController {
   private final TrainingRegistrationService trainingRegistrationService;
   private final TrainExamService trainExamService;
   private final ExamPaperService examPaperService;
   /**
    * 自定义分页
@@ -65,14 +91,27 @@
    */
   @PostMapping("/submit")
   public R submit(@RequestBody TrainingRegistration trainingRegistration) throws Exception {
      if (null!=trainingRegistration.getId()){
      if (null==trainingRegistration.getId()){
         //查询是否已报名
         Integer registration = trainingRegistrationService.getTrainingRegistrationInfo(trainingRegistration);
         if (null!=registration){
            return R.data(201,"已报名,不能重复报名");
         }
         trainingRegistration.setTrainingTime(new Date());
         trainingRegistration.setCancel(1);
         //默认报名未考试状态
         trainingRegistration.setIsExam(1);
         //去生成准考证号码
         trainingRegistration.setCandidateNo(getCandidateNo(trainingRegistration));
         //去生成考试编号
//         trainingRegistration.setApplyCode(getApplyCode(trainingRegistration));
      }
      arg.test01(arg.url+"/trainingRegistration/submit",trainingRegistration);
      return R.status(trainingRegistrationService.saveOrUpdate(trainingRegistration));
//      arg.test01(arg.url+"/trainingRegistration/submit",trainingRegistration);
      boolean status = trainingRegistrationService.saveOrUpdate(trainingRegistration);
      if (status){
         return R.data(200,"报名成功");
      }
      return R.data(202,"报名失败");
   }
   /**
@@ -96,4 +135,161 @@
      return R.data(detail);
   }
   /**
    * 导出报名数数据
    * @param response
    * @param trainingRegistration 培训报名对象
    */
   @GetMapping("export-apply")
   @ApiOperation(value = "导出报名清册")
   public void exportApply(HttpServletResponse response,TrainingRegistrationVo trainingRegistration) {
      List<ApplyInfoExcel> list = trainingRegistrationService.getTrainingRegistrationExcelList(trainingRegistration);
      ExcelUtil.export(response, "考试报名清册数据"+ DateUtil.time(), "考试报名清册数据表", list, ApplyInfoExcel.class);
   }
   /**
    * 导入报名考试数据
    */
   @PostMapping("import-trainingRegistration")
   @ApiOperation(value = "导入报名考试数据", notes = "传入excel")
   public R importUser(MultipartFile file, Integer isCovered) {
      TrainingRegistrationImporter trainingRegistrationImporter = new TrainingRegistrationImporter(trainingRegistrationService, false);
      ExcelUtil.save(file, trainingRegistrationImporter, ApplyInfoExcel.class);
      return R.success("操作成功");
   }
   /**
    * 生成准考证号码
    * @param trainingRegistration 考试报名信息对象
    */
   private String getCandidateNo(TrainingRegistration trainingRegistration) {
      //获取考试信息
      ExamPaper examPaper = examPaperService.getById(trainingRegistration.getTrainExamId());
      if (null!=examPaper.getStartTime()){
         String format = new SimpleDateFormat("yyyy-MM-dd").format(examPaper.getStartTime());
         String year = format.substring(2,4);
         String quarter  = null;
         String months = null;
         String days = null;
         int month = Integer.parseInt(format.substring(5,7));
         int day = Integer.parseInt(format.substring(8,10));
         if (month>0 && month<=3){
            quarter = "C";
         }
         if (month>3 && month<=6){
            quarter = "X";
         }
         if (month>6 && month<=9){
            quarter = "Q";
         }
         if (month>9 && month<=12){
            quarter = "D";
         }
         if (month<=9){
            months = "0" + month;
         }
         if (day<=9){
            days = "0" + day;
         }
         //获取考试名称前缀,去除数字,字母
         String examName
            = examPaper.getExamName().replaceAll("\\s*", "").replaceAll("[^(\\u4e00-\\u9fa5)]", "").substring(0,1);
         //前缀 = 年的最后两位  + 月份(两位) + 考试名称(中文拼音)首字母(去除数字,字母) + 考试类型 + 季度拼音首字母大写(春季就是 C)
         String result = year
            + months
            + toFirstChar(examName).toUpperCase()
            + examPaper.getExamType()
            + quarter;
         //查询是当前前缀已生成的数量
         int count = trainingRegistrationService.getCandidateNoCount(result);
         if (count==0){
            return result + "0000";
         }
         //格式化
         DecimalFormat decimalFormat = new DecimalFormat("0000");
         //返回
         return result + (decimalFormat.format(count++));
      }
      return null;
   }
   /**
    * 生成报名编号
    * @param apply 考试报名信息对象
    */
   private String getApplyCode(Apply apply) {
      //获取考试信息
      ExamPaper examPaper = examPaperService.getById(apply.getExamId());
      if (null!=examPaper.getStartTime()){
         String format = new SimpleDateFormat("yyyy-MM-dd").format(examPaper.getStartTime());
         String year = format.substring(2,4);
         String quarter  = null;
         String months = null;
         String days = null;
         int month = Integer.parseInt(format.substring(5,7));
         int day = Integer.parseInt(format.substring(8,10));
         if (month>0 && month<=3){
            quarter = "C";
         }
         if (month>3 && month<=6){
            quarter = "X";
         }
         if (month>6 && month<=9){
            quarter = "Q";
         }
         if (month>9 && month<=12){
            quarter = "D";
         }
         if (month<=9){
            months = "0" + month;
         }
         if (day<=9){
            days = "0" + day;
         }
         //获取考试名称前缀,去除数字,字母
         String examName
            = examPaper.getExamName().replaceAll("\\s*", "").replaceAll("[^(\\u4e00-\\u9fa5)]", "").substring(0,1);
         //前缀 = 年的最后两位  + 月份(两位) + 考试名称(中文拼音)首字母(去除数字,字母) + 考试类型 + 季度拼音首字母大写(春季就是 C)
         String result = year
            + months
            + toFirstChar(examName).toUpperCase()
            + examPaper.getExamType()
            + quarter;
         //生成随机数
         UUID uuid = UUID.randomUUID();
         //返回
         return result + uuid.toString().replaceAll("\\-","");
      }
      return null;
   }
   /**
    * 获取字符串拼音的第一个字母
    * @param chinese
    * @return
    */
   private String toFirstChar(String chinese){
      String pinyinStr = "";
      char[] newChar = chinese.toCharArray();
      //转为单个字符
      HanyuPinyinOutputFormat defaultFormat = new HanyuPinyinOutputFormat();
      defaultFormat.setCaseType(HanyuPinyinCaseType.LOWERCASE);
      defaultFormat.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
      for (int i = 0; i < newChar.length; i++) {
         if (newChar[i] > 128) {
            try {
               pinyinStr += PinyinHelper.toHanyuPinyinStringArray(newChar[i], defaultFormat)[0].charAt(0);
            } catch (BadHanyuPinyinOutputFormatCombination e) {
               e.printStackTrace();
            }
         }else{
            pinyinStr += newChar[i];
         }
      }
      return pinyinStr;
   }
}