package com.zy.asrs.service.impl; import com.baomidou.mybatisplus.mapper.EntityWrapper; import com.core.common.Cools; import com.core.exception.CoolException; import com.zy.asrs.mapper.DocTypeMapper; import com.zy.asrs.entity.DocType; import com.zy.asrs.service.DocTypeService; import com.baomidou.mybatisplus.service.impl.ServiceImpl; import org.springframework.stereotype.Service; import java.util.Date; @Service("docTypeService") public class DocTypeServiceImpl extends ServiceImpl implements DocTypeService { @Override public DocType selectOrAdd(String docName, Boolean pakin) { if (Cools.isEmpty(docName)) { return null; } DocType docType = this.selectOne(new EntityWrapper().eq("doc_name", docName)); if (docType == null) { long docId = 1L; DocType last = this.selectOne(new EntityWrapper().orderBy("doc_id", false)); if (last != null) { docId = last.getDocId() + 1; } docType = new DocType(); docType.setDocId(docId); docType.setDocName(docName); if (null != pakin) { docType.setPakin(pakin ? 1 : 0); docType.setPakout(pakin ? 0 : 1); } docType.setStatus(1); docType.setCreateTime(new Date()); docType.setUpdateTime(new Date()); if (!this.insert(docType)) { throw new CoolException("单据类型错误"); } } return docType; } }