fuint-application/application-web/src/main/java/com/fuint/application/web/rest/RegionController.java

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

package com.fuint.application.web.rest;

import com.fuint.application.dao.entities.MtRegion;
import com.fuint.application.dao.repositories.MtRegionRepository;
import com.fuint.application.dto.RegionDto;
import com.fuint.exception.BusinessCheckException;
import com.fuint.application.ResponseObject;
import com.fuint.application.BaseController;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 省/市/区controller
 * Created by FSQ
 * Contact wx fsq_better
 * Site https://www.fuint.cn
 */
@RestController
@RequestMapping(value = "/rest/region")
public class RegionController extends BaseController {

    private static final Logger log = LoggerFactory.getLogger(RegionController.class);

    @Autowired
    private MtRegionRepository regionRepository;

    /**
     * 获取树状结构列表
     */
    @RequestMapping(value = "/tree")
    @CrossOrigin
    public ResponseObject tree(HttpServletRequest request) throws BusinessCheckException {
        Map<String, Object> params = new HashMap<>();
        Specification<MtRegion> specification = regionRepository.buildSpecification(params);
        Sort sort = new Sort(Sort.Direction.ASC, "id");
        List<MtRegion> regionList = regionRepository.findAll(specification, sort);

        List<RegionDto> treeData = new ArrayList<>();

        for (MtRegion mtRegion : regionList) {
             if (mtRegion.getLevel().equals("1")) {
                 RegionDto dto = new RegionDto();
                 dto.setId(mtRegion.getId());
                 dto.setName(mtRegion.getName());
                 dto.setCode(mtRegion.getCode());
                 dto.setPid(mtRegion.getPid());
                 dto.setLevel(mtRegion.getLevel());
                 dto.setCity(new ArrayList<>());
                 treeData.add(dto);
             }
        }