jinpengyong
2023-10-30 49d7ce5a988596d7540083be0e1948bcb66af544
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,6 +30,9 @@
    @Autowired
    OrganizationMapper organizationMapper;
    @Autowired
    private DeviceService deviceService;
    @Override
    public List<Organization> getChildrenOrganizationsById(Integer id) {
@@ -40,6 +46,32 @@
        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中