package cn.teammodel.service.impl;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.lang.UUID;
import cn.teammodel.common.ErrorCode;
import cn.teammodel.common.FiveEducations;
import cn.teammodel.common.IdRequest;
import cn.teammodel.common.PK;
import cn.teammodel.config.exception.ServiceException;
import cn.teammodel.dao.*;
import cn.teammodel.model.dto.Appraise.*;
import cn.teammodel.model.entity.User;
import cn.teammodel.model.entity.appraise.*;
import cn.teammodel.model.entity.school.ClassInfo;
import cn.teammodel.model.entity.school.School;
import cn.teammodel.model.entity.school.Student;
import cn.teammodel.model.vo.appraise.AppraiseRecordVo;
import cn.teammodel.model.vo.appraise.StudentReportVo;
import cn.teammodel.security.utils.SecurityUtil;
import cn.teammodel.service.EvaluationService;
import cn.teammodel.utils.RepositoryUtil;
import cn.teammodel.utils.SchoolDateUtil;
import com.azure.cosmos.models.CosmosPatchOperations;
import com.azure.spring.data.cosmos.core.query.CosmosPageRequest;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.time.Instant;
import java.time.LocalDate;
import java.util.*;
import java.util.stream.Collectors;
/**
* @author winter
* @create 2023-11-20 17:47
*/
@Service
public class EvaluationServiceImpl implements EvaluationService {
/**
* 评价记录对象的类型
*/
private final static String TARGET_STUDENT = "student";
private final static String TARGET_CLASS = "class";
@Resource
private ClassRepository classRepository;
@Resource
private StudentRepository studentRepository;
@Resource
private SchoolRepository schoolRepository;
@Resource
private AppraiseRecordRepository appraiseRecordRepository;
@Resource
private AppraiseRepository appraiseRepository;
/**
* 通用的获取用户学校和学段对应的 appraise 的方法: 判断参数,判断数据是否为空
* 从 token 中获取 schoolId
*/
private Appraise findAppraise(String periodId) {
User loginUser = SecurityUtil.getLoginUser();
String schoolId = loginUser.getSchoolId();
// 拿到要新增节点的原始数据
Appraise appraise = appraiseRepository.findAppraiseBySchoolIdAndPeriodIdAndCode(schoolId, periodId, PK.PK_APPRAISE);
if (appraise == null) {
throw new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "该学段下评价项数据不存在");
}
return appraise;
}
@Override
public Appraise getTree(GetEvaluateTreeDto getEvaluateTreeDto) {
String periodId = getEvaluateTreeDto.getPeriodId();
User loginUser = SecurityUtil.getLoginUser();
String schoolId = loginUser.getSchoolId();
Appraise appraise = appraiseRepository.findAppraiseBySchoolIdAndPeriodIdAndCode(schoolId, periodId, PK.PK_APPRAISE);
// 如果传参 period 所在没有 tree 的话,则复制一份模板 tree 给他 (处理节点 id)
if (appraise == null) {
appraise = RepositoryUtil.findOne(appraiseRepository.findTemplateTree(), "获取模板评价树失败");
if (appraise == null) {
throw new ServiceException();
}
// refresh
refreshAppraiseTree(appraise);
appraise.setPeriodId(periodId);
appraise.setSchoolId(schoolId);
appraise.setId(null);
appraise = appraiseRepository.save(appraise);
}
return this.buildTree(appraise);
}
/**
* 将 appraise 进行替换(天才)
*/
private void refreshAppraiseTree(Appraise appraise) {
List nodes = appraise.getNodes();
List rules = appraise.getAchievementRules();
List children = nodes.stream().filter(item -> item.getPid() != null).collect(Collectors.toList());
// 将非 root 的 nodes 通过 pid 收集成 list, 再遍历 nodes, 将其每一个 id 对应 map 中的 list 的 pid修改成新的 id 即可
Map> pidNodeMap = children.stream().collect(Collectors.groupingBy(AppraiseTreeNode::getPid));
nodes.forEach(item -> {
String newId = UUID.randomUUID().toString();
String oldId = item.getId();
List appraiseTreeNodes = pidNodeMap.get(oldId);
if (appraiseTreeNodes != null) {
appraiseTreeNodes.forEach(node -> node.setPid(newId));
}
// 处理每个节点属性
item.setId(newId);
item.setCreator("template");
});
// refresh rules
rules.forEach(item -> item.setId(UUID.randomUUID().toString()));
}
@Override
public Appraise buildTree(Appraise appraise) {
if (appraise == null) {
return null;
}
List nodes = appraise.getNodes();
List parents = new ArrayList<>();
// pid 为 null 或者 "" 则为 parents
for (AppraiseTreeNode node : nodes) {
if (StringUtils.isBlank(node.getPid())) {
parents.add(node);
}
}
// 迭代构建孩子节点
for (AppraiseTreeNode parent : parents) {
buildChildren(parent, nodes);
}
// 排序
parents = parents.stream().sorted(Comparator.comparing(AppraiseTreeNode::getOrder)).collect(Collectors.toList());
appraise.setNodes(parents);
return appraise;
}
@Override
public void flattenTree(List trees, List nodes) {
// 递归结束条件
if (ObjectUtils.isEmpty(trees)) return;
// 这里的每个 tree 都是组装好的树,区别于单个节点
for (AppraiseTreeNode tree : trees) {
// 递归
flattenTree(tree.getChildren(), nodes);
// 回溯时删除 children,添加进列表
tree.setChildren(null);
nodes.add(tree);
}
}
@Override
public Appraise insertNode(InsertNodeDto insertNodeDto) {
Appraise appraise = findAppraise(insertNodeDto.getPeriodId());
User loginUser = SecurityUtil.getLoginUser();
List originNodes = appraise.getNodes();
// 拷贝数据到新节点
AppraiseTreeNode newNode = new AppraiseTreeNode();
BeanUtils.copyProperties(insertNodeDto, newNode);
String newNodePid = newNode.getPid();
// 根节点直接添加
if (StringUtils.isNotEmpty(newNodePid)) {
boolean invalid = originNodes.stream().noneMatch(item -> newNodePid.equals(item.getId()));
if (invalid) {
throw new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "父节点不存在");
}
}
newNode.setId(UUID.randomUUID().toString());
newNode.setCreatorId(loginUser.getId());
newNode.setCreator(loginUser.getName());
newNode.setCreateTime(Instant.now().toEpochMilli());
originNodes.add(newNode);
return buildTree(appraiseRepository.save(appraise));
}
@Override
public Appraise updateNode(UpdateNodeDto updateNodeDto) {
String updateNodeId = updateNodeDto.getId();
Appraise appraise = findAppraise(updateNodeDto.getPeriodId());
List originNodes = appraise.getNodes();
// 每个节点都有 id, 直接校验是否合法
AppraiseTreeNode updateNode = originNodes.stream()
.filter(item -> updateNodeId.equals(item.getId()))
.findFirst()
.orElseThrow(() -> new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "更新节点不存在"));
// 更新字段,考虑直接 copy properties
updateNode.setName(updateNodeDto.getName());
updateNode.setPath(updateNodeDto.getPath());
updateNode.setLogo(updateNodeDto.getLogo());
updateNode.setOrder(updateNodeDto.getOrder());
updateNode.setPraise(updateNodeDto.isPraise());
// todo: 为新节点赋值必须参数 (id, creator), 可不可以添加默认值 order ?
return buildTree(appraiseRepository.save(appraise));
}
@Override
public Appraise deleteNode(DeleteNodeDto deleteNodeDto) {
// 删除指定节点,可能是(一级,二级,三级),设计一个通用的
Appraise appraise = findAppraise(deleteNodeDto.getPeriodId());
List nodes = appraise.getNodes();
List nodesToDelete = new ArrayList<>();
// 迭代器安全删除
for (AppraiseTreeNode node : nodes) {
if (node.getId().equals(deleteNodeDto.getId())) {
// 删除当前节点
nodesToDelete.add(node);
// 递归删除其孩子节点
this.collectNodesToDelete(node.getId(), nodes, nodesToDelete);
}
}
if (ObjectUtils.isEmpty(nodesToDelete)) {
throw new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "删除节点不存在");
}
nodes.removeAll(nodesToDelete);
return buildTree(appraiseRepository.save(appraise));
}
@Override
public void vote(AppraiseVoteDto appraiseVoteDto) {
String targetId = appraiseVoteDto.getTargetId();
boolean spread = appraiseVoteDto.isSpread();
boolean pushParent = appraiseVoteDto.isPushParent();
String targetType = appraiseVoteDto.getTargetType();
String appraiseId = appraiseVoteDto.getAppraiseId();
User loginUser = SecurityUtil.getLoginUser();
String schoolId = loginUser.getSchoolId();
String classId;
String className;
String periodId;
String name;
String avatar = null;
String gender = null;
// 分别对班级和学生的关键信息取值
if (targetType.equals(TARGET_STUDENT)) {
List list = studentRepository.findByIdAndCode(targetId, String.format(PK.STUDENT, schoolId));
Student student = RepositoryUtil.findOne(list, "该学生不存在");
classId = student.getClassId();
periodId = student.getPeriodId();
name = student.getName();
avatar = student.getPicture();
gender = student.getGender();
// 获取班级信息
ClassInfo classInfo = classRepository.findClassByIdAndCode(classId, String.format(PK.CLASS, schoolId));
className = classInfo.getName();
} else if (targetType.equals(TARGET_CLASS)){
ClassInfo classInfo = classRepository.findClassByIdAndCode(targetId, String.format(PK.CLASS, schoolId));
if (classInfo == null) {
throw new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "班级不存在");
}
classId = targetId;
periodId = classInfo.getPeriodId();
name = classInfo.getName();
className = classInfo.getName();
} else {
throw new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "不受支持的评价对象");
}
// 获取评价项节点
List nodes = appraiseRepository.findNodeById(PK.PK_APPRAISE, appraiseId);
AppraiseTreeNode appraiseTreeNode = RepositoryUtil.findOne(nodes, "该评价项不存在");
if (appraiseTreeNode.getPath() == null) {
throw new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "仅能评价三级评价项");
}
// 通过 periodId 获取 semesters
List periodById = schoolRepository.findPeriodById(schoolId, periodId);
School.Period period = RepositoryUtil.findOne(periodById, "获取学段失败");
List semesters = period.getSemesters();
// 获取当前学年学期组合 ID
String academicYearId = SchoolDateUtil.calculateAcademicYearId(semesters, LocalDate.now());
// 查询是否存在记录
AppraiseRecord record = appraiseRecordRepository.findAppraiseRecordByTargetIdAndClassIdAndAcademicYearIdAndCode(
targetId,
classId,
academicYearId,
String.format(PK.PK_APPRAISE_RECORD,schoolId)
);
// 初始化新的评价节点
AppraiseRecordItem item = new AppraiseRecordItem();
item.setId(UUID.randomUUID().toString());
item.setAppraiseNode(appraiseTreeNode);
item.setCreator(loginUser.getName());
item.setCreatorId(loginUser.getId());
item.setCreateTime(Instant.now().toEpochMilli());
// 处理学校与学生的差异
if (targetType.equals(TARGET_CLASS)){
item.setSpread(spread);
} else {
item.setPushParent(pushParent);
}
// 不存在或者学生的班级不一样则创建一条新的,存在则处理一下分值后再向其 nodes 中插入一条 item
if (record == null || !classId.equals(record.getClassId())) {
List items = Collections.singletonList(item);
record = new AppraiseRecord();
record.setTargetId(targetId);
record.setTargetType(targetType);
record.setClassId(classId);
record.setClassName(className);
record.setName(name);
record.setAvatar(avatar);
record.setGender(gender);
record.setAcademicYearId(academicYearId);
record.setPraiseCount(appraiseTreeNode.isPraise() ? 1 : -1);
record.setScore(ObjectUtils.isEmpty(appraiseTreeNode.getScore()) ? 0 : appraiseTreeNode.getScore());
record.setNodes(items);
record.setCode(String.format(PK.PK_APPRAISE_RECORD, schoolId));
appraiseRecordRepository.save(record);
} else {
CosmosPatchOperations operations = CosmosPatchOperations.create();
operations.add("/nodes/-", item);
// 表扬 (待改进不会减少表扬数)
if (appraiseTreeNode.isPraise()) {
operations.increment("/praiseCount", 1);
}
// 加分
int scoreToPlus = ObjectUtils.isEmpty(appraiseTreeNode.getScore()) ? 0 : appraiseTreeNode.getScore();
operations.increment("/score", scoreToPlus);
// patch doc
appraiseRecordRepository.save(record.getId(), PK.buildOf(PK.PK_APPRAISE_RECORD, schoolId), AppraiseRecord.class, operations);
}
}
@Override
public List findVoteRecord(FindVoteRecordDto findVoteRecordDto) {
String periodId = findVoteRecordDto.getPeriodId();
String targetId = StringUtils.isBlank(findVoteRecordDto.getTargetId()) ? null : findVoteRecordDto.getTargetId();
String targetType = StringUtils.isBlank(findVoteRecordDto.getTargetType()) ? null : findVoteRecordDto.getTargetType();
String classId = StringUtils.isBlank(findVoteRecordDto.getClassId()) ? null : findVoteRecordDto.getClassId();
Boolean isPraise = findVoteRecordDto.getIsPraise();
User loginUser = SecurityUtil.getLoginUser();
String teacherId = loginUser.getId();
String schoolId = loginUser.getSchoolId();
List semesters = schoolRepository.findSemestersById(schoolId, periodId);
String academicYearId = SchoolDateUtil.calculateAcademicYearId(semesters, LocalDate.now());
// 分页
Sort sort = Sort.by("createTime").descending();
final CosmosPageRequest pageRequest = new CosmosPageRequest(findVoteRecordDto.getCurrent(), findVoteRecordDto.getSize(), null, sort);
Page appraiseRecordItemPage = appraiseRecordRepository.searchNodesByCondition(
targetId,
targetType,
classId,
teacherId,
academicYearId,
isPraise,
String.format(PK.PK_APPRAISE_RECORD, schoolId),
pageRequest
);
List content = appraiseRecordItemPage.getContent();
return content;
}
@Override
public void recallVote(RecallVoteDto recallVoteDto) {
String recordId = recallVoteDto.getRecordId();
String nodeId = recallVoteDto.getNodeId();
User loginUser = SecurityUtil.getLoginUser();
String schoolId = loginUser.getSchoolId();
String userId = loginUser.getId();
Optional optional = appraiseRecordRepository.findById(recordId, PK.buildOf(PK.PK_APPRAISE_RECORD, schoolId));
AppraiseRecord appraiseRecord = optional.orElseThrow(() -> new ServiceException("该记录不存在"));
AppraiseRecordItem record = appraiseRecord.getNodes()
.stream()
.filter(item -> nodeId.equals(item.getId()))
.findFirst().orElseThrow(() -> new ServiceException("该记录节点不存在"));
// 鉴权(不是创建老师不能撤回)
if (!loginUser.getRoles().contains("admin") && !userId.equals(record.getCreatorId())) {
throw new ServiceException(ErrorCode.NO_AUTH_ERROR.getCode(), "您不是创建老师,不能撤回");
}
// 删除评价项并且恢复评分
appraiseRecord.getNodes().removeIf(item -> nodeId.equals(item.getId()));
boolean praise = record.getAppraiseNode().isPraise();
Integer newPraiseCount = appraiseRecord.getPraiseCount() + (praise ? -1 : 1);
appraiseRecord.setPraiseCount(newPraiseCount);
int score = record.getAppraiseNode().getScore() == null ? 0 : record.getAppraiseNode().getScore();
Integer newScore = appraiseRecord.getScore() - score;
appraiseRecord.setScore(newScore);
// 保存
appraiseRecordRepository.save(appraiseRecord);
}
@Override
public StudentReportVo studentReport(IdRequest idRequest) {
String studentId = idRequest.getId();
User user = SecurityUtil.getLoginUser();
String schoolId = user.getSchoolId();
// 查询学生信息,学段等
Student student = RepositoryUtil.findOne(studentRepository.findByIdAndCode(studentId, String.format(PK.STUDENT, schoolId)), "当前学生不存在");
String periodId = student.getPeriodId();
String classId = student.getClassId();
List semesters = schoolRepository.findSemestersById(schoolId, periodId);
// 生成学年 id
String academicYearId = SchoolDateUtil.calculateAcademicYearId(semesters, LocalDate.now());
// 查询评价文档
AppraiseRecord appraiseRecord = appraiseRecordRepository.findAppraiseRecordByTargetIdAndClassIdAndAcademicYearIdAndCode(
studentId,
classId,
academicYearId,
String.format(PK.PK_APPRAISE_RECORD, schoolId)
);
// 班级内学生总数
int stuInClassCount = studentRepository.countByClassIdAndCode(classId, String.format(PK.STUDENT, schoolId));
if (appraiseRecord == null || appraiseRecord.getNodes() == null) {
throw new ServiceException(ErrorCode.OPERATION_ERROR.getCode(), "当前学生暂未产生评价数据,无法生成报告");
}
Integer praiseCount = appraiseRecord.getPraiseCount();
List records = appraiseRecord.getNodes();
records = records.stream().sorted(Comparator.comparing(AppraiseRecordItem::getCreateTime).reversed()).collect(Collectors.toList());
// 查询成就规则
Appraise appraise = RepositoryUtil.findOne(appraiseRepository.findRulesById(schoolId, periodId), "当前成就规则还未创建");
List rules = appraise.getAchievementRules();
if (CollectionUtil.isEmpty(rules)) {
throw new ServiceException(ErrorCode.OPERATION_ERROR.getCode(), "当前成就规则为空");
}
// 计算雷达图
Map praiseDistribution = StudentReportVo.ofFiveEducation();
Map criticalDistribution = StudentReportVo.ofFiveEducation();
// 计算数据
Map praiseTeacherMap = new HashMap<>();
Map criticalTeacherMap = new HashMap<>();
Map praiseNodeMap = new HashMap<>();
Map criticalNodeMap = new HashMap<>();
// 计算当前学生排名在全班次位
int greaterCount = appraiseRecordRepository.findClassRecord(String.format(PK.PK_APPRAISE_RECORD, schoolId), academicYearId, classId, praiseCount);
float beyondPercent = 0;
long currentRank = stuInClassCount - greaterCount;
beyondPercent = (float) currentRank / stuInClassCount;
// 根据全部数据计算结果
for (AppraiseRecordItem record : records) {
AppraiseTreeNode appraiseNode = record.getAppraiseNode();
String[] path = appraiseNode.getPath();
// 数据异常则跳过
if (path == null || path.length == 0){
continue;
}
String root = path[0];
// 一级评价项不在预期中
String key = FiveEducations.getCodeByName(root);
// 计算雷达图
if (key != null) {
if (appraiseNode.isPraise()) {
praiseDistribution.put(key, praiseDistribution.getOrDefault(key, 0) + 1);
} else {
criticalDistribution.put(key, criticalDistribution.getOrDefault(key, 0) + 1);
}
}
String teacherId = record.getCreatorId();
String nodeName = appraiseNode.getName();
// 根据评价是否为表扬计算数据
if (appraiseNode.isPraise()) {
praiseTeacherMap.put(teacherId, praiseTeacherMap.getOrDefault(teacherId, 0) + 1);
praiseNodeMap.put(nodeName, praiseNodeMap.getOrDefault(nodeName, 0) + 1);
} else {
criticalTeacherMap.put(teacherId, criticalTeacherMap.getOrDefault(teacherId, 0) + 1);
criticalNodeMap.put(nodeName, criticalNodeMap.getOrDefault(nodeName, 0) + 1);
}
}
// 计算成就项 (排序 120(20x3) 60(10x4) 25(5x5) 121
rules = rules.stream().sorted(Comparator.comparing(AchievementRule::getLevel).reversed()).collect(Collectors.toList());
int ruleSize = rules.size();
AchievementRule curAchievement = rules.get(ruleSize - 1);
int n = 1;
for (int i = 0; i <= ruleSize - 1; i++) {
// 头尾的情况要特判(理清逻辑就很简单)
AchievementRule curRule = rules.get(i);
int flag = praiseCount / curRule.getPromotionCount();
// 只有一个节点也特殊处理
if (ruleSize == 1) {
n = praiseCount / curRule.getLevelCount() + 1;
break;
}
if (flag >= 1) {
if (i == 0) {
curAchievement = curRule;
AchievementRule lastRule = rules.get(i + 1);
n = (praiseCount - lastRule.getPromotionCount()) / curRule.getLevelCount() + 1;
} else {
AchievementRule nextRule = rules.get(i - 1);
curAchievement = nextRule;
n = (praiseCount - curRule.getPromotionCount()) / nextRule.getLevelCount() + 1;
}
break;
}
if (i == ruleSize - 1) {
n = praiseCount / curRule.getLevelCount() + 1;
}
}
// 汇总数据
String topPraiseTeacherId = praiseTeacherMap.entrySet().stream().max(Map.Entry.comparingByValue()).map(Map.Entry::getKey).orElse(null);
AppraiseRecordItem tmp1 = records.stream().filter(item -> item.getCreatorId().equals(topPraiseTeacherId)).findFirst().orElse(null);
String topPraiseTeacherName = tmp1 == null ? null : tmp1.getCreator();
String topCriticalTeacherId = criticalTeacherMap.entrySet().stream().max(Map.Entry.comparingByValue()).map(Map.Entry::getKey).orElse(null);
AppraiseRecordItem tmp2 = records.stream().filter(item -> item.getCreatorId().equals(topCriticalTeacherId)).findFirst().orElse(null);
String topCriticalTeacherName = tmp2 == null ? null : tmp2.getCreator();
String topPraiseNode = praiseNodeMap.entrySet().stream().max(Map.Entry.comparingByValue()).map(Map.Entry::getKey).orElse(null);
String topCriticalNode = criticalNodeMap.entrySet().stream().max(Map.Entry.comparingByValue()).map(Map.Entry::getKey).orElse(null);
return StudentReportVo.builder()
.praiseCount(praiseCount)
.score(appraiseRecord.getScore())
.beyondPercent(beyondPercent)
.topPraiseTeacher(new StudentReportVo.Teacher(topPraiseTeacherId, topPraiseTeacherName))
.topCriticalTeacher(new StudentReportVo.Teacher(topCriticalTeacherId, topCriticalTeacherName))
.topPraiseNode(topPraiseNode)
.topCriticalNode(topCriticalNode)
.praiseDistribution(praiseDistribution)
.criticalDistribution(criticalDistribution)
.curAchievement(curAchievement)
.achievementN(n)
.records(records)
.build();
}
/**
* 递归收集 id 的节点及 id 节点的孩子节点 (迭代器删除居然也报错)
*/
private void collectNodesToDelete(String id, List nodes, List nodesToDelete) {
for (AppraiseTreeNode node : nodes) {
if (id.equals(node.getPid())) {
// 收集
nodesToDelete.add(node);
// 递归收集
collectNodesToDelete(node.getId(), nodes, nodesToDelete);
}
}
}
/**
* 递归的构建父亲节点的孩子,以及孩子的孩子 (理论支持无极树,但应该考虑是否增加递归深度)
*/
private void buildChildren(AppraiseTreeNode parent, List nodes) {
List children = getChildren(parent, nodes);
if (children.size() == 0) return;
parent.setChildren(children);
for (AppraiseTreeNode child : children) {
// 如果子节点还有孩子的话,就继续递归构建
if (getChildren(child, nodes).size() > 0) {
buildChildren(child, nodes);
}
}
}
/**
* 获取节点的孩子节点列表
*/
private List getChildren(AppraiseTreeNode parent, List nodes) {
List children = new ArrayList<>();
for (AppraiseTreeNode node : nodes) {
if (StringUtils.isNotBlank(node.getPid()) && node.getPid().equals(parent.getId())) {
children.add(node);
}
}
return children;
}
}