package org.jeecg.modules.base.service.impl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import org.jeecg.common.system.vo.SelectTreeModel; import org.jeecg.modules.base.entity.WarehouseClient; import org.jeecg.modules.base.mapper.WarehouseClientMapper; import org.jeecg.modules.base.service.IWarehouseClientService; import org.jeecg.modules.system.entity.SysDepart; import org.jeecg.modules.system.mapper.SysDepartMapper; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import java.util.ArrayList; import java.util.List; /** * @Description: 仓库服务对象 * @Author: jeecg-boot * @Date: 2022-11-07 * @Version: V1.0 */ @Service @Transactional public class WarehouseClientServiceImpl extends ServiceImpl implements IWarehouseClientService { @Autowired private WarehouseClientMapper warehouseClientMapper; @Autowired private SysDepartMapper sysDepartMapper; @Override public List selectByMainId(String mainId) { return warehouseClientMapper.selectByMainId(mainId); } @Override public List getDepartTreeByEnterpriseId(String enterpriseId) { return loadDepartTreeByParentId(enterpriseId); } @Override public List getDepartListByParentId(List parentIds) { List list = sysDepartMapper.getDepartListByParentId(parentIds); List list1 = new ArrayList<>(); for(SysDepart sysDepart:list){ list1.add(sysDepart.getId()); } List list2 = new ArrayList<>(); if(list1.size()!=0){ list2 = getDepartListByParentId(list1); list.addAll(list2); } return list; } public List loadDepartTreeByParentId(String parentId){ List list = sysDepartMapper.getDepartTreeByParentId(parentId); if(list.size()!=0){ for(SelectTreeModel selectTreeModel:list){ selectTreeModel.setChildren(loadDepartTreeByParentId(selectTreeModel.getKey())); } } return list; } }