智慧保安后台管理-外网
钟日健
2022-02-22 f12f168ae5354e573b82969184b7e22181b96d08
src/main/java/org/springblade/modules/vip/service/impl/UserVipServiceImpl.java
@@ -3,6 +3,8 @@
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;
@@ -11,6 +13,7 @@
import org.springframework.stereotype.Service;
import java.util.Date;
import java.util.List;
/**
 *  用户会员服务实现类
@@ -41,14 +44,29 @@
    */
   @Override
   public boolean insertUserVipInfo(TrainingRegistration trainingRegistration) {
      //1.创建用户会员对象
      UserVip userVip = new UserVip();
      //2.数据封装
      userVip.setApplyId(trainingRegistration.getId());
      userVip.setCreateTime(new Date());
      userVip.setUpdateTime(new Date());
      userVip.setUserId(Long.parseLong(trainingRegistration.getUserId()));
      //先查询之前是否有缴费,比如缴费了取消报名这种情况
      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 this.save(userVip);
      return status;
   }
}