com.github.fartherp.framework.tree.convert.ZtreeConvertTool Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2017. CK. All rights reserved.
 */

package com.github.fartherp.framework.tree.convert;

import com.github.fartherp.framework.common.util.JsonUtil;
import com.github.fartherp.framework.tree.bo.Treeable;
import com.github.fartherp.framework.tree.model.ztree.SimpleZtree;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created by framework .
 * Auth: hyssop
 * Date: 2016-09-09-17:29
 */
public class ZtreeConvertTool implements ConvertTool {
    public boolean[] booleans = new boolean[2];

    public String findTreeStr(List list, ModelCall mc) {
        return JsonUtil.toJson(findModel(list, mc).getChildren());
    }

    public List findChildren(List list, ModelCall mc) {
        return findModel(list, mc).getChildren();
    }

    private T findModel(List list, ConvertTool.ModelCall mc) {
        if (null == list) {
            throw new RuntimeException("没有属性结构");
        }
        Map p = new HashMap(list.size() + 1);
        // 最外层,默认为0
        T root = (T) new SimpleZtree();
        root.setId((ID) list.get(0).getParentId());
        p.put((ID) list.get(0).getParentId(), (T) root);
        findModel(list, p, mc);
        root.setId(null);
        return root;
    }

    /**
     * 递归遍历菜单
     *
     * @param list 菜单列表
     * @param p    最终的菜单map
     * @param mc   具体业务回调
     */
    private void findModel(List list, Map p, ConvertTool.ModelCall mc) {
        if (null == list || list.isEmpty()) {
            return;
        }
        List fail = new ArrayList();
        for (S t : list) {
            // 当前菜单没添加到map中
            setTreeModel(t, p, mc, fail);
        }
        if (!fail.isEmpty()) {
            // 递归
            findModel(fail, p, mc);
        }
    }

    /**
     * map中不包括当前菜单且父菜单存在,添加当前菜单到map中
     *
     * @param t    业务自己数据
     * @param p    最终map
     * @param mc   自己实现的业务回调接口
     * @param fail 没有找到父菜单的菜单列表
     */
    private void setTreeModel(S t, Map p, ConvertTool.ModelCall mc, List fail) {
        T model = mc.convert(t);
        if (model == null) {
            throw new RuntimeException("回调方法生成SimpleZtree错误");
        }
        // map中不包括当前菜单且包括父菜单
        if (!p.containsKey(model.getId()) && null != p.get((ID) model.getpId())) {
            // 添加到map中
            p.put((ID) model.getId(), model);
            T obj = p.get(model.getpId());
            if (null != obj) {
                obj.addChildren(model);
            }
        } else {
            fail.add(t);
        }
    }
}