AttendanceStatisticsServiceImpl.java 32 KB

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