package com.chinaztt.mes.basic.service.impl;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
|
import com.chinaztt.mes.basic.entity.Company;
|
import com.chinaztt.mes.basic.entity.Division;
|
import com.chinaztt.mes.basic.entity.Factory;
|
import com.chinaztt.mes.basic.entity.Post;
|
import com.chinaztt.mes.basic.excel.PostData;
|
import com.chinaztt.mes.basic.mapper.CompanyMapper;
|
import com.chinaztt.mes.basic.mapper.DivisionMapper;
|
import com.chinaztt.mes.basic.mapper.FactoryMapper;
|
import com.chinaztt.mes.basic.mapper.PostMapper;
|
import com.chinaztt.mes.basic.service.PostService;
|
import com.chinaztt.mes.common.numgen.NumberGenerator;
|
import lombok.AllArgsConstructor;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.List;
|
import java.util.Objects;
|
|
/**
|
* 岗位表
|
*
|
* @author cxf
|
* @date 2021-06-23 17:17:26
|
*/
|
@Service
|
@AllArgsConstructor
|
@Transactional(rollbackFor = Exception.class)
|
public class PostServiceImpl extends ServiceImpl<PostMapper, Post> implements PostService {
|
private NumberGenerator<Post> postNumberGenerator;
|
private DivisionMapper divisionMapper;
|
private PostMapper postMapper;
|
private CompanyMapper companyMapper;
|
private FactoryMapper factoryMapper;
|
|
@Override
|
public boolean save(Post post) {
|
checkPostNo(post);
|
return SqlHelper.retBool(baseMapper.insert(post));
|
}
|
|
@Override
|
public boolean updateById(Post post) {
|
checkPostNo(post);
|
return SqlHelper.retBool(baseMapper.updateById(post));
|
}
|
|
private void checkPostNo(Post post) {
|
if (StringUtils.isBlank(post.getPostNo())) {
|
// 1. 自动生成编号
|
post.setPostNo(postNumberGenerator.generateNumberWithPrefix(Post.DIGIT, Post.PREFIX, Post::getPostNo));
|
} else {
|
// 2.判断是否重复
|
List<Post> repeatNos = baseMapper.selectList(Wrappers.<Post>query().lambda().eq(Post::getPostNo, post.getPostNo()).ne(Post::getId, post.getId()));
|
if (CollectionUtil.isNotEmpty(repeatNos)) {
|
throw new RuntimeException("编号重复");
|
}
|
}
|
}
|
|
//excel导入数据
|
@Override
|
public void importWorkstationExcel(List<PostData> list) {
|
if (CollectionUtil.isEmpty(list)) {
|
return;
|
}
|
List<Company> companies = companyMapper.selectList(null);
|
List<Factory> factories = factoryMapper.selectList(null);
|
List<Division> divisions = divisionMapper.selectList(null);
|
for (PostData data : list) {
|
QueryWrapper<Post>QueryWrapper = new QueryWrapper<>();
|
QueryWrapper.lambda().eq(Post::getPostNo,data.getPostNo());
|
Post post1 = baseMapper.selectOne(QueryWrapper);
|
if(!Objects.isNull(post1)){
|
log.error("岗位编号重复=====》"+data.getPostNo());
|
continue;
|
}
|
Post post = new Post();
|
BeanUtils.copyProperties(data, post);
|
//TODO: data.getDivisionName()+"-"+ 岗位组合 部门+岗位
|
post.setPostName(data.getPostName());
|
for (int i = 0; i <companies.size() ; i++) {
|
if (companies.get(i).getCompanyName().equals(data.getCompanyName())){
|
post.setCompanyId(companies.get(i).getId());
|
}
|
}
|
if (post.getCompanyId()==null){
|
log.error("公司:" + data.getCompanyName() + "不存在");
|
continue ;
|
}
|
for (int i = 0; i <factories.size() ; i++) {
|
if (factories.get(i).getFactoryName().equals(data.getFactoryName())){
|
post.setFactoryId(factories.get(i).getId());
|
}
|
}
|
if (post.getFactoryId()==null){
|
log.error("工厂:" + data.getFactoryName() + "不存在");
|
continue ;
|
}
|
for (int i = 0; i < divisions.size(); i++) {
|
if (divisions.get(i).getDivisionName().equals(data.getDivisionName())) {
|
post.setDivisionId(divisions.get(i).getId());
|
}
|
}
|
if (post.getDivisionId() == null) {
|
log.error("部门:" + data.getDivisionName() + "不存在");
|
continue;
|
}
|
baseMapper.insert(post);
|
}
|
}
|
}
|
