AttendanceStatisticsServiceImpl.java 31 KB

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