From 15e30e0207156758d8b714a1ad00232aac7b8cf0 Mon Sep 17 00:00:00 2001
From: yz <yz_0812@outlook.com>
Date: 星期二, 02 三月 2021 01:04:51 +0800
Subject: [PATCH] 修改bug

---
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpBadRecordServiceImpl.java |   24 +++++++++++++++++++++---
 1 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpBadRecordServiceImpl.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpBadRecordServiceImpl.java
index 9be1689..94d7f6d 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpBadRecordServiceImpl.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpBadRecordServiceImpl.java
@@ -1,6 +1,9 @@
 package cc.mrbird.febs.server.hr.service.impl;
 
 import cc.mrbird.febs.common.core.constant.ModuleCode;
+import cc.mrbird.febs.common.core.exception.FebsException;
+import cc.mrbird.febs.server.hr.constant.MyConstant;
+import cc.mrbird.febs.server.hr.entity.EmpAccidentCases;
 import cc.mrbird.febs.server.hr.entity.EmpBadRecord;
 import cc.mrbird.febs.server.hr.entity.EmpBadRecord;
 import cc.mrbird.febs.server.hr.mapper.EmpBadRecordMapper;
@@ -81,7 +84,14 @@
 
     @Override
     @Transactional(rollbackFor = Exception.class)
-    public void createEmpBadrecord(EmpBadRecord empBadrecord) {
+    public void createEmpBadrecord(EmpBadRecord empBadrecord) throws FebsException {
+        LambdaQueryWrapper<EmpBadRecord> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+        lambdaQueryWrapper.eq(EmpBadRecord::getEmpId, empBadrecord.getEmpId());
+        lambdaQueryWrapper.eq(EmpBadRecord::getBadDate, empBadrecord.getBadDate());
+        lambdaQueryWrapper.ne(EmpBadRecord::getDelFlag,1);
+        if (this.count(lambdaQueryWrapper) > 0) {
+            throw new FebsException(MyConstant.MESSAGE);
+        }
         empBadrecord.setBadId(SequenceUtil.generateId(0L, ModuleCode.HR_EMPLOYEE));
         empBadrecord.setCreator(operatorId);
         empBadrecord.setModifier(operatorId);
@@ -90,7 +100,15 @@
 
     @Override
     @Transactional(rollbackFor = Exception.class)
-    public void updateEmpBadrecord(EmpBadRecord empBadrecord) {
+    public void updateEmpBadrecord(EmpBadRecord empBadrecord) throws FebsException {
+        LambdaQueryWrapper<EmpBadRecord> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+        lambdaQueryWrapper.eq(EmpBadRecord::getEmpId, empBadrecord.getEmpId());
+        lambdaQueryWrapper.eq(EmpBadRecord::getBadDate, empBadrecord.getBadDate());
+        lambdaQueryWrapper.ne(EmpBadRecord::getDelFlag,1);
+        lambdaQueryWrapper.ne(EmpBadRecord::getBadId,empBadrecord.getBadId());
+        if (this.count(lambdaQueryWrapper) > 0) {
+            throw new FebsException(MyConstant.MESSAGE);
+        }
         EmpBadRecord dbData = this.getById(empBadrecord.getBadId());
         empBadrecord.setCreateTime(dbData.getCreateTime());
         empBadrecord.setCreator(dbData.getCreator());
@@ -156,4 +174,4 @@
 			this.save(empBadrecord);
     	}
 	}
-}
\ No newline at end of file
+}

--
Gitblit v1.8.0