package com.ruoyi.system.web;

import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.system.domain.company.AreaInfo;
import com.ruoyi.system.domain.dto.AreaInfoDto;
import com.ruoyi.system.service.company.IAreaInfoService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@RequestMapping("/areaInfo")
@RestController
public class AreaInfoController extends BaseController {
    @Autowired
    IAreaInfoService areaInfoService;

    /***
     * 功能描述: <查询所有的区域信息>
     * 〈〉
     *
     * @Param: []
     * @Return: com.ruoyi.common.core.domain.AjaxResult
     * @Author: chenqf
     * @Date: 2022/6/2 17:48
     */
    @RequestMapping("/getAreaInfo")
    public AjaxResult getAreaInfoMap() {
        List<AreaInfo> areaInfoList = areaInfoService.list();
        Map<String, List<AreaInfo>> areaInfoMap = areaInfoList.stream().collect(Collectors.groupingBy(AreaInfo::getCity));
        List<AreaInfoDto> result = new ArrayList<>();
        areaInfoMap.keySet().forEach(city -> {
            AreaInfoDto areaInfoDto = new AreaInfoDto();
            areaInfoDto.setLabel(city);
            List<AreaInfo> areaInfos = areaInfoMap.get(city);
            List<AreaInfoDto> children = new ArrayList<>();
            areaInfos.forEach(areaInfo -> {
                AreaInfoDto childAreaInfoDto = new AreaInfoDto();
                childAreaInfoDto.setLabel(areaInfo.getArea());
                childAreaInfoDto.setValue(areaInfo.getId());
                children.add(childAreaInfoDto);
            });
            areaInfoDto.setChildren(children);
            result.add(areaInfoDto);
        });
        return AjaxResult.success(result);
    }
}