AreaServiceImpl.java

package com.govmade.modules.system.service.impl;

import java.util.List;

import org.springframework.stereotype.Service;

import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.service.impl.ServiceImpl;
import com.google.common.collect.Lists;
import com.govmade.modules.system.dao.AreaDao;
import com.govmade.modules.system.entity.AreaEntity;
import com.govmade.modules.system.service.AreaService;

/**
 * 系统管理 - 行政区划设置
 * 
 * @author 刘弈臻
 * @date 2018年8月9日
 */
@Service("AreaService")
public class AreaServiceImpl extends ServiceImpl<AreaDao,AreaEntity> implements AreaService{
     
	@Override
    public List<AreaEntity> queryAll() {
		
		List<AreaEntity> areaList = this.selectList(new EntityWrapper<AreaEntity>().eq("state", 1));
		
        List<AreaEntity> aTreeList = buildAreaTree(areaList, 0L);

        return aTreeList;
    }
	
	
/**
 *递归方法
 */
	 private List<AreaEntity> buildAreaTree(List<AreaEntity> list, long id) {
	        List<AreaEntity> aList = Lists.newArrayList();

	        for (AreaEntity a : list) {
	            if (id == a.getPid()) {
	                a.setChildren(buildAreaTree(list, a.getId()));
	                aList.add(a);
	            }
	        }
	        return aList;
	    }
		
	
	@Override
	public void save(AreaEntity area) {
		super.insertOrUpdate(area);
	}
	
	@Override
	public void deleteBatch(Long[] areaIds) {
		// TODO Auto-generated method stub
	}
	
	@Override
	public Integer checkArea(String name) {
		return super.selectCount(new EntityWrapper<AreaEntity>().eq("name",name));
	}
	
}