api/src/main/java/com/xjrsoft/module/attendance/controller/StatisticsController.java

package com.xjrsoft.module.attendance.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.github.yulichang.toolkit.MPJWrappers;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.xjrsoft.common.annotation.XjrLog;
import com.xjrsoft.common.enums.DeleteMark;
import com.xjrsoft.common.enums.OutInStatusEnum;
import com.xjrsoft.common.enums.StudyStatusEnum;
import com.xjrsoft.common.model.result.R;
import com.xjrsoft.common.model.result.RT;
import com.xjrsoft.common.utils.VoToColumnUtil;
import com.xjrsoft.module.attendance.dto.AttendanceStatisticDto;
import com.xjrsoft.module.attendance.dto.TeacherDetailsDto;
import com.xjrsoft.module.attendance.entity.AttendanceRuleCategory;
import com.xjrsoft.module.attendance.entity.AttendanceUserRelation;
import com.xjrsoft.module.attendance.service.IAttendanceRuleCategoryService;
import com.xjrsoft.module.attendance.service.ITeacherAttendanceRecordService;
import com.xjrsoft.module.attendance.vo.TeacherStatisticsPageVo;
import com.xjrsoft.module.attendance.vo.TeacherStatisticsVo;
import com.xjrsoft.module.attendance.vo.VisitorInfoVo;
import com.xjrsoft.module.base.entity.BaseClass;
import com.xjrsoft.module.concat.service.IXjrUserService;
import com.xjrsoft.module.organization.entity.UserDeptRelation;
import com.xjrsoft.module.outint.entity.StudentOutInRecord;
import com.xjrsoft.module.outint.service.IStudentOutInRecordService;
import com.xjrsoft.module.personnel.service.IReservationSchoolService;
import com.xjrsoft.module.student.entity.BaseStudent;
import com.xjrsoft.module.student.entity.BaseStudentSchoolRoll;
import com.xjrsoft.module.student.service.IStudentLeaveService;
import com.xjrsoft.module.teacher.dto.BaseTeacherPageDto;
import com.xjrsoft.module.teacher.entity.BaseTeacher;
import com.xjrsoft.module.teacher.entity.XjrUser;
import com.xjrsoft.module.teacher.service.ITeacherbaseManagerService;
import com.xjrsoft.module.teacher.vo.XjrUserPageVo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.validation.Valid;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @title: 考勤消息设置
 * @Author dzx
 * @Date: 2024-05-21
 * @Version 1.0
 */
@RestController
@RequestMapping("/statistics")
@Api(value = "/statistics", tags = "考勤统计")
@AllArgsConstructor
public class StatisticsController {


    private final IXjrUserService xjrUserService;
    private final IStudentOutInRecordService studentOutInRecordService;
    private final IStudentLeaveService studentLeaveService;
    private final IReservationSchoolService reservationSchoolService;
    private final ITeacherAttendanceRecordService attendanceRecordService;
    private final ITeacherbaseManagerService teacherbaseManagerService;
    private final IAttendanceRuleCategoryService ruleCategoryService;

    @GetMapping(value = "/teacher-statistics")
    @ApiOperation(value = "教职工考勤统计")
    @SaCheckPermission("statistics:detail")
    @XjrLog(value = "教职工考勤统计", saveResponseData = true)
    public RT<TeacherStatisticsVo> teacherStatistics(@Valid AttendanceStatisticDto dto) {
        TeacherStatisticsVo statisticsVo = new TeacherStatisticsVo();

        if(dto.getRuleCategoryId() == null){
            AttendanceRuleCategory ruleCategory = ruleCategoryService.getOne(
                    new QueryWrapper<AttendanceRuleCategory>().lambda()
                            .eq(AttendanceRuleCategory::getIsDefaultShow, 1)
                            .eq(AttendanceRuleCategory::getRoleId, 2)
            );
            if(ruleCategory != null){
                dto.setRuleCategoryId(ruleCategory.getId());
            }
        }

        //查询总人数
        MPJLambdaWrapper<XjrUser> queryWrapper = MPJWrappers.<XjrUser>lambdaJoin()
                .disableLogicDel()
                .eq(ObjectUtil.isNotNull(dto.getDeptId()), UserDeptRelation::getDeptId, dto.getDeptId())
                .eq(AttendanceUserRelation::getAttendanceRuleCategoryId, dto.getRuleCategoryId())
                .innerJoin(BaseTeacher.class, BaseTeacher::getUserId, XjrUser::getId)
                .innerJoin(UserDeptRelation.class, UserDeptRelation::getUserId, XjrUser::getId)
                .innerJoin(AttendanceUserRelation.class, AttendanceUserRelation::getUserId, XjrUser::getId);
        long allCount = xjrUserService.count(queryWrapper);
        statisticsVo.setAllCount(allCount);

        TeacherDetailsDto detailsDto = new TeacherDetailsDto();
        detailsDto.setDate(dto.getDate());
        detailsDto.setDeptId(dto.getDeptId());
        detailsDto.setTimePeriod(dto.getTimePeriod());
        detailsDto.setRuleCategoryId(dto.getRuleCategoryId());
        List<TeacherStatisticsPageVo> list = attendanceRecordService.getList(detailsDto);

        Set<Long> actualCount = new HashSet<>();
        Set<Long> lateCount = new HashSet<>();
        Set<Long> leaveCount = new HashSet<>();
        Set<Long> absenteeismCount = new HashSet<>();
        Set<Long> dividendCount = new HashSet<>();
        Set<Long> teacherIds = new HashSet<>();
        for (TeacherStatisticsPageVo recordVo : list) {
            teacherIds.add(recordVo.getUserId());
            if ("迟到".equals(recordVo.getStatus())) {
                lateCount.add(recordVo.getUserId());
                dividendCount.add(recordVo.getUserId());
            } else if ("请假".equals(recordVo.getStatus())) {
                leaveCount.add(recordVo.getUserId());
                dividendCount.add(recordVo.getUserId());
            } else if ("缺勤".equals(recordVo.getStatus())) {
                absenteeismCount.add(recordVo.getUserId());
                dividendCount.add(recordVo.getUserId());
            }
            if (recordVo.getRecordTime() != null) {
                actualCount.add(recordVo.getUserId());
                dividendCount.add(recordVo.getUserId());
            }
        }
        statisticsVo.setLateCount(lateCount.stream().count());
        statisticsVo.setActualCount(actualCount.stream().count());
        statisticsVo.setLeaveCount(leaveCount.stream().count());
        statisticsVo.setAbsenteeismCount(absenteeismCount.stream().count());

//        //查询不需要考勤的人
//        List<AttendanceRuleCategory> ruleCategories = ruleCategoryService.list(
//                new QueryWrapper<AttendanceRuleCategory>().lambda()
//                        .eq(AttendanceRuleCategory::getDeleteMark, DeleteMark.NODELETE.getCode())
//                        .eq(AttendanceRuleCategory::getRoleId, 2)
//                        .le(AttendanceRuleCategory::getStartDate, LocalDate.now())
//                        .ge(AttendanceRuleCategory::getEndDate, LocalDate.now())
//        );
//        if(!ruleCategories.isEmpty()){
//            //查询所有需要考勤的人
//            BaseTeacherPageDto teacherPageDto = new BaseTeacherPageDto();
//            List<Long> userIds = new ArrayList<>();
//            List<Long> deptIds = new ArrayList<>();
//            for (AttendanceRuleCategory ruleCategory : ruleCategories) {
//                JsonArray userRelation = new JsonParser().parse(ruleCategory.getUserRelation()).getAsJsonArray();
//
//                for (JsonElement jsonElement : userRelation) {
//                    JsonObject jsonObject = jsonElement.getAsJsonObject();
//                    if(jsonObject.get("deptId") != null && !jsonObject.get("deptId").isJsonNull()){
//                        deptIds.add(jsonObject.get("deptId").getAsLong());
//                    }
//                    if(jsonObject.get("userId") != null && !jsonObject.get("userId").isJsonNull()){
//                        userIds.add(jsonObject.get("userId").getAsLong());
//                    }
//                }
//            }
//            teacherPageDto.setDeptIds(deptIds);
//            List<XjrUserPageVo> attendanceTeacherList = teacherbaseManagerService.getList(teacherPageDto);
//
//            teacherPageDto = new BaseTeacherPageDto();
//            teacherPageDto.setUserIds(userIds);
//            attendanceTeacherList.addAll(teacherbaseManagerService.getList(teacherPageDto));
//            //将所有需要考勤的人的id存入set集合
//            Set<String> attendanceUserIdSet = attendanceTeacherList.stream().map(XjrUserPageVo::getId).collect(Collectors.toSet());
//
//            //查询所有人
//            teacherPageDto = new BaseTeacherPageDto();
//            if(dto.getDeptId() != null){
//                List<Long> dtoDeptIds = new ArrayList<>();
//                dtoDeptIds.add(dto.getDeptId());
//                teacherPageDto.setDeptIds(dtoDeptIds);
//            }
//
//            List<XjrUserPageVo> allTeacherList = teacherbaseManagerService.getList(teacherPageDto);
//            statisticsVo.setNotAttendanceCount(allTeacherList.stream().filter(x -> !attendanceUserIdSet.contains(x.getId())).collect(Collectors.toList()).size());
//        }


        //计算出勤率
        BigDecimal divide = BigDecimal.ZERO;
        if (!dividendCount.isEmpty()) {
            divide = BigDecimal.valueOf(statisticsVo.getActualCount()).divide(BigDecimal.valueOf(dividendCount.size()), 4, RoundingMode.HALF_UP);
        }
        statisticsVo.setAttendanceRate(divide.doubleValue() + "");

        return RT.ok(statisticsVo);
    }


    @GetMapping(value = "/student-statistics")
    @ApiOperation(value = "学生考勤统计")
    @SaCheckPermission("statistics:detail")
    @XjrLog(value = "学生考勤统计", saveResponseData = true)
    public RT<TeacherStatisticsVo> studentStatistics(@Valid AttendanceStatisticDto dto) {
        TeacherStatisticsVo statisticsVo = new TeacherStatisticsVo();
        //查询总人数
        MPJLambdaWrapper<XjrUser> queryWrapper = MPJWrappers.<XjrUser>lambdaJoin()
                .disableSubLogicDel()
                .distinct()
                .eq(ObjectUtil.isNotNull(dto.getGradeId()), BaseClass::getGradeId, dto.getGradeId())
                .eq(BaseStudentSchoolRoll::getArchivesStatus, "FB2901")
                .eq(ObjectUtil.isNotNull(dto.getClassId()), BaseStudentSchoolRoll::getClassId, dto.getClassId())
                .innerJoin(BaseStudentSchoolRoll.class, BaseStudentSchoolRoll::getUserId, XjrUser::getId)
                .innerJoin(BaseClass.class, BaseClass::getId, BaseStudentSchoolRoll::getClassId)
                .innerJoin(BaseStudent.class, BaseStudent::getUserId, XjrUser::getId);
        long allCount = xjrUserService.count(queryWrapper);

        queryWrapper.eq(BaseStudentSchoolRoll::getStduyStatus, StudyStatusEnum.AttendDaySchool.getCode());
        long attendDaySchoolCount = xjrUserService.count(queryWrapper);
        statisticsVo.setAllCount(allCount);
        if (dto.getDate() != null && !"".equals(dto.getDate())) {
            DateTimeFormatter formatter = DateTimeFormatter.ISO_DATE;
            LocalDate queryDate = LocalDate.parse(dto.getDate(), formatter);
            LocalDateTime startTime, endTime;
            if (dto.getTimePeriod() != null && dto.getTimePeriod() == 1) {
                startTime = queryDate.atTime(5, 0, 0);
                endTime = queryDate.atTime(12, 0, 0);
            } else if (dto.getTimePeriod() != null && dto.getTimePeriod() == 2) {
                startTime = queryDate.atTime(12, 0, 0);
                endTime = queryDate.atTime(18, 0, 0);
            } else {
                startTime = queryDate.atTime(0, 0, 0);
                endTime = queryDate.atTime(23, 59, 59);
            }

            List<StudentOutInRecord> outInRecords = studentOutInRecordService.list(
                    new MPJLambdaWrapper<StudentOutInRecord>()
                            .select(StudentOutInRecord.class, x -> VoToColumnUtil.fieldsToColumns(StudentOutInRecord.class).contains(x.getProperty()))
                            .innerJoin(BaseStudentSchoolRoll.class, BaseStudentSchoolRoll::getUserId, StudentOutInRecord::getUserId)
                            .innerJoin(BaseClass.class, BaseClass::getId, BaseStudentSchoolRoll::getClassId)
                            .le(StudentOutInRecord::getRecordTime, endTime)
                            .eq("DATE_FORMAT(record_time, '%Y-%m-%d')", endTime.toLocalDate())
                            .eq(StudentOutInRecord::getStatus, OutInStatusEnum.enter.getCode())
                            .eq(ObjectUtil.isNotNull(dto.getClassId()), BaseStudentSchoolRoll::getClassId, dto.getClassId())
                            .eq(ObjectUtil.isNotNull(dto.getGradeId()), BaseClass::getGradeId, dto.getGradeId())
            );
            //实到人数
            Set<Long> collect = outInRecords.stream().map(StudentOutInRecord::getUserId).collect(Collectors.toSet());
            statisticsVo.setActualCount(Long.valueOf(collect.size()));

            //查询学生请假人数
            Long leaveCount = studentLeaveService.getLeaveCount(startTime, endTime, dto);
            if (leaveCount == null) {
                leaveCount = 0L;
            }
            statisticsVo.setLeaveCount(leaveCount);

            //通过考勤规则和实到人数信息，计算迟到的
            Long lateCount = 0L;
            Integer playTruantCount = 0;
            for (StudentOutInRecord outInRecord : outInRecords) {
                if ("迟到".equals(outInRecord.getAttendanceStatus())) {
                    lateCount++;
                } else if ("旷课".equals(outInRecord.getAttendanceStatus())) {
                    playTruantCount++;
                }
            }
            statisticsVo.setPlayTruantCount(playTruantCount);
            statisticsVo.setLateCount(lateCount);


            //计算出勤率
            if (statisticsVo.getAllCount() != null && statisticsVo.getAllCount() != 0) {
                //最后通过总人数-实到人数-请假人数计算出缺勤人数
                statisticsVo.setAbsenteeismCount(attendDaySchoolCount - statisticsVo.getLeaveCount() - statisticsVo.getActualCount());
                BigDecimal divide = BigDecimal.valueOf(statisticsVo.getActualCount()).divide(BigDecimal.valueOf(attendDaySchoolCount), 4, RoundingMode.HALF_UP);
                statisticsVo.setAttendanceRate(divide.doubleValue() + "");
            } else {
                statisticsVo.setAttendanceRate("0");
                statisticsVo.setAbsenteeismCount(0L);
            }

        }

        return RT.ok(statisticsVo);
    }


    @GetMapping(value = "/visitor-list")
    @ApiOperation(value = "访客列表")
    @SaCheckPermission("statistics:detail")
    @XjrLog(value = "访客列表")
    public RT<List<VisitorInfoVo>> visitorList(@Valid AttendanceStatisticDto dto) {
        List<VisitorInfoVo> visionList = reservationSchoolService.getVisionList(dto.getDate());
        return RT.ok(visionList);
    }

}