/* * Copyright (c) 2018-2028, Chill Zhuang All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * Neither the name of the dreamlu.net developer nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * Author: Chill 庄骞 (smallchill@163.com) */ package org.springblade.modules.assessment.wrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import org.springblade.common.cache.SysCache; import org.springblade.common.cache.UserCache; import org.springblade.core.mp.support.BaseEntityWrapper; import org.springblade.core.tool.utils.BeanUtil; import org.springblade.core.tool.utils.SpringUtil; import org.springblade.modules.assessment.entity.AssessmentScoreEntity; import org.springblade.modules.assessment.entity.AssessmentSetDeptEntity; import org.springblade.modules.assessment.entity.AssessmentSetEntity; import org.springblade.modules.assessment.service.IAssessmentScoreService; import org.springblade.modules.assessment.service.IAssessmentSetDeptService; import org.springblade.modules.assessment.service.IAssessmentSetService; import org.springblade.modules.assessment.service.IAssessmentTaskService; import org.springblade.modules.assessment.vo.AssessmentScoreVO; import org.springblade.modules.assessment.vo.AssessmentSetVO; import org.springblade.modules.system.entity.User; import java.util.List; import java.util.Objects; import java.util.stream.Collectors; /** * 考核评分 包装类,返回视图层所需的字段 * * @author aix * @since 2023-12-13 */ public class AssessmentScoreWrapper extends BaseEntityWrapper { private static final IAssessmentSetService assessmentSetService; private static final IAssessmentSetDeptService assessmentSetDeptService; private static final IAssessmentScoreService assessmentScoreService; private static final IAssessmentTaskService assessmentTaskService; static { assessmentSetService = SpringUtil.getBean(IAssessmentSetService.class); assessmentSetDeptService = SpringUtil.getBean(IAssessmentSetDeptService.class); assessmentScoreService = SpringUtil.getBean(IAssessmentScoreService.class); assessmentTaskService = SpringUtil.getBean(IAssessmentTaskService.class); } public static AssessmentScoreWrapper build() { return new AssessmentScoreWrapper(); } @Override public AssessmentScoreVO entityVO(AssessmentScoreEntity assessmentScore) { AssessmentScoreVO assessmentScoreVO = Objects.requireNonNull(BeanUtil.copy(assessmentScore, AssessmentScoreVO.class)); //User createUser = UserCache.getUser(assessmentScore.getCreateUser()); //User updateUser = UserCache.getUser(assessmentScore.getUpdateUser()); //assessmentScoreVO.setCreateUserName(createUser.getName()); //assessmentScoreVO.setUpdateUserName(updateUser.getName()); return assessmentScoreVO; } @Override public IPage pageVO(IPage pages) { List records = listVO(pages.getRecords()); //开始处理 for (AssessmentScoreVO vo: records) { if (vo.getType() == 0) { User user = UserCache.getUser(vo.getBeId()); //被考核人 vo.setUserName(user.getName()); vo.setCode(user.getCode()); vo.setDeptName(SysCache.getDeptName(Long.valueOf(user.getDeptId()))); vo.setPostName(SysCache.getPostName(Long.valueOf(user.getPostId()))); } else { vo.setDeptName(SysCache.getDeptName(Long.valueOf(vo.getBeId()))); } User scoreUser = UserCache.getUser(vo.getScoreUserId()); //考核人 vo.setScoreUserName(scoreUser.getName()); vo.setScoreDeptName(SysCache.getDeptName(Long.valueOf(scoreUser.getDeptId()))); vo.setScorePostName(SysCache.getPostName(Long.valueOf(scoreUser.getPostId()))); } IPage pageVo = new Page<>(pages.getCurrent(), pages.getSize(), pages.getTotal()); pageVo.setRecords(records); return pageVo; } @Override public List listVO(List list) { return list.stream().map(this::entityVO).collect(Collectors.toList()); } }