package com.ycl.service.region.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ycl.entity.region.SccgRegion;
import com.ycl.mapper.region.SccgRegionMapper;
import com.ycl.service.redis.RedisService;
import com.ycl.service.region.ISccgRegionService;
import com.ycl.utils.common.LiveTimeMillisecond;
import com.ycl.utils.redis.RedisKey;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 遂昌行政区域 服务实现类
 * </p>
 *
 * @author lyq
 * @since 2022-09-16
 */
@Service
public class SccgRegionServiceImpl extends ServiceImpl<SccgRegionMapper, SccgRegion> implements ISccgRegionService {

    @Resource
    private SccgRegionMapper sccgRegionMapper;
    @Resource
    private RedisService redisService;

    @Override
    public List<SccgRegion> getTree() {
        String result = (String) redisService.get(RedisKey.SCCG_REGION);
        if (StringUtils.isBlank(result)) {
            // 1.查出所有网格
            List<SccgRegion> list = this.list();

            // 2.组装成父子的树型结构
            // 2.1、找到所有的一级网格:使用jdk8的stream流进行过滤
            List<SccgRegion> collect = list.stream().filter(griddingEntity -> {
                //分类父id为0就表示该网格为 一级网格
                return griddingEntity.getParentId() == 0;
            }).map(item -> {
                // 找到子网格 set进children属性中
                item.setChildren(getChildrens(item, list));
                return item;
            }).collect(Collectors.toList());
            List<SccgRegion> treeData = getTreeData(collect);
            redisService.set(RedisKey.SCCG_REGION, JSON.toJSONString(treeData), LiveTimeMillisecond.s5184000.time);
            return treeData;
        }
        return JSON.parseArray(result, SccgRegion.class);
    }


    public List<SccgRegion> getChildrens(SccgRegion root, List<SccgRegion> all) {
        List<SccgRegion> collect = all.stream().filter(e -> {
            return e.getParentId().equals(root.getId());
        }).map(item -> {
            item.setChildren(getChildrens(item, all));
            return item;
        }).collect(Collectors.toList());
        return collect;
    }

    /**
     * 利用递归将最后一级空集合变为null，前端联级选择器最后才不会出现 暂无数据的bug
     *
     * @return
     */
    public List<SccgRegion> getTreeData(List<SccgRegion> e) {
        for (int i = 0; i < e.size(); i++) {
            //如果Children的size小于1就说明为空，设置为null
            if (e.get(i).getChildren().size() < 1) {
                e.get(i).setChildren(null);
            } else {
                //子网格有数据就递归
                getTreeData(e.get(i).getChildren());
            }
        }
        return e;
    }

    @Override
    public IPage<SccgRegion> list(SccgRegion sccgRegion) {
        return null;
    }

    @Override
    public List<SccgRegion> getChildren(Long parentId) {
        LambdaQueryWrapper<SccgRegion> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(SccgRegion::getParentId, parentId);
        List<SccgRegion> regions = baseMapper.selectList(wrapper);
        return regions;
    }
}