AttendanceStatisticsServiceImpl.java 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706
  1. package com.xjrsoft.module.attendance.service.impl;
  2. import cn.hutool.core.bean.BeanUtil;
  3. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
  4. import com.baomidou.mybatisplus.core.toolkit.Wrappers;
  5. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  6. import com.github.yulichang.base.MPJBaseServiceImpl;
  7. import com.github.yulichang.wrapper.MPJLambdaWrapper;
  8. import com.google.gson.JsonArray;
  9. import com.google.gson.JsonElement;
  10. import com.google.gson.JsonObject;
  11. import com.google.gson.JsonParser;
  12. import com.xjrsoft.common.enums.DeleteMark;
  13. import com.xjrsoft.common.exception.MyException;
  14. import com.xjrsoft.common.mybatis.SqlRunnerAdapter;
  15. import com.xjrsoft.common.utils.VoToColumnUtil;
  16. import com.xjrsoft.module.attendance.dto.AddAttendanceStatisticsDto;
  17. import com.xjrsoft.module.attendance.dto.AttendanceStatisticsPageDto;
  18. import com.xjrsoft.module.attendance.dto.AttendanceStatisticsRecordDto;
  19. import com.xjrsoft.module.attendance.entity.AttendanceRuleDetails;
  20. import com.xjrsoft.module.attendance.entity.AttendanceStatistics;
  21. import com.xjrsoft.module.attendance.entity.AttendanceStatisticsRecord;
  22. import com.xjrsoft.module.attendance.entity.AttendanceUserRelation;
  23. import com.xjrsoft.module.attendance.entity.TeacherAttendanceRecord;
  24. import com.xjrsoft.module.attendance.mapper.AttendanceStatisticsMapper;
  25. import com.xjrsoft.module.attendance.service.IAttendanceRuleCategoryService;
  26. import com.xjrsoft.module.attendance.service.IAttendanceStatisticsRecordService;
  27. import com.xjrsoft.module.attendance.service.IAttendanceStatisticsService;
  28. import com.xjrsoft.module.attendance.service.ITeacherAttendanceRecordService;
  29. import com.xjrsoft.module.attendance.vo.AttendanceRuleDetailsUserVo;
  30. import com.xjrsoft.module.attendance.vo.AttendanceStatisticsPageVo;
  31. import com.xjrsoft.module.attendance.vo.AttendanceStatisticsRecordVo;
  32. import com.xjrsoft.module.hikvision.util.OutInRecordUtil;
  33. import com.xjrsoft.module.holiday.entity.HolidayDate;
  34. import com.xjrsoft.module.holiday.service.IHolidayDateService;
  35. import com.xjrsoft.module.job.AttendanceRecordTask;
  36. import com.xjrsoft.module.organization.entity.User;
  37. import com.xjrsoft.module.organization.service.IUserService;
  38. import com.xjrsoft.module.organization.vo.UserIdDeptNameVo;
  39. import com.xjrsoft.module.outint.service.ICarOutInRecordService;
  40. import com.xjrsoft.module.outint.service.ITeacherOutInRecordService;
  41. import com.xjrsoft.module.teacher.entity.BaseTeacher;
  42. import com.xjrsoft.module.teacher.mapper.FaceImportMapper;
  43. import com.xjrsoft.module.teacher.service.IWfTeacherleaveService;
  44. import lombok.AllArgsConstructor;
  45. import me.zhyd.oauth.log.Log;
  46. import org.apache.poi.ss.usermodel.Cell;
  47. import org.apache.poi.ss.usermodel.CellStyle;
  48. import org.apache.poi.ss.usermodel.Font;
  49. import org.apache.poi.ss.usermodel.HorizontalAlignment;
  50. import org.apache.poi.ss.usermodel.IndexedColors;
  51. import org.apache.poi.ss.usermodel.Row;
  52. import org.apache.poi.ss.usermodel.Sheet;
  53. import org.apache.poi.ss.usermodel.VerticalAlignment;
  54. import org.apache.poi.ss.usermodel.Workbook;
  55. import org.apache.poi.ss.util.CellRangeAddress;
  56. import org.apache.poi.xssf.usermodel.XSSFWorkbook;
  57. import org.springframework.stereotype.Service;
  58. import org.springframework.transaction.annotation.Transactional;
  59. import java.io.ByteArrayOutputStream;
  60. import java.io.IOException;
  61. import java.text.ParseException;
  62. import java.time.LocalDate;
  63. import java.time.LocalDateTime;
  64. import java.time.format.DateTimeFormatter;
  65. import java.time.temporal.ChronoUnit;
  66. import java.util.ArrayList;
  67. import java.util.Arrays;
  68. import java.util.Date;
  69. import java.util.HashMap;
  70. import java.util.List;
  71. import java.util.Map;
  72. import java.util.Objects;
  73. import java.util.stream.Collectors;
  74. /**
  75. * @title: 考勤统计
  76. * @Author dzx
  77. * @Date: 2024-10-19
  78. * @Version 1.0
  79. */
  80. @Service
  81. @AllArgsConstructor
  82. public class AttendanceStatisticsServiceImpl extends MPJBaseServiceImpl<AttendanceStatisticsMapper, AttendanceStatistics> implements IAttendanceStatisticsService {
  83. private final AttendanceStatisticsMapper statisticsMapper;
  84. private final IAttendanceStatisticsRecordService statisticsRecordService;
  85. private final ITeacherAttendanceRecordService recordService;
  86. private final IUserService userService;
  87. private final IAttendanceRuleCategoryService ruleCategoryService;
  88. private final IHolidayDateService holidayDateService;
  89. private final FaceImportMapper faceImportMapper;
  90. private final IWfTeacherleaveService wfTeacherleaveService;
  91. private final ITeacherOutInRecordService outInRecordService;
  92. private final ICarOutInRecordService carOutInRecordService;
  93. @Override
  94. @Transactional(rollbackFor = Exception.class)
  95. public Boolean add(AddAttendanceStatisticsDto dto) {
  96. AttendanceStatistics statistics = BeanUtil.toBean(dto, AttendanceStatistics.class);
  97. statistics.setCreateDate(new Date());
  98. statistics.setStatus(0);
  99. statisticsMapper.insert(statistics);
  100. //1、同步出入记录的数据
  101. //2、固化考勤数据
  102. refreshRecord(statistics.getId());
  103. return true;
  104. }
  105. @Override
  106. @Transactional(rollbackFor = Exception.class)
  107. public Boolean update(AttendanceStatistics attendanceStatistics) {
  108. statisticsMapper.updateById(attendanceStatistics);
  109. //********************************* AttendanceStatisticsRecord 增删改 开始 *******************************************/
  110. {
  111. // 查出所有子级的id
  112. List<AttendanceStatisticsRecord> attendanceStatisticsRecordList = statisticsRecordService.list(Wrappers.lambdaQuery(AttendanceStatisticsRecord.class).eq(AttendanceStatisticsRecord::getAttendanceStatisticsId, attendanceStatistics.getId()).select(AttendanceStatisticsRecord::getId));
  113. List<Long> attendanceStatisticsRecordIds = attendanceStatisticsRecordList.stream().map(AttendanceStatisticsRecord::getId).collect(Collectors.toList());
  114. //原有子表单 没有被删除的主键
  115. List<Long> attendanceStatisticsRecordOldIds = attendanceStatistics.getAttendanceStatisticsRecordList().stream().map(AttendanceStatisticsRecord::getId).filter(Objects::nonNull).collect(Collectors.toList());
  116. //找到需要删除的id
  117. List<Long> attendanceStatisticsRecordRemoveIds = attendanceStatisticsRecordIds.stream().filter(item -> !attendanceStatisticsRecordOldIds.contains(item)).collect(Collectors.toList());
  118. for (AttendanceStatisticsRecord attendanceStatisticsRecord : attendanceStatistics.getAttendanceStatisticsRecordList()) {
  119. //如果不等于空则修改
  120. if (attendanceStatisticsRecord.getId() != null) {
  121. statisticsRecordService.updateById(attendanceStatisticsRecord);
  122. }
  123. //如果等于空 则新增
  124. else {
  125. //已经不存在的id 删除
  126. attendanceStatisticsRecord.setAttendanceStatisticsId(attendanceStatistics.getId());
  127. statisticsRecordService.save(attendanceStatisticsRecord);
  128. }
  129. }
  130. //已经不存在的id 删除
  131. if (attendanceStatisticsRecordRemoveIds.size() > 0) {
  132. statisticsRecordService.removeBatchByIds(attendanceStatisticsRecordRemoveIds);
  133. }
  134. }
  135. //********************************* AttendanceStatisticsRecord 增删改 结束 *******************************************/
  136. return true;
  137. }
  138. @Override
  139. @Transactional(rollbackFor = Exception.class)
  140. public Boolean delete(List<Long> ids) {
  141. statisticsMapper.deleteBatchIds(ids);
  142. statisticsRecordService.remove(Wrappers.lambdaQuery(AttendanceStatisticsRecord.class).in(AttendanceStatisticsRecord::getAttendanceStatisticsId, ids));
  143. return true;
  144. }
  145. @Override
  146. public Page<AttendanceStatisticsPageVo> getPage(Page<AttendanceStatisticsPageVo> page, AttendanceStatisticsPageDto dto) {
  147. return this.baseMapper.getPage(page, dto);
  148. }
  149. @Override
  150. public List<AttendanceStatisticsRecordVo> getRecordList(AttendanceStatisticsRecordDto dto) {
  151. return this.baseMapper.getRecordList(dto);
  152. }
  153. /**
  154. * 1、修改状态为统计中
  155. * 2、删除以前的数据
  156. * 3、重新计算数据并入库
  157. * 4、将状态改为统计完成
  158. */
  159. @Override
  160. @Transactional(rollbackFor = Exception.class)
  161. public Boolean refreshRecord(Long id) {
  162. String remarks = null;
  163. int personCount = 0;
  164. int attendanceDays = 0;
  165. try {
  166. //1、修改状态为统计中
  167. AttendanceStatistics statistics = this.getById(id);
  168. statistics.setStatus(0);
  169. statistics.setModifyDate(new Date());
  170. this.updateById(statistics);
  171. List<LocalDate> dateList2 = getDatesBetween(statistics.getStartDate(), statistics.getEndDate());
  172. LocalDateTime now = LocalDateTime.now();
  173. if (dateList2.contains(now.toLocalDate())) {
  174. DateTimeFormatter dtf = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  175. String sql = "DELETE FROM teacher_attendance_record WHERE attendance_date = '" + now.toLocalDate().format(dtf) + "'" +
  176. " AND time_interval = " + statistics.getTimePeriod();
  177. SqlRunnerAdapter.db().delete(sql);
  178. OutInRecordUtil outInRecordUtil = new OutInRecordUtil();
  179. LocalDateTime startDateTime = now.withHour(0).withMinute(0).withSecond(0).withNano(0);
  180. LocalDateTime endDateTime = startDateTime.plusDays(1).plusSeconds(-1);
  181. Map<Long, AttendanceRuleDetailsUserVo> teacherRules = ruleCategoryService.getAllTeacherTodyRule(now.getDayOfWeek().name());
  182. try {
  183. //教师
  184. outInRecordUtil.getTeacherRecords(faceImportMapper, startDateTime, endDateTime, teacherRules);
  185. //拉取车辆数据
  186. outInRecordUtil.GetVehicleRecord(faceImportMapper, startDateTime, endDateTime);
  187. }catch (Exception e){
  188. Log.error(e.getMessage());
  189. }
  190. AttendanceRecordTask recordTask = new AttendanceRecordTask();
  191. recordTask.teacherAttendanceRecord(startDateTime, statistics.getTimePeriod(),
  192. userService,
  193. ruleCategoryService,
  194. holidayDateService,
  195. wfTeacherleaveService,
  196. outInRecordService,
  197. carOutInRecordService,
  198. recordService
  199. );
  200. }
  201. //2、删除以前的数据
  202. statisticsRecordService.remove(
  203. new QueryWrapper<AttendanceStatisticsRecord>().lambda()
  204. .eq(AttendanceStatisticsRecord::getAttendanceStatisticsId, id)
  205. );
  206. /**
  207. * 3、重新计算数据并入库
  208. */
  209. //3.1、根据考勤规则和统计的时间段查询这个规则下面所涉及到的老师
  210. List<User> userList = userService.list(
  211. new MPJLambdaWrapper<User>()
  212. .select(User::getId)
  213. .select(User.class, x -> VoToColumnUtil.fieldsToColumns(User.class).contains(x.getProperty()))
  214. .innerJoin(AttendanceUserRelation.class, AttendanceUserRelation::getUserId, User::getId)
  215. .innerJoin(BaseTeacher.class, BaseTeacher::getUserId, User::getId)
  216. .eq(AttendanceUserRelation::getAttendanceRuleCategoryId, statistics.getAttendanceRuleCategoryId())
  217. .eq(AttendanceUserRelation::getDeleteMark, DeleteMark.NODELETE.getCode())
  218. );
  219. if (userList.isEmpty()) {
  220. throw new MyException("该规则下无考勤人员");
  221. }
  222. List<Long> userIds = userList.stream().map(User::getId).collect(Collectors.toList());
  223. //查询固化的考勤数据
  224. List<TeacherAttendanceRecord> attendanceRecords = recordService.list(
  225. new QueryWrapper<TeacherAttendanceRecord>().lambda()
  226. .eq(TeacherAttendanceRecord::getTimeInterval, statistics.getTimePeriod())
  227. .in(TeacherAttendanceRecord::getUserId, userIds)
  228. .between(TeacherAttendanceRecord::getAttendanceDate, statistics.getStartDate(), statistics.getEndDate())
  229. .eq(TeacherAttendanceRecord::getDeleteMark, DeleteMark.NODELETE.getCode())
  230. );
  231. //3.2、准备相关需要的数据
  232. DateTimeFormatter dtf = DateTimeFormatter.ofPattern("yyyy年MM月dd日");
  233. DateTimeFormatter timeDtf = DateTimeFormatter.ofPattern("HH:mm");
  234. Map<String, String> weekCnMap = initWeekCn();
  235. Map<String, String> weekEnMap = initWeekEn();
  236. List<UserIdDeptNameVo> teacherDeptName = userService.getTeacherDeptName();
  237. Map<Long, UserIdDeptNameVo> userDeptMap = teacherDeptName.stream().collect(Collectors.toMap(UserIdDeptNameVo::getId, x -> x));
  238. //3.3、查询这个考勤规则下面的所有细则并转换成map
  239. List<AttendanceRuleDetails> ruleList = ruleCategoryService.getRules(statistics.getAttendanceRuleCategoryId());
  240. Map<String, AttendanceRuleDetails> ruleMap = ruleList.stream().collect(Collectors.toMap(AttendanceRuleDetails::getDateType, x -> x));
  241. //3.4、计算出所有的天数
  242. List<LocalDate> dateList = getDatesBetween(statistics.getStartDate(), statistics.getEndDate(), ruleMap, weekEnMap);
  243. attendanceDays = dateList.size();
  244. //3.5、循环计算,并存入insertList,方便后续批量入库
  245. List<AttendanceStatisticsRecord> insertList = new ArrayList<>();
  246. List<String> leaveTypeList = new ArrayList<>();
  247. leaveTypeList.add("事假");
  248. leaveTypeList.add("公假");
  249. leaveTypeList.add("病假");
  250. leaveTypeList.add("会议活动");
  251. for (User user : userList) {
  252. AttendanceStatisticsRecord record = new AttendanceStatisticsRecord();
  253. record.setUserId(user.getId());
  254. record.setAttendanceStatisticsId(statistics.getId());
  255. Long normalCount = 0L;
  256. Long leaveCount = 0L;
  257. Long absenteeCount = 0L;
  258. JsonArray daysData = new JsonArray();
  259. int sortCode = 1;
  260. for (LocalDate localDate : dateList) {
  261. String dayOfWeekName = localDate.getDayOfWeek().name();
  262. AttendanceRuleDetails ruleDetails = ruleMap.get(weekEnMap.get(dayOfWeekName));
  263. if (ruleDetails != null && (ruleDetails.getIsAttendance() == null || ruleDetails.getIsAttendance() == 0)) {
  264. continue;
  265. }
  266. JsonObject daysJson = new JsonObject();
  267. daysJson.addProperty("date", localDate.format(dtf));
  268. daysJson.addProperty("week", weekCnMap.get(dayOfWeekName));
  269. String content = "";
  270. if (statistics.getTimePeriod() == 1) {
  271. content += "(" + ruleDetails.getAmStartTime().toLocalTime().format(timeDtf) + ")上班 ";
  272. } else if (statistics.getTimePeriod() == 2) {
  273. content += "(" + ruleDetails.getPmStartTime().toLocalTime().format(timeDtf) + ")上班 ";
  274. }
  275. List<TeacherAttendanceRecord> collect = attendanceRecords.stream()
  276. .filter(el -> el.getAttendanceDate().equals(localDate) && el.getUserId().equals(user.getId())
  277. && !"不考勤".equals(el.getAttendanceStatus())
  278. )
  279. .collect(Collectors.toList());
  280. for (TeacherAttendanceRecord attendanceRecord : collect) {
  281. if ("到校".equals(attendanceRecord.getAttendanceStatus())) {
  282. normalCount++;
  283. content += "正常(" + attendanceRecord.getRecordTime().toLocalTime().format(timeDtf) + ")";
  284. } else if (leaveTypeList.contains(attendanceRecord.getAttendanceStatus())) {
  285. leaveCount++;
  286. content += "请假";
  287. } else {
  288. absenteeCount++;
  289. content += "缺勤";
  290. }
  291. }
  292. daysJson.addProperty("content", content);
  293. daysJson.addProperty("sortCode", sortCode);
  294. daysData.add(daysJson);
  295. sortCode++;
  296. }
  297. record.setNormalCount(normalCount);
  298. record.setLeaveCount(leaveCount);
  299. record.setAbsenteeCount(absenteeCount);
  300. record.setDaysData(daysData.toString());
  301. UserIdDeptNameVo deptNameVo = userDeptMap.get(record.getUserId());
  302. if (deptNameVo != null) {
  303. record.setDeptIds(deptNameVo.getDeptIds());
  304. record.setDeptName(deptNameVo.getDeptName());
  305. }
  306. insertList.add(record);
  307. }
  308. //3.6、插入数据
  309. if (!insertList.isEmpty()) {
  310. statisticsRecordService.saveBatch(insertList);
  311. personCount = insertList.size();
  312. }
  313. return true;
  314. } catch (Exception e) {
  315. Log.error(e.getMessage(), e);
  316. if (e.getClass().equals(MyException.class)) {
  317. throw new MyException(e.getMessage());
  318. }else if(e.getClass().equals(ParseException.class) || e.getClass().equals(ClassCastException.class)){
  319. remarks = "未能拉取海康数据,统计可能不准确";
  320. } else {
  321. throw new MyException("刷新出错,请联系管理员");
  322. }
  323. }finally {
  324. //4、将状态改为统计完成
  325. AttendanceStatistics statistics = this.getById(id);
  326. statistics.setStatus(1);
  327. statistics.setModifyDate(new Date());
  328. statistics.setPersonCount(personCount);
  329. statistics.setAttendanceDays(attendanceDays);
  330. statistics.setRemarks(remarks);
  331. this.updateById(statistics);
  332. return true;
  333. }
  334. }
  335. private List<LocalDate> getDatesBetween(LocalDate startDate, LocalDate endDate, Map<String, AttendanceRuleDetails> ruleMap, Map<String, String> weekEnMap) {
  336. List<LocalDate> dates = new ArrayList<>();
  337. List<Integer> list = Arrays.asList(1, 3);
  338. List<HolidayDate> holidayDates = holidayDateService.list(
  339. new QueryWrapper<HolidayDate>().lambda()
  340. .in(HolidayDate::getStatus, list)
  341. );
  342. List<String> holidayDateList = holidayDates.stream().map(HolidayDate::getDate).collect(Collectors.toList());
  343. List<HolidayDate> workHolidayDates = holidayDateService.list(
  344. new QueryWrapper<HolidayDate>().lambda()
  345. .eq(HolidayDate::getStatus, 2)
  346. );
  347. List<String> workHolidayDateListg = workHolidayDates.stream().map(HolidayDate::getDate).collect(Collectors.toList());
  348. DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  349. List<String> weekList = Arrays.asList("SUNDAY", "SATURDAY");
  350. long numOfDaysBetween = ChronoUnit.DAYS.between(startDate, endDate) + 1; // +1 包含结束日期
  351. for (long i = 0; i < numOfDaysBetween; i++) {
  352. LocalDate localDate = startDate.plusDays(i);
  353. AttendanceRuleDetails holidays = ruleMap.get("holidays");
  354. Integer holidaysIsAttendance = holidays.getIsAttendance();
  355. if (
  356. (
  357. (!workHolidayDateListg.contains(localDate.format(formatter)) && weekList.contains(localDate.getDayOfWeek().name()))
  358. || holidayDateList.contains(localDate.format(formatter)))
  359. && (holidaysIsAttendance == null || holidaysIsAttendance == 0)
  360. ) {
  361. continue;
  362. }
  363. AttendanceRuleDetails ruleDetails = ruleMap.get(weekEnMap.get(localDate.getDayOfWeek().name()));
  364. if (ruleDetails != null && ruleDetails.getIsAttendance() != null && ruleDetails.getIsAttendance() == 1) {
  365. dates.add(localDate);
  366. }
  367. }
  368. return dates;
  369. }
  370. private List<LocalDate> getDatesBetween(LocalDate startDate, LocalDate endDate) {
  371. List<LocalDate> dates = new ArrayList<>();
  372. long numOfDaysBetween = ChronoUnit.DAYS.between(startDate, endDate) + 1; // +1 包含结束日期
  373. for (long i = 0; i < numOfDaysBetween; i++) {
  374. LocalDate localDate = startDate.plusDays(i);
  375. dates.add(localDate);
  376. }
  377. return dates;
  378. }
  379. private Map<String, String> initWeekCn() {
  380. Map<String, String> result = new HashMap<>();
  381. result.put("MONDAY", "周一");
  382. result.put("TUESDAY", "周二");
  383. result.put("WEDNESDAY", "周三");
  384. result.put("THURSDAY", "周四");
  385. result.put("FRIDAY", "周五");
  386. result.put("SATURDAY", "周六");
  387. result.put("SUNDAY", "周日");
  388. return result;
  389. }
  390. private Map<String, String> initWeekEn() {
  391. Map<String, String> result = new HashMap<>();
  392. result.put("MONDAY", "monday");
  393. result.put("TUESDAY", "tuesday");
  394. result.put("WEDNESDAY", "wednesday");
  395. result.put("THURSDAY", "thursday");
  396. result.put("FRIDAY", "friday");
  397. result.put("SATURDAY", "saturday");
  398. result.put("SUNDAY", "sunday");
  399. return result;
  400. }
  401. @Override
  402. public byte[] recordExport(Long id) throws IOException {
  403. AttendanceStatistics statistics = this.getById(id);
  404. List<AttendanceStatisticsRecordVo> recordList = this.getRecordList(new AttendanceStatisticsRecordDto() {{
  405. setId(id);
  406. }});
  407. List<ArrayList<String>> dataList = new ArrayList<>();
  408. int sortCode = 1;
  409. JsonParser parser = new JsonParser();
  410. List<String> dateList = new ArrayList<>();
  411. List<String> weekList = new ArrayList<>();
  412. for (AttendanceStatisticsRecordVo recordVo : recordList) {
  413. ArrayList<String> data = new ArrayList<>();
  414. data.add(sortCode + "");
  415. data.add(recordVo.getName());
  416. data.add(recordVo.getUserName());
  417. data.add(recordVo.getDeptName());
  418. data.add(statistics.getAttendanceDays() + "");
  419. data.add(recordVo.getNormalCount() + "");
  420. data.add(recordVo.getLeaveCount() + "");
  421. data.add(recordVo.getAbsenteeCount() + "");
  422. JsonArray daysData = parser.parse(recordVo.getDaysData()).getAsJsonArray();
  423. for (JsonElement daysDatum : daysData) {
  424. JsonObject daysJson = daysDatum.getAsJsonObject();
  425. data.add(daysJson.get("content").getAsString());
  426. if (!dateList.contains(daysJson.get("date").getAsString())) {
  427. dateList.add(daysJson.get("date").getAsString());
  428. }
  429. if (!weekList.contains(daysJson.get("week").getAsString())) {
  430. weekList.add(daysJson.get("week").getAsString());
  431. }
  432. }
  433. dataList.add(data);
  434. sortCode++;
  435. }
  436. int allColumn = 8 + statistics.getAttendanceDays();//总列数
  437. Workbook workbook = new XSSFWorkbook();
  438. // 创建一个工作表(sheet)
  439. String sheetName = "数据";
  440. Sheet sheet = workbook.createSheet(sheetName);
  441. // 第一行表头
  442. createFirstTitle(workbook, sheet, allColumn - 1);
  443. // 第二行表头
  444. createSecondTitle(workbook, sheet, statistics, 7);
  445. // 第三行表头
  446. createThirdTitle(workbook, sheet, weekList);
  447. // 第四行表头
  448. createFourthTitle(workbook, sheet, 8, dateList);
  449. //生成数据
  450. int dataRowNumber = 4;
  451. for (ArrayList<String> rowData : dataList) {
  452. Row dataRow = sheet.createRow(dataRowNumber);
  453. for (int i = 0; i < rowData.size(); i++) {
  454. if (i > 7) {
  455. sheet.autoSizeColumn(i);
  456. }
  457. String content = rowData.get(i);
  458. Font font = workbook.createFont();
  459. font.setBold(false);// 设置为粗体
  460. font.setFontName("宋体");
  461. if (content != null && content.contains("缺勤")) {
  462. font.setColor(IndexedColors.RED.getIndex()); // 设置字体颜色为红色
  463. } else if (content != null && content.contains("正常")) {
  464. font.setColor(IndexedColors.BLUE.getIndex()); // 设置字体颜色为蓝色
  465. } else if (content != null && content.contains("请假")) {
  466. font.setColor(IndexedColors.BROWN.getIndex()); // 设置字体颜色为橙色
  467. }
  468. font.setFontHeightInPoints((short) 12);
  469. CellStyle cellStyle = workbook.createCellStyle();
  470. cellStyle.setFont(font); // 将字体应用到样式
  471. cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
  472. if (i > 3 && i < 8) {
  473. cellStyle.setAlignment(HorizontalAlignment.CENTER);
  474. }
  475. Cell row1cell2 = dataRow.createCell(i);
  476. row1cell2.setCellValue(content);
  477. row1cell2.setCellStyle(cellStyle);
  478. }
  479. dataRowNumber++;
  480. }
  481. //写入文件
  482. ByteArrayOutputStream bot = new ByteArrayOutputStream();
  483. workbook.write(bot);
  484. return bot.toByteArray();
  485. }
  486. /**
  487. * 创建第一行表头
  488. *
  489. * @param mergeCoulmn 合并后面多少列(不包含自己)
  490. */
  491. void createFirstTitle(Workbook workbook, Sheet sheet, int mergeCoulmn) {
  492. int rowNumber = 0;
  493. Font font = workbook.createFont();
  494. font.setBold(true);// 设置为粗体
  495. font.setFontName("宋体");
  496. //font.setColor(IndexedColors.RED.getIndex()); // 设置字体颜色为红色
  497. font.setFontHeightInPoints((short) 24);
  498. CellStyle cellStyle = workbook.createCellStyle();
  499. cellStyle.setFont(font); // 将字体应用到样式
  500. cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
  501. cellStyle.setAlignment(HorizontalAlignment.CENTER);
  502. Row row = sheet.createRow(rowNumber);
  503. //合并第一行的列
  504. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber, 0, mergeCoulmn));
  505. //3、处理表头
  506. String title1 = "坐班考勤 - 汇总统计";
  507. // 创建单元格并设置值
  508. Cell cell = row.createCell(0);
  509. cell.setCellValue(title1);
  510. cell.setCellStyle(cellStyle);
  511. }
  512. /**
  513. * 生成第二行表头
  514. *
  515. * @param mergeCoulmn 合并后面多少列(不包含自己)
  516. */
  517. void createSecondTitle(Workbook workbook, Sheet sheet, AttendanceStatistics statistics, int mergeCoulmn) {
  518. int rowNumber = 1;
  519. Font font = workbook.createFont();
  520. font.setFontName("宋体");
  521. font.setFontHeightInPoints((short) 12);
  522. CellStyle cellStyle = workbook.createCellStyle();
  523. cellStyle.setFont(font); // 将字体应用到样式
  524. cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
  525. cellStyle.setAlignment(HorizontalAlignment.CENTER);
  526. Row row1 = sheet.createRow(rowNumber);
  527. Cell row1cell1 = row1.createCell(0);
  528. DateTimeFormatter dtf = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  529. String content = "查询的考勤日期范围:" + statistics.getStartDate().format(dtf) + "至" + statistics.getEndDate() + ";";
  530. if (statistics.getTimePeriod() == 1) {
  531. content += "时间段:上午";
  532. } else if (statistics.getTimePeriod() == 2) {
  533. content += "时间段:下午";
  534. }
  535. row1cell1.setCellValue(content);
  536. row1cell1.setCellStyle(cellStyle);
  537. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber, 0, mergeCoulmn));
  538. }
  539. /**
  540. * 生成第三行表头
  541. * 表头:序号、姓名、工号、部门、正常考勤次数、请假次数、缺勤次数,后续周几
  542. *
  543. * @param weekList 周几
  544. */
  545. void createThirdTitle(Workbook workbook, Sheet sheet, List<String> weekList) {
  546. int rowNumber = 2;
  547. Font font = workbook.createFont();
  548. font.setFontName("宋体");
  549. font.setFontHeightInPoints((short) 12);
  550. CellStyle cellStyle = workbook.createCellStyle();
  551. cellStyle.setFont(font); // 将字体应用到样式
  552. cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
  553. cellStyle.setAlignment(HorizontalAlignment.CENTER);
  554. Row row1 = sheet.createRow(rowNumber);
  555. Cell row1cell1 = row1.createCell(0);
  556. row1cell1.setCellValue("序号");
  557. row1cell1.setCellStyle(cellStyle);
  558. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 0, 0));
  559. Cell row1cell2 = row1.createCell(1);
  560. row1cell2.setCellValue("姓名");
  561. row1cell2.setCellStyle(cellStyle);
  562. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 1, 1));
  563. Cell row1cell3 = row1.createCell(2);
  564. row1cell3.setCellValue("工号");
  565. row1cell3.setCellStyle(cellStyle);
  566. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 2, 2));
  567. Cell row1cell4 = row1.createCell(3);
  568. row1cell4.setCellValue("组织机构");
  569. row1cell4.setCellStyle(cellStyle);
  570. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 3, 3));
  571. Cell row1cell8 = row1.createCell(4);
  572. row1cell8.setCellValue("考勤组应出勤天数");
  573. row1cell8.setCellStyle(cellStyle);
  574. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 4, 4));
  575. Cell row1cell5 = row1.createCell(5);
  576. row1cell5.setCellValue("正常考勤次数");
  577. row1cell5.setCellStyle(cellStyle);
  578. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 5, 5));
  579. Cell row1cell6 = row1.createCell(6);
  580. row1cell6.setCellValue("请假次数");
  581. row1cell6.setCellStyle(cellStyle);
  582. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 6, 6));
  583. Cell row1cell7 = row1.createCell(7);
  584. row1cell7.setCellValue("缺勤次数");
  585. row1cell7.setCellStyle(cellStyle);
  586. sheet.addMergedRegion(new CellRangeAddress(rowNumber, rowNumber + 1, 7, 7));
  587. for (int i = 0; i < 7; i++) {
  588. sheet.autoSizeColumn(i);
  589. }
  590. sheet.setColumnWidth(4, 24 * 256);
  591. sheet.setColumnWidth(5, 24 * 256);
  592. int cellNumber = 8;
  593. for (String dayOfWeek : weekList) {
  594. Cell row1cell = row1.createCell(cellNumber);
  595. row1cell.setCellValue(dayOfWeek);
  596. row1cell.setCellStyle(cellStyle);
  597. cellNumber++;
  598. }
  599. }
  600. /**
  601. * @param startColumn 开始列
  602. * @param dateList 2024年10月20日
  603. */
  604. void createFourthTitle(Workbook workbook, Sheet sheet, int startColumn, List<String> dateList) {
  605. int rowNumber = 3;
  606. Font font = workbook.createFont();
  607. font.setFontName("宋体");
  608. font.setFontHeightInPoints((short) 12);
  609. CellStyle cellStyle = workbook.createCellStyle();
  610. cellStyle.setFont(font); // 将字体应用到样式
  611. cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
  612. cellStyle.setAlignment(HorizontalAlignment.CENTER);
  613. Row row1 = sheet.createRow(rowNumber);
  614. for (String date : dateList) {
  615. Cell row1cell = row1.createCell(startColumn);
  616. row1cell.setCellValue(date);
  617. row1cell.setCellStyle(cellStyle);
  618. startColumn++;
  619. }
  620. }
  621. }