From 49d7ce5a988596d7540083be0e1948bcb66af544 Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Mon, 30 Oct 2023 17:12:03 +0800
Subject: [PATCH] chore:补充提交

---
 screen-api/src/main/java/com/moral/api/service/impl/OrganizationServiceImpl.java |   40 ++++++++++++++++++++++++++++++++++++++++
 1 files changed, 40 insertions(+), 0 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/OrganizationServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/OrganizationServiceImpl.java
index f407aa4..b009485 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/OrganizationServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/OrganizationServiceImpl.java
@@ -1,8 +1,11 @@
 package com.moral.api.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.moral.api.config.Interceptor.UserHelper;
 import com.moral.api.entity.Organization;
 import com.moral.api.mapper.OrganizationMapper;
+import com.moral.api.pojo.vo.user.QxUser;
+import com.moral.api.service.DeviceService;
 import com.moral.api.service.OrganizationService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.moral.constant.Constants;
@@ -27,12 +30,48 @@
     @Autowired
     OrganizationMapper organizationMapper;
 
+    @Autowired
+    private DeviceService deviceService;
+
+
     @Override
     public List<Organization> getChildrenOrganizationsById(Integer id) {
         List<Organization> childrenOrganization = new ArrayList<>();
         recursionQueryChildren(id,childrenOrganization);
         return childrenOrganization;
     }
+
+    @Override
+    public Organization getOrganizationById(Integer id) {
+        return organizationMapper.selectById(id);
+    }
+
+    /**
+     * ������������������������
+     *
+     * @return
+     */
+    @Override
+    public List<Organization> getOrganizations() {
+        QueryWrapper<Organization> wrapper = new QueryWrapper<>();
+        wrapper.select("id","name");
+        QxUser user = UserHelper.getCurrentUser();
+        Integer organizationId = user.getOrganizationId();
+        if (organizationId!=24){
+            wrapper.eq("id",user.getOrganizationId());
+        }
+            wrapper.eq("is_delete",Constants.NOT_DELETE);
+        List<Organization> organizations = organizationMapper.selectList(wrapper);
+        for (int i=0; i<organizations.size(); i++) {
+            List<String> macs = deviceService.getMacsByOrganizationId(organizations.get(i).getId());
+            if (ObjectUtils.isEmpty(macs)){
+                organizations.remove(i);
+                i--;
+            }
+        }
+        return organizations;
+    }
+
 
     /**
     * @Description: ���������������������������������������������������children���
@@ -43,6 +82,7 @@
             */
     private void recursionQueryChildren(Integer parentId, List<Organization> children) {
         QueryWrapper<Organization> queryWrapper = new QueryWrapper();
+        queryWrapper.select("id");
         queryWrapper.eq("is_delete", Constants.NOT_DELETE);
         queryWrapper.eq("parent_id", parentId);
         List<Organization> organizations = organizationMapper.selectList(queryWrapper);

--
Gitblit v1.8.0