gunshi-project-ss/src/main/java/com/gunshi/project/ss/service/RegistrationInfoSerivce.java

37 lines
1.3 KiB
Java
Raw Normal View History

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<RegistrationInfoMapper, RegistrationInfo> {
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;
}
}
}