package com.ruoyi.admin.service.impl;
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.admin.entity.Region;
|
import com.ruoyi.admin.mapper.RegionMapper;
|
import com.ruoyi.admin.service.RegionService;
|
import com.ruoyi.common.core.constant.RedisConstants;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.TimeUnit;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 省市区三级联动 服务实现类
|
* </p>
|
*
|
* @author hjl
|
* @since 2024-06-12
|
*/
|
@Service
|
public class RegionServiceImpl extends ServiceImpl<RegionMapper, Region> implements RegionService {
|
|
@Resource
|
private RedisTemplate<Object, Object> redisTemplate;
|
|
@Override
|
public List<Region> addressTree(List<String> cityCodeList) {
|
// 获取所有地区信息 省市区三级
|
List<Region> regions = getReginList(cityCodeList);
|
// 所有地区
|
Map<String, Region> courseTypeMap = regions.stream().
|
collect(Collectors.toMap(region -> region.getId().toString()
|
, region -> region));
|
redisTemplate.opsForHash().putAll(RedisConstants.ADDRESS_TREE, courseTypeMap);
|
redisTemplate.expire(RedisConstants.ADDRESS_TREE, 30, TimeUnit.MINUTES);
|
// 生成map集合
|
Map<Integer, Region> map = regions.stream().collect(Collectors.toMap(Region::getId, region -> region));
|
// 存放无限级树
|
List<Region> treeData = new ArrayList<>();
|
// 遍历地区集合
|
regions.forEach(e -> {
|
if (e.getParentId() == null || e.getParentId().equals(0)) {
|
treeData.add(e);
|
} else {
|
Region region = map.get(e.getParentId());
|
region.getChildren().add(e);
|
}
|
});
|
for (Region region : treeData) {
|
List<Region> children = region.getChildren();
|
for (Region child : children) {
|
child.setChildren(new ArrayList<>());
|
}
|
}
|
return treeData;
|
}
|
|
/**
|
* 获取redis数据进行封装
|
*/
|
private List<Region> getReginList(List<String> cityCodeList) {
|
List<Region> regions;
|
if (cityCodeList.isEmpty()) {
|
regions = this.list();
|
} else {
|
regions = this.lambdaQuery()
|
.in(Region::getCode, cityCodeList).list();
|
}
|
return regions;
|
}
|
|
}
|
