#

luxiaotao1123

2022-03-31 ab9e3b55225455a503d40aec1a603f355e208ea7

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package com.zy.common.utils;
 
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.zy.asrs.entity.Tag;
import com.zy.asrs.service.TagService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.stereotype.Component;
 
import java.util.*;
 
/**
 * 树形图工具
 * Created by vincent on 2020/10/16
 */
@Component
public class TreeUtils {
 
    @Autowired
    private TagService tagService;
 
    /******************************** 归类树 *********************************/
 
    /**
     * 获取树图数据结构
     */
    @Cacheable(cacheNames="tagTree",key="#id")
    public ArrayList<Map> getTree(String id){
        ArrayList<Map> result = new ArrayList<>();
        Tag tag = tagService.selectById(id);
        // 主节点
        Map<String, Object> map = new HashMap<>();
        map.put("title", tag.getName());
        map.put("id", tag.getId());
        map.put("spread", true);
        List<Map> childrens = new ArrayList<>();
        map.put("children", childrens);
        dealTag(tag, childrens);
        result.add(map);
        // 开始处理字节点
//        deal(tag, childrens);
        return result;
    }
 
    /**
     * 递归获取子节点数据
     */
    public void dealTag(Tag parent, List<Map> list) {
        List<Tag> tags = tagService.selectList(
                new EntityWrapper<Tag>()
                        .eq("parent_id", parent.getId())
                        .eq("status", "1"));
        for (Tag tag : tags) {
            Map<String, Object> map = new HashMap<>();
            map.put("title", tag.getName());
            map.put("id", tag.getId());
            map.put("spread", true);
            List<Map> childrens = new ArrayList<>();
            map.put("children", childrens);
            dealTag(tag, childrens);
            list.add(map);
        }
    }
 
 
    // -------------------------------------------------------------------------------------------------------
 
    /**
     * 条件筛选
     */
    @SuppressWarnings("unchecked")
    public void remove(String condition, List<Map> list) {
        Iterator<Map> iterator = list.iterator();
        while (iterator.hasNext()) {
            Map map = iterator.next();
            if (map.get("children") != null) {
                List<Map> children = (List<Map>) map.get("children");
                if (children.size() > 0) {
                    remove(condition, children);
                } else {
                    if (!String.valueOf(map.get("title")).contains(condition)) {
                        iterator.remove();
                    }
                }
            }
        }
    }
 
}