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<DocTypeMapper, DocType> implements DocTypeService { 
 | 
  
 | 
    @Override 
 | 
    public DocType selectOrAdd(String docName, Boolean pakin) { 
 | 
        if (Cools.isEmpty(docName)) { 
 | 
            return null; 
 | 
        } 
 | 
        DocType docType = this.selectOne(new EntityWrapper<DocType>().eq("doc_name", docName)); 
 | 
        if (docType == null) { 
 | 
            long docId = 1L; 
 | 
            DocType last = this.selectOne(new EntityWrapper<DocType>().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; 
 | 
    } 
 | 
} 
 |