Your Name
2023-06-30 c32e21f2eab25be19b337d546e67e1441733dd4a
service-community/src/main/java/com/java110/community/smo/impl/DataPrivilegeUnitV1InnerServiceSMOImpl.java
@@ -17,14 +17,16 @@
import com.java110.community.dao.IDataPrivilegeUnitV1ServiceDao;
import com.java110.dto.UnitDto;
import com.java110.dto.unit.UnitDto;
import com.java110.dto.data.DataPrivilegeStaffDto;
import com.java110.intf.user.IDataPrivilegeStaffV1InnerServiceSMO;
import com.java110.intf.community.IDataPrivilegeUnitV1InnerServiceSMO;
import com.java110.dto.dataPrivilegeUnit.DataPrivilegeUnitDto;
import com.java110.po.dataPrivilegeUnit.DataPrivilegeUnitPo;
import com.java110.dto.data.DataPrivilegeUnitDto;
import com.java110.po.privilege.DataPrivilegeUnitPo;
import com.java110.utils.util.BeanConvertUtil;
import com.java110.core.base.smo.BaseServiceSMO;
import com.java110.dto.user.UserDto;
import com.java110.dto.PageDto;
import com.java110.utils.util.StringUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
@@ -45,6 +47,9 @@
    @Autowired
    private IDataPrivilegeUnitV1ServiceDao dataPrivilegeUnitV1ServiceDaoImpl;
    @Autowired
    private IDataPrivilegeStaffV1InnerServiceSMO dataPrivilegeStaffV1InnerServiceSMOImpl;
    @Override
@@ -109,4 +114,38 @@
        return unitDtos;
    }
    @Override
    public String[] queryDataPrivilegeUnitsByStaff(@RequestBody DataPrivilegeStaffDto dataPrivilegeStaffDto) {
        if(StringUtil.isEmpty(dataPrivilegeStaffDto.getStaffId())){
            return new String[0];
        }
        List<DataPrivilegeStaffDto> dataPrivilegeStaffDtos = dataPrivilegeStaffV1InnerServiceSMOImpl.queryDataPrivilegeStaffs(dataPrivilegeStaffDto);
        if(dataPrivilegeStaffDtos == null || dataPrivilegeStaffDtos.size()<1){
            return new String[0];
        }
        List<String> dpIds = new ArrayList<>();
        for(DataPrivilegeStaffDto dataPrivilegeStaffDto1 : dataPrivilegeStaffDtos){
            dpIds.add(dataPrivilegeStaffDto1.getDpId());
        }
        DataPrivilegeUnitDto dataPrivilegeUnitDto = new DataPrivilegeUnitDto();
        dataPrivilegeUnitDto.setDpIds(dpIds.toArray(new String[dpIds.size()]));
        List<DataPrivilegeUnitDto> dataPrivilegeUnitDtos = queryDataPrivilegeUnits(dataPrivilegeUnitDto);
        if(dataPrivilegeUnitDtos == null || dataPrivilegeUnitDtos.size()<1){
            return new String[0];
        }
        List<String> unitIds = new ArrayList<>();
        for(DataPrivilegeUnitDto dataPrivilegeUnitDto1 : dataPrivilegeUnitDtos){
            unitIds.add(dataPrivilegeUnitDto1.getUnitId());
        }
        return unitIds.toArray(new String[unitIds.size()]);
    }
}