RepairClassesSettingsController.java 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717
  1. package com.repair.controller;
  2. import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
  3. import com.repair.api.RepairClassesSettingsControllerAPI;
  4. import com.repair.common.utils.ExcelExportUtil;
  5. import com.repair.common.utils.ExcelUtils;
  6. import com.repair.common.utils.TimeExchange;
  7. import com.repair.common.utils.paramUtils;
  8. import com.repair.model.enumModel.eFileType;
  9. import com.repair.model.enumModel.eSchool;
  10. import com.repair.model.pojo.RepairClassesSettings;
  11. import com.repair.model.pojo.RepairShiftSettings;
  12. import com.repair.model.pojo.RepairUser;
  13. import com.repair.model.request.insertRepairClassesRequest;
  14. import com.repair.model.request.updateRepairClassesRequest;
  15. import com.repair.model.result.CommonResult;
  16. import com.repair.model.result.PageUtils;
  17. import com.repair.model.vo.*;
  18. import com.repair.services.RepairClassesSettingsService;
  19. import com.repair.services.RepairShiftSettingsService;
  20. import com.repair.services.RepairUserService;
  21. import org.apache.commons.lang3.StringUtils;
  22. import org.apache.poi.hssf.usermodel.HSSFRow;
  23. import org.apache.poi.hssf.usermodel.HSSFSheet;
  24. import org.apache.poi.hssf.usermodel.HSSFWorkbook;
  25. import org.apache.poi.ss.usermodel.CellType;
  26. import org.apache.poi.xssf.usermodel.XSSFRow;
  27. import org.apache.poi.xssf.usermodel.XSSFSheet;
  28. import org.apache.poi.xssf.usermodel.XSSFWorkbook;
  29. import org.springframework.beans.factory.annotation.Autowired;
  30. import org.springframework.validation.BindingResult;
  31. import org.springframework.web.bind.annotation.RestController;
  32. import org.springframework.web.multipart.MultipartFile;
  33. import javax.servlet.http.HttpServletResponse;
  34. import java.io.IOException;
  35. import java.io.InputStream;
  36. import java.lang.reflect.InvocationTargetException;
  37. import java.text.ParseException;
  38. import java.text.SimpleDateFormat;
  39. import java.time.Instant;
  40. import java.time.LocalDate;
  41. import java.time.LocalDateTime;
  42. import java.time.ZoneId;
  43. import java.time.format.DateTimeFormatter;
  44. import java.util.*;
  45. import java.util.stream.Collectors;
  46. /**
  47. * <p>
  48. * 前端控制器
  49. * </p>
  50. *
  51. * @author ceshi
  52. * @since 2023-07-13
  53. */
  54. @RestController
  55. public class RepairClassesSettingsController implements RepairClassesSettingsControllerAPI {
  56. @Autowired
  57. private RepairUserService repairUserService;
  58. @Autowired
  59. private RepairShiftSettingsService repairShiftSettingsService;
  60. @Autowired
  61. private RepairClassesSettingsService repairClassesSettingsService;
  62. /**
  63. * 新增排班数据
  64. * @param ircr 排版数据
  65. * @param bindingResult 是否为空判断
  66. * @return
  67. * @throws ParseException
  68. */
  69. @Override
  70. public CommonResult InsertRepairClassesSettings(insertRepairClassesRequest ircr, BindingResult bindingResult) throws ParseException {
  71. if (bindingResult.hasErrors()) {
  72. String st = paramUtils.getParamError(bindingResult);
  73. return CommonResult.fail(st);
  74. }
  75. if (ircr.getStartDate() == null) {
  76. return CommonResult.fail("开始日期不能为空");
  77. }
  78. //查找该用户是否已存储过当天排班
  79. DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  80. String dateString = ircr.getStartDate().format(formatter);
  81. int shiftDate = repairClassesSettingsService.getRepairByUserId(ircr.getUserId(),dateString);
  82. if(shiftDate > 0){
  83. return CommonResult.fail("该用户"+dateString+"日已排班,请刷新重试!");
  84. }
  85. //防止选择的班次重叠
  86. List<RepairShiftSettings> datas = repairShiftSettingsService.getRepairsByShiftIds(ircr.getShiftIds());
  87. for (int i = 0; i < datas.size(); i++) {
  88. for (int j = i + 1; j < datas.size(); j++) {
  89. boolean isOver = TimeExchange.overlapping(datas.get(i).getStartTime(), datas.get(i).getEndTime(), datas.get(j).getStartTime(), datas.get(j).getEndTime());
  90. if (isOver) {
  91. return CommonResult.fail(datas.get(i).getStartTime() + "-" + datas.get(i).getEndTime() + "时间段和" + datas.get(j).getStartTime() + "-" + datas.get(j).getEndTime() + "时间段存在重叠");
  92. }
  93. }
  94. }
  95. RepairClassesSettings rcs = new RepairClassesSettings();
  96. rcs.setUserId(ircr.getUserId());
  97. rcs.setShiftId(StringUtils.join(ircr.getShiftIds(), ','));
  98. rcs.setStartDate(ircr.getStartDate());
  99. int result = repairClassesSettingsService.insertRepairClassesSettings(rcs);
  100. return result > 0 ? CommonResult.ok("添加成功") : CommonResult.fail("添加失败");
  101. }
  102. /**
  103. * 根据ID更新排班数据
  104. * @param urcr 排班数据
  105. * @param bindingResult 是否为空判断
  106. * @return
  107. * @throws ParseException
  108. */
  109. @Override
  110. public CommonResult updateRepairClassesSettingsById(updateRepairClassesRequest urcr, BindingResult bindingResult) throws ParseException {
  111. if (bindingResult.hasErrors()) {
  112. String st = paramUtils.getParamError(bindingResult);
  113. return CommonResult.fail(st);
  114. }
  115. if (urcr.getStartDate() == null) {
  116. return CommonResult.fail("开始日期不能为空");
  117. }
  118. RepairClassesSettings rcs = repairClassesSettingsService.getRepairById(urcr.getId());
  119. if (rcs == null) {
  120. return CommonResult.fail("当前数据不存在,修改失败!");
  121. }
  122. LocalDate dateNow = LocalDate.now();
  123. //小于等于当天时间的数据不能修改
  124. if(rcs.getStartDate().isBefore(dateNow) || rcs.getStartDate().isEqual(dateNow)){
  125. return CommonResult.fail("当天及当天以前的排班数据无法进行修改操作!");
  126. }
  127. //防止选择的班次重叠
  128. List<RepairShiftSettings> datas = repairShiftSettingsService.getRepairsByShiftIds(urcr.getShiftIds());
  129. for (int i = 0; i < datas.size(); i++) {
  130. for (int j = i + 1; j < datas.size(); j++) {
  131. boolean isOver = TimeExchange.overlapping(datas.get(i).getStartTime(), datas.get(i).getEndTime(), datas.get(j).getStartTime(), datas.get(j).getEndTime());
  132. if (isOver) {
  133. return CommonResult.fail(datas.get(i).getStartTime() + "-" + datas.get(i).getEndTime() + "时间段和" + datas.get(j).getStartTime() + "-" + datas.get(j).getEndTime() + "时间段存在重叠");
  134. }
  135. }
  136. }
  137. rcs.setId(urcr.getId());
  138. rcs.setUserId(urcr.getUserId());
  139. rcs.setShiftId(StringUtils.join(urcr.getShiftIds(), ','));
  140. rcs.setStartDate(urcr.getStartDate());
  141. int result = repairClassesSettingsService.updateRepairClassesSettings(rcs);
  142. return result > 0 ? CommonResult.ok("修改成功") : CommonResult.fail("修改失败");
  143. }
  144. /**
  145. * 查询排班分页数据
  146. * @param currentPage 当前页数
  147. * @param pageCount 一页数据条数
  148. * @param startTime 开始时间
  149. * @param endTime 结束时间
  150. * @return
  151. */
  152. @Override
  153. public CommonResult queryPageRepairClassesSettingss(int currentPage, int pageCount, String startTime, String endTime) {
  154. PageUtils<RepairUser> result = repairUserService.queryPageRepairUser(currentPage, pageCount);
  155. List<Integer> userIds = result.getList().stream().map(RepairUser::getId).collect(Collectors.toList());
  156. if (startTime == null || endTime == null) {
  157. String[] weeks = TimeExchange.getCurrentWeekTimeFrame();
  158. startTime = weeks[0];
  159. endTime = weeks[1];
  160. }
  161. LocalDate StartDate = LocalDate.parse(startTime, DateTimeFormatter.ofPattern("yyyy-MM-dd"));
  162. LocalDate EndDate = LocalDate.parse(endTime, DateTimeFormatter.ofPattern("yyyy-MM-dd"));
  163. List<ClassSettingDateVo> dates = TimeExchange.getDatesBetween(StartDate, EndDate);
  164. try {
  165. startTime = startTime == null ? startTime : TimeExchange.getStartOfDayStr(TimeExchange.StringToDate(startTime, "yyyy-MM-dd"));
  166. endTime = endTime == null ? endTime : TimeExchange.getEndOfDayStr(TimeExchange.StringToDate(endTime, "yyyy-MM-dd"));
  167. } catch (Exception e) {
  168. }
  169. List<ClassesSettingVo> datas = repairClassesSettingsService.getRepairByUserIds(userIds, startTime, endTime);
  170. List<ClassesSettingPageDataVo> cspds = new ArrayList<>();
  171. for (RepairUser ru : result.getList()) {
  172. ClassesSettingPageDataVo cspd = new ClassesSettingPageDataVo();
  173. cspd.setSchoolId(ru.getSchoolId());
  174. cspd.setSchoolName(eSchool.stringOf(ru.getSchoolId()));
  175. cspd.setUserId(ru.getId());
  176. cspd.setUserName(ru.getUserName());
  177. List<ClassesSettingVo> classDatas = datas.stream().filter(e -> e.getUserId() == ru.getId()).collect(Collectors.toList());
  178. for (ClassSettingDateVo d : dates) {
  179. Optional<ClassesSettingVo> o = classDatas.stream().filter(e -> e.getStartDate().equals(d.getDateStr())).findFirst();
  180. if (!(o != null && o.isPresent())) {
  181. ClassesSettingVo newData = new ClassesSettingVo();
  182. newData.setStartDate(d.getDateStr());
  183. newData.setUserId(ru.getId());
  184. classDatas.add(newData);
  185. }
  186. }
  187. classDatas.sort(Comparator.comparing(ClassesSettingVo::getStartDate));
  188. cspd.setClassSettings(classDatas);
  189. cspds.add(cspd);
  190. }
  191. ClassesSettingPageVo data = new ClassesSettingPageVo();
  192. data.setDates(dates);
  193. PageUtils ssss = new PageUtils(cspds, result.getTotalCount(), result.getTotalPage(), result.getCurrPage());
  194. data.setPageDatas(ssss);
  195. return CommonResult.ok(data);
  196. }
  197. /**
  198. * 排班数据导出
  199. * @param startTime 开始时间
  200. * @param endTime 结束时间
  201. * @param response 响应流
  202. */
  203. @Override
  204. public void downloadRepairClassesExcel(String startTime, String endTime, HttpServletResponse response){
  205. List<RepairUser> result = repairUserService.queryRepairUsers();
  206. List<Integer> userIds = result.stream().map(RepairUser::getId).collect(Collectors.toList());
  207. if (startTime == null || endTime == null) {
  208. String[] weeks = TimeExchange.getCurrentWeekTimeFrame();
  209. startTime = weeks[0];
  210. endTime = weeks[1];
  211. }
  212. LocalDate StartDate = LocalDate.parse(startTime, DateTimeFormatter.ofPattern("yyyy-MM-dd"));
  213. LocalDate EndDate = LocalDate.parse(endTime, DateTimeFormatter.ofPattern("yyyy-MM-dd"));
  214. List<ClassSettingDateVo> dates = TimeExchange.getDatesBetween(StartDate, EndDate);
  215. try {
  216. startTime = startTime == null ? startTime : TimeExchange.getStartOfDayStr(TimeExchange.StringToDate(startTime, "yyyy-MM-dd"));
  217. endTime = endTime == null ? endTime : TimeExchange.getEndOfDayStr(TimeExchange.StringToDate(endTime, "yyyy-MM-dd"));
  218. } catch (Exception e) {
  219. }
  220. List<ClassesSettingVo> datas = repairClassesSettingsService.getRepairByUserIds(userIds, startTime, endTime);
  221. List<Map<String, Object>> list = new ArrayList<>();
  222. for (RepairUser ru : result) {
  223. Map<String, Object> map = new LinkedHashMap<>();
  224. map.put("名称", ru.getUserName());
  225. List<ClassesSettingVo> classDatas = datas.stream().filter(e -> e.getUserId() == ru.getId()).collect(Collectors.toList());
  226. for (ClassSettingDateVo d : dates) {
  227. Optional<ClassesSettingVo> o = classDatas.stream().filter(e -> e.getStartDate().equals(d.getDateStr())).findFirst();
  228. if (!(o != null && o.isPresent())) {
  229. map.put(d.getDateStr() + "(" + d.getWeekStr() + ")", "");
  230. } else {
  231. map.put(d.getDateStr() + "(" + d.getWeekStr() + ")", o.get().getShiftName());
  232. }
  233. }
  234. list.add(map);
  235. }
  236. ExcelExportUtil.createExcel(list, "testExcel.xlsx", "排班文档", response);
  237. }
  238. /**
  239. * 根据数据ID删除排班数据
  240. * @param id 数据ID
  241. * @return
  242. */
  243. @Override
  244. public CommonResult deleteRepairClassesSettingsById(int id) {
  245. RepairClassesSettings data = repairClassesSettingsService.getRepairById(id);
  246. if (data == null) {
  247. return CommonResult.fail("当前数据不存在,删除失败!");
  248. }
  249. LocalDate dateNow = LocalDate.now();
  250. //小于等于当天时间的数据不能删除
  251. if(data.getStartDate().isBefore(dateNow) || data.getStartDate().isEqual(dateNow)){
  252. return CommonResult.fail("当天及当天以前的排班数据无法进行删除操作!");
  253. }
  254. int result = repairClassesSettingsService.deleteRepairClassesSettingsById(id);
  255. return result > 0 ? CommonResult.ok("删除成功") : CommonResult.fail("删除失败");
  256. }
  257. /**
  258. * 排班模板导出
  259. * @param response 响应流
  260. */
  261. @Override
  262. public void downloadRepairClassesSettingExcel(HttpServletResponse response) {
  263. ExcelUtils.fileDownload("/自定义排班模板.xlsx", response);
  264. }
  265. /**
  266. * 维修师傅首页当天排班分页数据
  267. * @param currentPage 当前页数
  268. * @param pageCount 一页数据条数
  269. * @param schoolId 校区ID
  270. * @return
  271. */
  272. @Override
  273. public CommonResult queryPageUserClasses(int currentPage, int pageCount, int schoolId, Integer userId) {
  274. PageUtils<AddressBookVo> result = repairUserService.queryUserPageBySchoolId(currentPage, pageCount, schoolId, userId);
  275. List<Integer> userIds = result.getList().stream().map(AddressBookVo::getId).collect(Collectors.toList());
  276. List<ClassesDataVo> users = repairClassesSettingsService.queryClassesDatas(userIds, schoolId);
  277. List<UserScheduleVo> results = new ArrayList<>();
  278. for (AddressBookVo ru : result.getList()) {
  279. UserScheduleVo dataV = new UserScheduleVo();
  280. dataV.setId(ru.getId());
  281. dataV.setUserName(ru.getUserName());
  282. dataV.setUserPhone(ru.getUserPhone());
  283. dataV.setWorkTypeName(ru.getWorkTypeName());
  284. dataV.setState(ru.getState());
  285. dataV.setStateStr(ru.getStateStr());
  286. Optional<ClassesDataVo> data = users.stream().filter(e -> Integer.valueOf(e.getUserId()).equals(ru.getId())).findFirst();
  287. if (data != null && data.isPresent()) {
  288. dataV.setShifts(data.get().getShifts());
  289. dataV.setClassId(data.get().getId());
  290. }
  291. results.add(dataV);
  292. }
  293. PageUtils resultPage = new PageUtils(results, result.getTotalCount(), result.getTotalPage(), result.getCurrPage());
  294. return CommonResult.ok(resultPage);
  295. }
  296. /**
  297. * 后勤首页修改或新增当天排班数据
  298. * @param usv 排班数据
  299. * @param bindingResult 是否为空判断
  300. * @return
  301. * @throws ParseException
  302. */
  303. @Override
  304. public CommonResult updateUserClasses(UserScheduleVo usv, BindingResult bindingResult) throws ParseException {
  305. if (bindingResult.hasErrors()) {
  306. String st = paramUtils.getParamError(bindingResult);
  307. return CommonResult.fail(st);
  308. }
  309. RepairUser user = repairUserService.getRepairById(usv.getId());
  310. if (user == null) {
  311. return CommonResult.fail("用户信息失效,修改失败");
  312. }
  313. if (!user.getState().equals(usv.getState())) {
  314. user.setState(usv.getState());
  315. int updateUser = repairUserService.updateRepairUser(user);
  316. if (updateUser <= 0) {
  317. return CommonResult.fail("用户信息更新失败");
  318. }
  319. }
  320. if (usv.getShifts() != null && usv.getShifts().size() > 0) {
  321. List<String> shifts = usv.getShifts().stream().map(ShiftSettingDataVo::getId).distinct().collect(Collectors.toList()).stream().map(String::valueOf).collect(Collectors.toList());
  322. List<RepairShiftSettings> datas = repairShiftSettingsService.getRepairsByShiftIds(shifts);
  323. for (int i = 0; i < datas.size(); i++) {
  324. for (int j = i + 1; j < datas.size(); j++) {
  325. boolean isOver = TimeExchange.overlapping(datas.get(i).getStartTime(), datas.get(i).getEndTime(), datas.get(j).getStartTime(), datas.get(j).getEndTime());
  326. if (isOver) {
  327. return CommonResult.fail(datas.get(i).getStartTime() + "-" + datas.get(i).getEndTime() + "时间段和" + datas.get(j).getStartTime() + "-" + datas.get(j).getEndTime() + "时间段存在重叠");
  328. }
  329. }
  330. }
  331. //为空则新增排班数据
  332. if (usv.getClassId() == null) {
  333. RepairClassesSettings rcs = new RepairClassesSettings();
  334. rcs.setUserId(usv.getId());
  335. Date date = new Date();
  336. Instant instant = date.toInstant();
  337. ZoneId zoneId = ZoneId.systemDefault();
  338. // atZone()方法返回在指定时区从此Instant生成的ZonedDateTime。
  339. LocalDate localDate = instant.atZone(zoneId).toLocalDate();
  340. rcs.setStartDate(localDate);
  341. rcs.setShiftId(StringUtils.join(shifts, ","));
  342. int insertClasses = repairClassesSettingsService.insertRepairClassesSettings(rcs);
  343. if (insertClasses <= 0) {
  344. return CommonResult.fail("排班信息新增失败");
  345. }
  346. } else {
  347. //判断对应师傅当天排班数据是否存在
  348. RepairClassesSettings csv = repairClassesSettingsService.getRepairById(usv.getClassId());
  349. if(csv == null){
  350. return CommonResult.fail("排班数据不存在更新失败");
  351. }
  352. csv.setShiftId(StringUtils.join(shifts, ","));
  353. int updateClasses = repairClassesSettingsService.updateRepairClassesSettings(csv);
  354. if (updateClasses <= 0) {
  355. return CommonResult.fail("排班信息更新失败");
  356. }
  357. }
  358. }
  359. return CommonResult.ok("操作成功");
  360. }
  361. /**
  362. * 排班数据导入
  363. * @param file 导入文件
  364. * @return
  365. * @throws IOException
  366. * @throws ParseException
  367. */
  368. @Override
  369. public CommonResult importRepairClassesSettingExcel(MultipartFile file) throws IOException, ParseException {
  370. System.out.println("导入排班");
  371. if (file.isEmpty() || file.getSize() == 0) {
  372. return CommonResult.fail("导入文件不能为空");
  373. }
  374. String ContentType = file.getContentType();
  375. InputStream inputStream = file.getInputStream();
  376. List<RepairClassesSettings> result = new ArrayList<>();
  377. //xls格式文件
  378. if (ContentType.equals(eFileType.Xls.getValue())) {
  379. CommonResult<List<RepairClassesSettings>> resultData = readXls(inputStream);
  380. if (!resultData.isSuccess()) {
  381. return resultData;
  382. }
  383. result = resultData.getData();
  384. } else if (ContentType.equals(eFileType.Xlsx.getValue())) {
  385. CommonResult<List<RepairClassesSettings>> resultData = readXlsx(inputStream);
  386. if (!resultData.isSuccess()) {
  387. return resultData;
  388. }
  389. result = resultData.getData();
  390. } else {
  391. return CommonResult.fail("排班数据导入只支持Xls、Xlsx");
  392. }
  393. boolean resultBool = repairClassesSettingsService.saveBatch(result);
  394. System.out.println("导入排班1");
  395. return resultBool ? CommonResult.ok("导入成功") : CommonResult.fail("导入失败");
  396. }
  397. /**
  398. * xls文件读取方法
  399. * @param inputStream
  400. * @return
  401. * @throws IOException
  402. * @throws ParseException
  403. */
  404. private CommonResult<List<RepairClassesSettings>> readXls(InputStream inputStream) throws IOException, ParseException {
  405. int month = 0;
  406. int year = 0;
  407. List<String> datas = new ArrayList<>();
  408. HSSFWorkbook sheets = new HSSFWorkbook(inputStream);
  409. List<RepairClassesSettings> result = new ArrayList<>();
  410. //读取第一张sheet
  411. HSSFSheet sheetAt = sheets.getSheetAt(0);
  412. List<String> UserNames = new ArrayList<>();
  413. List<Integer> userZZs = new ArrayList<>();
  414. userZZs.add(1);
  415. userZZs.add(2);
  416. List<RepairUser> users = repairUserService.getRepairByUserZZ(userZZs, null);
  417. List<RepairShiftSettings> shifts = repairShiftSettingsService.getRepairs();
  418. List<RepairClassesSettings> clssses = new ArrayList<>();
  419. //rowNum = 3 从第三行开始获取值
  420. //sheetAt.getLastRowNum():从0开始统计数量 所以得+1
  421. for (int rowNum = 0; rowNum < sheetAt.getLastRowNum() + 1; rowNum++) {
  422. HSSFRow row = sheetAt.getRow(rowNum);
  423. if (row != null) {
  424. //使用了getStringCellValue()方法来获取值,POI会判断单元格的类型,如果非字符串类型就会抛出上面的异常。
  425. //所以先使用setCellType()方法先将该单元格的类型设置为STRING
  426. //然后poi会根据字符串读取它
  427. //第一行数据获取月份
  428. if (rowNum == 0) {
  429. try {
  430. row.getCell(0).setCellType(CellType.STRING);
  431. month = Integer.valueOf(row.getCell(0).getStringCellValue().substring(4, 6));
  432. year = Integer.valueOf(row.getCell(0).getStringCellValue().substring(0, 4));
  433. } catch (Exception e) {
  434. return CommonResult.fail("请按模板格式导入数据");
  435. }
  436. clssses = repairClassesSettingsService.getRepairByMonths(year + "-" + String.format("%02d", month));
  437. } else if (rowNum == 3) {//日期
  438. for (int d = 3; d <= row.getPhysicalNumberOfCells() - 1; d++) {
  439. row.getCell(d).setCellType(CellType.STRING);
  440. String stringCellValue = year + "-" + String.format("%02d", month) + "-" + String.format("%02d", Integer.valueOf(row.getCell(d).getStringCellValue()));
  441. datas.add(stringCellValue);
  442. }
  443. } else if (rowNum >= 4) {
  444. if(row.getCell(0)==null){
  445. continue;
  446. }
  447. row.getCell(0).setCellType(CellType.STRING);
  448. String UserName = row.getCell(0).getStringCellValue();
  449. if(ObjectUtils.isEmpty(UserName)){
  450. continue;
  451. }
  452. row.getCell(1).setCellType(CellType.STRING);
  453. String CardNumber = row.getCell(1).getStringCellValue();
  454. int userId = 0;
  455. Optional<RepairUser> userData = users.stream().filter(e -> e.getUserName().equals(UserName)).findFirst();
  456. if (userData != null && userData.isPresent()) {
  457. userId = userData.get().getId();
  458. }
  459. if (userId == 0) {
  460. return CommonResult.fail(UserName + "用户数据不存在");
  461. }
  462. UserNames.add(UserName);
  463. if (UserNames.stream().distinct().count() != UserNames.size()) {
  464. return CommonResult.fail("导入的Excel中" + UserName + "存在重复数据");
  465. }
  466. for (int d = 1; d <= datas.size(); d++) {
  467. Integer userid = userId;
  468. String dateStr = datas.get(d - 1);
  469. DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  470. //重复数据不让导入
  471. long exist = clssses.stream().filter(e -> e.getUserId().equals(userid) && e.getStartDate().format(formatter).equals(dateStr)).count();
  472. if (exist > 0) {
  473. return CommonResult.fail(UserName + "的" + dateStr + "日数据在系统中已存在,请勿重复导入");
  474. }
  475. //用户ID
  476. RepairClassesSettings classData = new RepairClassesSettings();
  477. classData.setUserId(userId);
  478. //日期
  479. DateTimeFormatter df = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  480. LocalDate dateParam = LocalDate.parse(datas.get(d - 1), df);
  481. classData.setStartDate(dateParam);
  482. //班次ID
  483. row.getCell(d+2).setCellType(CellType.STRING);
  484. String stringCellValue = row.getCell(d+2).getStringCellValue();
  485. String[] shiftStrs = stringCellValue.split("、");
  486. List<String> shiftIds = new ArrayList<>();
  487. List<RepairShiftSettings> shiftDatas = new ArrayList<>();
  488. for (String shiftStr : shiftStrs) {
  489. if (!shiftStr.equals("")) {
  490. Optional<RepairShiftSettings> shiftData = shifts.stream().filter(e -> e.getName().equals(shiftStr)).findFirst();
  491. if (shiftData != null && shiftData.isPresent()) {
  492. shiftIds.add(shiftData.get().getId().toString());
  493. shiftDatas.add(shiftData.get());
  494. } else {
  495. return CommonResult.fail(UserName + "的" + datas.get(d - 1) + "的" + shiftStr + "数据不存在");
  496. }
  497. }
  498. }
  499. for (int i = 0; i < shiftDatas.size(); i++) {
  500. for (int j = i + 1; j < shiftDatas.size(); j++) {
  501. boolean isOver = TimeExchange.overlapping(shiftDatas.get(i).getStartTime(), shiftDatas.get(i).getEndTime(), shiftDatas.get(j).getStartTime(), shiftDatas.get(j).getEndTime());
  502. if (isOver) {
  503. return CommonResult.fail(UserName + "师傅的" + datas.get(d - 1) + "日" + shiftDatas.get(i).getStartTime() + "-" + shiftDatas.get(i).getEndTime() + "时间段和" + shiftDatas.get(j).getStartTime() + "-" + shiftDatas.get(j).getEndTime() + "时间段存在重叠");
  504. }
  505. }
  506. }
  507. classData.setShiftId(StringUtils.join(shiftIds, ','));
  508. result.add(classData);
  509. }
  510. }
  511. }
  512. }
  513. return CommonResult.ok(result);
  514. }
  515. /**
  516. * Xlsx文件读取方法
  517. * @param inputStream
  518. * @return
  519. * @throws IOException
  520. * @throws ParseException
  521. */
  522. private CommonResult<List<RepairClassesSettings>> readXlsx(InputStream inputStream) throws IOException, ParseException {
  523. int month = 0;
  524. int year = 0;
  525. List<String> datas = new ArrayList<>();
  526. XSSFWorkbook sheets = new XSSFWorkbook(inputStream);
  527. List<RepairClassesSettings> result = new ArrayList<>();
  528. //读取第一张sheet
  529. XSSFSheet sheetAt = sheets.getSheetAt(0);
  530. List<String> UserNames = new ArrayList<>();
  531. List<Integer> userZZs = new ArrayList<>();
  532. userZZs.add(1);
  533. userZZs.add(2);
  534. List<RepairUser> users = repairUserService.getRepairByUserZZ(userZZs, null);
  535. List<RepairShiftSettings> shifts = repairShiftSettingsService.getRepairs();
  536. List<RepairClassesSettings> clssses = new ArrayList<>();
  537. //rowNum = 3 从第三行开始获取值
  538. //sheetAt.getLastRowNum():从0开始统计数量 所以得+1
  539. for (int rowNum = 0; rowNum < sheetAt.getLastRowNum() + 1; rowNum++) {
  540. XSSFRow row = sheetAt.getRow(rowNum);
  541. if (row != null) {
  542. //使用了getStringCellValue()方法来获取值,POI会判断单元格的类型,如果非字符串类型就会抛出上面的异常。
  543. //所以先使用setCellType()方法先将该单元格的类型设置为STRING
  544. //然后poi会根据字符串读取它
  545. //第一行数据获取月份
  546. if (rowNum == 0) {
  547. try {
  548. row.getCell(0).setCellType(CellType.STRING);
  549. month = Integer.valueOf(row.getCell(0).getStringCellValue().substring(4, 6));
  550. year = Integer.valueOf(row.getCell(0).getStringCellValue().substring(0, 4));
  551. } catch (Exception e) {
  552. return CommonResult.fail("请按模板格式导入数据");
  553. }
  554. clssses = repairClassesSettingsService.getRepairByMonths(year + "-" + String.format("%02d", month));
  555. } else if (rowNum == 3) {//日期
  556. for (int d = 3; d <= row.getPhysicalNumberOfCells() - 1; d++) {
  557. row.getCell(d).setCellType(CellType.STRING);
  558. String stringCellValue = year + "-" + String.format("%02d", month) + "-" + String.format("%02d", Integer.valueOf(row.getCell(d).getStringCellValue()));
  559. datas.add(stringCellValue);
  560. }
  561. } else if (rowNum >= 4) {
  562. if(row.getCell(0)==null){
  563. continue;
  564. }
  565. row.getCell(0).setCellType(CellType.STRING);
  566. String UserName = row.getCell(0).getStringCellValue();
  567. if(ObjectUtils.isEmpty(UserName)){
  568. continue;
  569. }
  570. row.getCell(1).setCellType(CellType.STRING);
  571. String CardNumber = row.getCell(1).getStringCellValue();
  572. int userId = 0;
  573. Optional<RepairUser> userData = users.stream().filter(e -> e.getUserName().equals(UserName) && e.getCardNumber().equals(CardNumber)).findFirst();
  574. if (userData != null && userData.isPresent()) {
  575. userId = userData.get().getId();
  576. }
  577. if (userId == 0) {
  578. return CommonResult.fail(UserName + "用户数据不存在");
  579. }
  580. UserNames.add(UserName);
  581. if (UserNames.stream().distinct().count() != UserNames.size()) {
  582. return CommonResult.fail("导入的Excel中" + UserName + "存在重复数据");
  583. }
  584. for (int d = 1; d <= datas.size(); d++) {
  585. Integer userid = userId;
  586. String dateStr = datas.get(d - 1);
  587. DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  588. //重复数据不让导入
  589. long exist = clssses.stream().filter(e -> e.getUserId().equals(userid) && e.getStartDate().format(formatter).equals(dateStr)).count();
  590. if (exist > 0) {
  591. return CommonResult.fail(UserName + "的" + dateStr + "日数据在系统中已存在,请勿重复导入");
  592. }
  593. //用户ID
  594. RepairClassesSettings classData = new RepairClassesSettings();
  595. classData.setUserId(userId);
  596. //日期
  597. DateTimeFormatter df = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  598. LocalDate dateParam = LocalDate.parse(datas.get(d - 1), df);
  599. classData.setStartDate(dateParam);
  600. //班次ID
  601. row.getCell(d+2).setCellType(CellType.STRING);
  602. String stringCellValue = row.getCell(d+2).getStringCellValue();
  603. String[] shiftStrs = stringCellValue.split("、");
  604. List<String> shiftIds = new ArrayList<>();
  605. List<RepairShiftSettings> shiftDatas = new ArrayList<>();
  606. for (String shiftStr : shiftStrs) {
  607. if (!shiftStr.equals("")) {
  608. Optional<RepairShiftSettings> shiftData = shifts.stream().filter(e -> e.getName().equals(shiftStr)).findFirst();
  609. if (shiftData != null && shiftData.isPresent()) {
  610. shiftIds.add(shiftData.get().getId().toString());
  611. shiftDatas.add(shiftData.get());
  612. } else {
  613. return CommonResult.fail(UserName + "的" + datas.get(d - 1) + "的" + shiftStr + "数据不存在");
  614. }
  615. }
  616. }
  617. for (int i = 0; i < shiftDatas.size(); i++) {
  618. for (int j = i + 1; j < shiftDatas.size(); j++) {
  619. boolean isOver = TimeExchange.overlapping(shiftDatas.get(i).getStartTime(), shiftDatas.get(i).getEndTime(), shiftDatas.get(j).getStartTime(), shiftDatas.get(j).getEndTime());
  620. if (isOver) {
  621. return CommonResult.fail(UserName + "师傅的" + datas.get(d - 1) + "日" + shiftDatas.get(i).getStartTime() + "-" + shiftDatas.get(i).getEndTime() + "时间段和" + shiftDatas.get(j).getStartTime() + "-" + shiftDatas.get(j).getEndTime() + "时间段存在重叠");
  622. }
  623. }
  624. }
  625. classData.setShiftId(StringUtils.join(shiftIds, ','));
  626. result.add(classData);
  627. }
  628. }
  629. }
  630. }
  631. return CommonResult.ok(result);
  632. }
  633. }