package com.dsh.other.controller;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.dsh.other.entity.Region;
import com.dsh.other.model.CityDataAndProvinceDataVo;
import com.dsh.other.service.IRegionService;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

@RestController
@RequestMapping("")
public class TRegionController {


    @Resource
    private IRegionService regionService;


    @PostMapping("/base/region/getCityAndProvince")
    public List<CityDataAndProvinceDataVo> getCityAndProvince(@RequestBody List<Integer> ids) {
        List<CityDataAndProvinceDataVo> cityDataAndProvinceDataVos = new ArrayList<>();
        try {
            List<Region> list = regionService.list(new LambdaQueryWrapper<Region>()
                    .in(Region::getId, ids));
            if (list.size() > 0) {
                for (Region region : list) {
                    CityDataAndProvinceDataVo dataVo = new CityDataAndProvinceDataVo();
                    dataVo.setCityCode(region.getCode());
                    dataVo.setCityName(region.getName());
                    Region byId = regionService.getById(region.getParentId());
                    dataVo.setProvinceCode(byId.getCode());
                    dataVo.setProvinceName(byId.getName());
                    cityDataAndProvinceDataVos.add(dataVo);
                }
            }
            return cityDataAndProvinceDataVos;
        } catch (Exception e) {
            return cityDataAndProvinceDataVos;
        }
    }

}