设备模块修改bug

zss

2025-02-17 087991c76f078defe5eb55d84223021b4199fb3d

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
90
91
package com.yuanchu.mom.service.impl;
 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yuanchu.mom.mapper.ProcurementSuppliesContentsMapper;
import com.yuanchu.mom.mapper.UserMapper;
import com.yuanchu.mom.pojo.ProcurementSuppliesContents;
import com.yuanchu.mom.pojo.User;
import com.yuanchu.mom.service.ProcurementSuppliesContentsService;
import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;
 
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
 
@Service
@AllArgsConstructor
public class ProcurementSuppliesContentsServiceImpl extends ServiceImpl<ProcurementSuppliesContentsMapper, ProcurementSuppliesContents>
        implements ProcurementSuppliesContentsService {
 
    private UserMapper userMapper;
 
    @Override
    public List<ProcurementSuppliesContents> directoryListing() {
        ArrayList<ProcurementSuppliesContents> list = new ArrayList<>();
        // 查出一级目录
        List<ProcurementSuppliesContents> firstLevel = baseMapper.selectList(new LambdaQueryWrapper<ProcurementSuppliesContents>()
                .isNull(ProcurementSuppliesContents::getParentId));
 
        // 查出含有父节点的 并通过父节点分组
        List<ProcurementSuppliesContents> seconds = baseMapper.selectList(new LambdaQueryWrapper<ProcurementSuppliesContents>()
                .isNotNull(ProcurementSuppliesContents::getParentId));
        if(firstLevel.size() > 0){
            Map<Integer, List<ProcurementSuppliesContents>> collect = seconds.stream()
                    .collect(Collectors.groupingBy(ProcurementSuppliesContents::getParentId));
            // 根据分组的key 赋值对应的children
            for (int i = 0; i < firstLevel.size(); i++) {
                recursion(firstLevel.get(i),collect);
                list.add(firstLevel.get(i));
            }
        }
        return list;
    }
 
    // 递归赋值children
    public ProcurementSuppliesContents recursion(ProcurementSuppliesContents firstLevel, Map<Integer, List<ProcurementSuppliesContents>> collect) {
        // 将父节点的children赋值
        if(collect.containsKey(firstLevel.getId())) {
            List<ProcurementSuppliesContents> procurementSupplies = collect.get(firstLevel.getId());
            firstLevel.setChildren(procurementSupplies);
            for (int i = 0; i < procurementSupplies.size(); i++) {
                recursion(procurementSupplies.get(i),collect);
            }
        }
        return firstLevel;
    }
 
    @Override
    public List<Map<String, Object>> getNodeNames() {
        List<ProcurementSuppliesContents> procurementSuppliesContents = baseMapper.selectList(null);
        List<Map<String, Object>> collect = new ArrayList<>();
        if(procurementSuppliesContents.size() > 0) {
            collect = procurementSuppliesContents.stream().map(item -> {
                HashMap<String, Object> map = new HashMap<>();
                map.put("id", item.getId());
                map.put("nodeName", item.getNodeName());
                return map;
            }).collect(Collectors.toList());
        }
        return collect;
    }
 
    @Override
    public List<Map<String, Object>> getUserList() {
        // 0 停用
        List<User> users = userMapper.selectList(new LambdaQueryWrapper<User>().eq(User::getState, 1));
        List<Map<String, Object>> collect = new ArrayList<>();
        if(users.size() > 0) {
            collect = users.stream().map(item -> {
                HashMap<String, Object> map = new HashMap<>();
                map.put("id", item.getId());
                map.put("nodeName", item.getName());
                return map;
            }).collect(Collectors.toList());
        }
        return collect;
    }
}