From 65ad59fc53b0fee72bb06da3b89bcc4867848ee8 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期六, 09 四月 2022 14:06:15 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/asrs/service/impl/DocTypeServiceImpl.java |   29 +++++++++++++++++++++++++++++
 1 files changed, 29 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/zy/asrs/service/impl/DocTypeServiceImpl.java b/src/main/java/com/zy/asrs/service/impl/DocTypeServiceImpl.java
index 0192d5c..b7eef61 100644
--- a/src/main/java/com/zy/asrs/service/impl/DocTypeServiceImpl.java
+++ b/src/main/java/com/zy/asrs/service/impl/DocTypeServiceImpl.java
@@ -1,12 +1,41 @@
 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) {
+        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);
+            docType.setStatus(1);
+            docType.setCreateTime(new Date());
+            docType.setUpdateTime(new Date());
+            if (!this.insert(docType)) {
+                throw new CoolException("鍗曟嵁绫诲瀷閿欒");
+            }
+        }
+        return docType;
+    }
 }

--
Gitblit v1.9.1