From aae41c881d4082a5ee6200cd2ef7990faa06306b Mon Sep 17 00:00:00 2001
From: jiangping <jp@doumee.com>
Date: 星期五, 26 一月 2024 11:22:49 +0800
Subject: [PATCH] 开发业务接口

---
 server/service/src/main/java/com/doumee/service/system/impl/SystemUserServiceImpl.java |   34 +++++++++++++++++++++++++++++++++-
 1 files changed, 33 insertions(+), 1 deletions(-)

diff --git a/server/service/src/main/java/com/doumee/service/system/impl/SystemUserServiceImpl.java b/server/service/src/main/java/com/doumee/service/system/impl/SystemUserServiceImpl.java
index 01e6014..7160b8b 100644
--- a/server/service/src/main/java/com/doumee/service/system/impl/SystemUserServiceImpl.java
+++ b/server/service/src/main/java/com/doumee/service/system/impl/SystemUserServiceImpl.java
@@ -5,6 +5,7 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.doumee.biz.system.SystemDictDataBiz;
+import com.doumee.core.model.LoginUserInfo;
 import com.doumee.core.model.PageData;
 import com.doumee.core.model.PageWrap;
 import com.doumee.core.utils.Constants;
@@ -25,6 +26,7 @@
 import com.github.pagehelper.PageInfo;
 import com.github.yulichang.wrapper.MPJLambdaWrapper;
 import org.apache.commons.lang3.StringUtils;
+import org.apache.shiro.SecurityUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -102,9 +104,16 @@
 
     @Override
     public SystemUser findById(Integer id) {
+        LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+
         MPJLambdaWrapper<SystemUser> queryWrapper = new MPJLambdaWrapper<>();
         queryWrapper.selectAll(SystemUser.class);
         queryWrapper.eq(SystemUser::getId,id );
+        queryWrapper.eq(SystemUser::getType,loginUserInfo.getType() );
+        if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){
+            //濡傛灉鏄紒涓氳处鍙�
+            queryWrapper.eq(SystemUser::getCompanyId,loginUserInfo.getCompanyId() );
+        }
         queryWrapper.orderByDesc(SystemUser::getCreateTime);
         SystemUser  result = systemUserJoinMapper.selectJoinOne( SystemUser.class, queryWrapper);
         return result;
@@ -118,6 +127,12 @@
 
     @Override
     public List<SystemUser> findList(SystemUser systemUser) {
+        LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+        systemUser.setType(loginUserInfo.getType());
+        if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){
+            //濡傛灉鏄紒涓氳处鍙�
+            systemUser.setCompanyId(loginUserInfo.getCompanyId());
+        }
         Wrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser);
         return systemUserMapper.selectList(wrapper);
     }
@@ -131,6 +146,12 @@
 //            List<SystemDepartment> list = systemDepartmentService.findList(new QueryWrapper<>());
 //            List<Integer> collect = list.stream().map(s -> s.getId()).collect(Collectors.toList());
 //            pageWrap.getModel().setDepartmentIds(collect);
+        }
+        LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+        pageWrap.getModel().setType(loginUserInfo.getType());
+        if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){
+            //濡傛灉鏄紒涓氳处鍙�
+            pageWrap.getModel().setCompanyId(loginUserInfo.getCompanyId());
         }
         // 鎵ц鏌ヨ
         PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity());
@@ -202,7 +223,12 @@
 
     @Override
     public PageData<SystemUser> findAllList(PageWrap<SystemUser> pageWrap) {
-
+        LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+        pageWrap.getModel().setType(loginUserInfo.getType());
+        if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){
+            //濡傛灉鏄紒涓氳处鍙�
+            pageWrap.getModel().setCompanyId(loginUserInfo.getCompanyId());
+        }
         IPage<SystemUser> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());
         MPJLambdaWrapper<SystemUser> queryWrapper = new MPJLambdaWrapper<>();
         Utils.MP.blankToNull(pageWrap.getModel());
@@ -224,6 +250,12 @@
 
     @Override
     public List<SystemUser> findAllList(SystemUser pageWrap){
+        LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+        pageWrap.setType(loginUserInfo.getType());
+        if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){
+            //濡傛灉鏄紒涓氳处鍙�
+            pageWrap.setCompanyId(loginUserInfo.getCompanyId());
+        }
         MPJLambdaWrapper<SystemUser> queryWrapper = new MPJLambdaWrapper<>();
 
         queryWrapper.selectAll(SystemUser.class);

--
Gitblit v1.9.3