package org.springblade.modules.vip.service.impl; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import org.springblade.core.mp.support.Condition; import org.springblade.modules.system.entity.User; import org.springblade.modules.training.entity.TrainingRegistration; import org.springblade.modules.vip.entity.UserVip; import org.springblade.modules.vip.mapper.UserVipMapper; import org.springblade.modules.vip.service.UserVipService; import org.springblade.modules.vip.vo.UserVipVO; import org.springframework.stereotype.Service; import java.util.Date; import java.util.List; /** * 用户会员服务实现类 * * @author zhongrj * @since 2022-02-21 */ @Service public class UserVipServiceImpl extends ServiceImpl implements UserVipService { /** * 自定义分页查询用户会员数据 * @param page * @param userVip * @return */ @Override public IPage selectUserVipPage(IPage page, UserVipVO userVip) { return page.setRecords(baseMapper.selectUserVipPage(page, userVip)); } /** * 新增会员信息 * @param trainingRegistration 报名信息对象 * @since 2022-02-22 * @return */ @Override public boolean insertUserVipInfo(TrainingRegistration trainingRegistration) { //先查询之前是否有缴费,比如缴费了取消报名这种情况 UserVip userVip1 = new UserVip(); userVip1.setUserId(Long.parseLong(trainingRegistration.getUserId())); List list = this.list(Condition.getQueryWrapper(userVip1)); boolean status = false; if (list.size()>0){ //更新数据 UserVip userVip = list.get(0); userVip.setApplyId(trainingRegistration.getId()); userVip.setUpdateTime(new Date()); //更新 status = this.updateById(userVip); }else { //1.创建用户会员对象 UserVip userVip = new UserVip(); //2.数据封装 userVip.setApplyId(trainingRegistration.getId()); userVip.setCreateTime(new Date()); userVip.setUpdateTime(new Date()); userVip.setUserId(Long.parseLong(trainingRegistration.getUserId())); status = this.save(userVip); } //3.返回 return status; } }