backend_dormitory/src/main/java/com/template/controller/WelcomeCityController.java

package com.template.controller;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.template.api.WelcomeCityControllerAPI;
import com.template.model.pojo.WelcomeCity;
import com.template.model.result.CommonResult;
import com.template.model.vo.ListVo;
import com.template.services.WelcomeCityService;
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;

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author ceshi
 * @since 2025-06-16
 */
@RestController
public class WelcomeCityController implements WelcomeCityControllerAPI {

    @Autowired
    private WelcomeCityService welcomeCityService;

    @Override
    public CommonResult getProvinceLevel() {
        List<WelcomeCity> citys = welcomeCityService.queryAllProvinces();
        List<ListVo> lists = new ArrayList<>();
        if(citys != null && citys.size() > 0){
            for (WelcomeCity city:citys) {
                ListVo data = new ListVo();
                data.setId(city.getId());
                data.setName(city.getProvinceZh());
                lists.add(data);
            }
        }
        return CommonResult.ok(lists);
    }

    @Override
    public CommonResult getCityLevel(int provinceId) {
        //直辖市
        List<String> cityStrs = new ArrayList<>();
        cityStrs.add("北京市");
        cityStrs.add("天津市");
        cityStrs.add("上海市");
        cityStrs.add("重庆市");
        List<WelcomeCity> citys = welcomeCityService.queryAllCitys(provinceId);
        List<ListVo> lists = new ArrayList<>();
        if(citys != null && citys.size() > 0){
            for (WelcomeCity city:citys) {
                ListVo data = new ListVo();
                if(cityStrs.contains(city.getCityZh())){
                    data.setId(city.getParentId());
                    data.setName(city.getCityZh());
                    lists.add(data);
                    break;
                }

                data.setId(city.getId());
                data.setName(city.getCityZh());
                lists.add(data);
            }
        }


        return CommonResult.ok(lists);
    }

    @Override
    public CommonResult getDistrictLevel(int cityId) {
        List<WelcomeCity> citys = welcomeCityService.queryAllCitys(cityId);
        List<ListVo> lists = new ArrayList<>();
        if(citys != null && citys.size() > 0){
            for (WelcomeCity city:citys) {
                ListVo data = new ListVo();
                data.setId(city.getId());
                data.setName(city.getDistrictZh());
                lists.add(data);
            }
        }
        return CommonResult.ok(lists);
    }
}