package com.gunshi.project.ss.service; import com.baomidou.mybatisplus.core.toolkit.IdWorker; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.gunshi.project.ss.mapper.PersonnelPlanMapper; import com.gunshi.project.ss.mapper.RegistrationInfoMapper; import com.gunshi.project.ss.model.PersonnelPlan; import com.gunshi.project.ss.model.RegistrationInfo; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; @Service @Slf4j @Transactional(rollbackFor = Exception.class) public class RegistrationInfoSerivce extends ServiceImpl { public RegistrationInfo addOrEdit(RegistrationInfo registrationInfo) { if(registrationInfo.getId()==null){ RegistrationInfo queryOne = this.baseMapper.selectOne(null); if(queryOne==null){ RegistrationInfo info = new RegistrationInfo(); long id = IdWorker.getId(); info.setId(id); save(info); return info; }else{ return queryOne; } }else{ RegistrationInfo byId = getById(registrationInfo.getId()); return byId; } } }