package com.ruoyi.system.service.impl;
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.system.mapper.TRegionMapper;
|
import com.ruoyi.system.model.TRegion;
|
import com.ruoyi.system.service.TRegionService;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 省市管理 服务实现类
|
* </p>
|
*
|
* @author xiaochen
|
* @since 2024-02-29
|
*/
|
@Service
|
public class TRegionServiceImpl extends ServiceImpl<TRegionMapper, TRegion> implements TRegionService {
|
|
@Override
|
public List<TRegion> getRegion() {
|
List<TRegion> tRegions = this.baseMapper.selectList(Wrappers.lambdaQuery(TRegion.class));
|
List<TRegion> parent = tRegions.stream().filter(e -> e.getParentId().equals(0L)).collect(Collectors.toList());
|
for (TRegion region : parent) {
|
List<TRegion> children = tRegions.stream().filter(e -> e.getParentId().equals(region.getId())).collect(Collectors.toList());
|
region.setChildren(children);
|
}
|
return parent;
|
}
|
}
|
