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.ArrayList;
|
import java.util.Collections;
|
import java.util.LinkedHashSet;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 服务实现类
|
* </p>
|
*
|
* @author kimi
|
* @since 2019-12-16
|
*/
|
@Service
|
public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> implements ISysMenuService {
|
@Autowired
|
ISysMenuService menuService;
|
|
@Override
|
public LinkedHashSet<String> getByParentId(LinkedHashSet<String> parentIdSet) {
|
if (parentIdSet.size() == 0) {
|
return null;
|
}
|
LinkedHashSet<String> 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<String> byParentId = getByParentId(subSet);
|
if (byParentId != null) {
|
parentIdSet.addAll(byParentId);
|
}
|
|
return parentIdSet;
|
}
|
|
@Override
|
public List<SysMenu> getMenuByParentId(LinkedHashSet<String> parentIdSet) {
|
LinkedHashSet<String> byParentId = getByParentId(parentIdSet);
|
List<SysMenu> sysMenus = selectBatchIds(byParentId);
|
List<SysMenu> result = new ArrayList<SysMenu>();
|
for (String s : byParentId) {
|
List<SysMenu> collect = sysMenus.stream().filter(sysMenu -> sysMenu.getId().equalsIgnoreCase(s)).collect(Collectors.toList());
|
result.add(collect.get(0));
|
}
|
Collections.reverse(result);
|
return result;
|
}
|
|
|
}
|
