后端封装分类树形实现

0.加载分类的树形节点

0.1 创建接口

/**
 * @author MaoLin Wang
 * @date 2019/9/2910:20
 */
public interface CategoryTreeData<T> {
    String getId();
    String getParentId();

    void setChildren(List<T> children);
    List<T> getChildren();
}

0.2 实体类继承接口

/**
 * category实体类
 * @author Administrator
 *
 */
@Entity
@Table(name="tb_category")
@Data
public class Category implements Serializable, CategoryTreeData<Category> {

	@Id
	private String id;//分类id
	private String cname;//分类名称
	private String parentId;//父级id
	private String isparent;//是否为父级
	private String state;//状态 0弃用 1使用

    @Transient
    private List<Category> children;
}

0.3 封装数据工具类

/**
 * @author MaoLin Wang
 * @date 2019/9/2910:21
 */
public class CategoryTreeUtil {


    public static <T extends CategoryTreeData<T>> List<T> getTreeList(String parentid, List<T> categoryList){
        //存储顶层节点
        List<T> resultList=new ArrayList<>();

        Map<Object,T> categoryMap=new HashMap<>();
        T category;

        for(int i=0;i<categoryList.size();i++){
            category=categoryList.get(i);
            categoryMap.put(category.getId(),category);
            if(parentid.equals(category.getParentId())){
                resultList.add(category);
            }
        }


        for(int i=0;i<categoryList.size();i++){
            category=categoryList.get(i);
            T parentCategory=categoryMap.get(category.getParentId());
            if(parentCategory!=null){
                //有父节点
                if(parentCategory.getChildren()==null){
                    parentCategory.setChildren(new ArrayList<>());
                }
                parentCategory.getChildren().add(category);
                categoryMap.put(category.getParentId(),parentCategory);
            }
        }
        return resultList;
    }
}

0.4 Controller

 /**
     * 获取封装树形数据
     * @return
     */
    @PostMapping("/tree")
    public Result fetchCategoryTree(){
        if( redisTemplate.opsForList().range("weblog_category_tree",0,-1).size()>0){
            return new Result(true,StatusCode.OK,ResultMessage.QUERY_SUCCESS,(List<Category>)redisTemplate.opsForList().index("weblog_category_tree",0));
        }
        System.out.println(redisTemplate.opsForList().range("weblog_category_tree",0,-1));
        //缓存不存在
        List<Category>categoryTree=categoryService.fetchCategoryTree();
        redisTemplate.opsForList().leftPush("weblog_category_tree",categoryTree);
        return new Result(true,StatusCode.OK,ResultMessage.QUERY_SUCCESS,categoryTree);
    }

0.5 Service

/**
     * 查询分类数据（树形）
     * @return
     */
    public List<Category> fetchCategoryTree() {
        List<Category> data=categoryDao.findByState("1");
        return CategoryTreeUtil.getTreeList("0",data);
    }

1.修改分类

1.1修改工具类

    private List<Category>categories=new ArrayList<>();//存储修改时待修改集合

 /**
     * 修改操作
     * @param category
     */
    public void handleUpdate(Category category){
        Category proCategory = categoryDao.findById(category.getId()).get();//修改前的分类
        if (!proCategory.getCname().equals(category.getCname()) || !proCategory.getState().equals(category.getState())) {
            //修改了
            if(category.getState().equals(proCategory.getState())){
                //只修改了名字,未修改状态
                categoryDao.save(category);//直接更新
            }else {
                //修改了状态
                //约定前端在修改操作中只能添加使用，不能进行删除操作，所以状态只能被改为1
                categoryDao.save(category);
                List<Category> categories = diguiUp(category.getParentId(),false);
                for (Category c :categories){
                    categoryDao.updateStateById("1",c.getId());
                }
            }

        }
        //未作修改


    }

    /**
     * 向上递归查询待修改分类
     * @param parentid
     * @param isDigui
     * @return 返回待修改分类集合
     */
    private List<Category> diguiUp(String parentid,boolean isDigui){
        Category category = categoryDao.findById(parentid).get();//查询上级类目
       List<String>diguiCategory=new ArrayList<>();
        if("0".equals(category.getParentId())){
            //父类目为顶级类目
            categories.add(category);

            return categories;
        }else if("1".equals(category.getState())){
            //上级状态已经为1
            return categories;
        }else {
            categories.add(category);

            diguiCategory.add(category.getParentId());
            if(diguiCategory!=null&&diguiCategory.size()>0){
                diguiUp(category.getParentId(),true);
            }
            return categories;
        }


    }