From d83deacfb6d7ca1a4a8e3133c6e79ab818de9d5e Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期一, 29 四月 2024 11:06:11 +0800
Subject: [PATCH] 通知改造

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java |   22 ++++++++++++++++++++++
 1 files changed, 22 insertions(+), 0 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java
index a0fb489..4ffd0d0 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java
@@ -20,6 +20,7 @@
 import com.ycl.system.mapper.SysRoleMapper;
 import com.ycl.system.mapper.SysUserMapper;
 import com.ycl.system.mapper.SysUserRoleMapper;
+import com.ycl.system.model.LoginUser;
 import com.ycl.system.page.PageUtil;
 import com.ycl.platform.service.YwUnitService;
 import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -217,6 +218,27 @@
     }
 
     @Override
+    public void setUnitInfo(LoginUser loginUser) {
+        // 鍗曚綅绠$悊鍛樻煡璇�
+        YwUnit unit = new LambdaQueryChainWrapper<>(baseMapper)
+                .eq(YwUnit::getUnitAdminAccount, loginUser.getUsername())
+                .one();
+        if (Objects.nonNull(unit)) {
+            loginUser.setUnitAdmin(Boolean.TRUE);
+            loginUser.setUnitId(unit.getId());
+        } else {
+            unit = baseMapper.getByUserId(loginUser.getUserId());
+            if (Objects.nonNull(unit)) {
+                loginUser.setUnitAdmin(Boolean.FALSE);
+                loginUser.setUnitId(unit.getId());
+            }
+            else {
+                loginUser.setUnitAdmin(Boolean.FALSE);
+            }
+        }
+    }
+
+    @Override
     public List<YwUnitVO> export(YwUnitQuery query) {
         // 鍒嗛〉鏉′欢鏌ヨ浣�
         IPage<YwUnit> page = new LambdaQueryChainWrapper<>(baseMapper)

--
Gitblit v1.8.0