/* * 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.evaluate.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.core.mp.support.BaseEntityWrapper; import org.springblade.core.tool.utils.BeanUtil; import org.springblade.core.tool.utils.SpringUtil; import org.springblade.modules.evaluate.entity.EvaluateResultEntity; import org.springblade.modules.evaluate.entity.EvaluateTaskSetEntity; import org.springblade.modules.evaluate.service.IEvaluateResultService; import org.springblade.modules.evaluate.service.IEvaluateTaskService; import org.springblade.modules.evaluate.service.IEvaluateTaskSetService; import org.springblade.modules.evaluate.vo.EvaluateTaskSetVO; import java.util.List; import java.util.Objects; /** * 评优任务配置 包装类,返回视图层所需的字段 * * @author aix * @since 2023-12-21 */ public class EvaluateTaskSetWrapper extends BaseEntityWrapper { private static final IEvaluateResultService evaluateResultService; static { evaluateResultService = SpringUtil.getBean(IEvaluateResultService.class); } public static EvaluateTaskSetWrapper build() { return new EvaluateTaskSetWrapper(); } @Override public EvaluateTaskSetVO entityVO(EvaluateTaskSetEntity evaluateTaskSet) { EvaluateTaskSetVO evaluateTaskSetVO = Objects.requireNonNull(BeanUtil.copy(evaluateTaskSet, EvaluateTaskSetVO.class)); //User createUser = UserCache.getUser(evaluateTaskSet.getCreateUser()); //User updateUser = UserCache.getUser(evaluateTaskSet.getUpdateUser()); //evaluateTaskSetVO.setCreateUserName(createUser.getName()); //evaluateTaskSetVO.setUpdateUserName(updateUser.getName()); return evaluateTaskSetVO; } @Override public IPage pageVO(IPage pages) { List records = listVO(pages.getRecords()); for (EvaluateTaskSetVO vo : records) { QueryWrapper queryWrapper = new QueryWrapper<>(); queryWrapper.eq("be_id",vo.getUserId()); queryWrapper.eq("evaluate_task_id",vo.getEvaluateTaskId()); queryWrapper.eq("type", vo.getType()); vo.setVoteNum(evaluateResultService.count(queryWrapper)); } IPage pageVo = new Page<>(pages.getCurrent(), pages.getSize(), pages.getTotal()); pageVo.setRecords(records); return pageVo; } }