智慧保安后台管理-外网项目备份
钟日健
2022-02-22 a3d222eaa8204c783a4b4efb0df0eeb38fc639db
src/main/java/org/springblade/modules/vip/service/impl/UserVipServiceImpl.java
@@ -3,11 +3,17 @@
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;
/**
 *  用户会员服务实现类
@@ -29,4 +35,38 @@
   public IPage<UserVipVO> selectUserVipPage(IPage<UserVipVO> 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<UserVip> 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;
   }
}