package com.highdatas.mdm.service.impl; import com.baomidou.mybatisplus.service.impl.ServiceImpl; import com.highdatas.mdm.entity.SysMenu; import com.highdatas.mdm.mapper.SysMenuMapper; import com.highdatas.mdm.service.ISysMenuService; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.LinkedHashSet; /** *

* 服务实现类 *

* * @author kimi * @since 2019-12-16 */ @Service public class SysMenuServiceImpl extends ServiceImpl implements ISysMenuService { @Autowired ISysMenuService menuService; @Override public LinkedHashSet getByParentId(LinkedHashSet parentIdSet) { if (parentIdSet.size() == 0) { return null; } LinkedHashSet subSet = new LinkedHashSet<>(); for (String s : parentIdSet) { SysMenu sysMenu = menuService.selectById(s); if (sysMenu == null) { continue; } String parentId = sysMenu.getParentId(); if (StringUtils.isEmpty(parentId)) { continue; } subSet.add(parentId); } LinkedHashSet byParentId = getByParentId(subSet); if (byParentId != null) { parentIdSet.addAll(byParentId); } return parentIdSet; } }