李林
2023-10-07 658d4927d468c47208fd012d9128b09249c07eff
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
/*
 *    Copyright (c) 2018-2025, ztt All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * Author: ztt
 */
package com.chinaztt.mes.production.service.impl;
 
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.production.dto.ArtificialInformationJoinTreeNode;
import com.chinaztt.mes.production.entity.ArtificialInformationJoin;
import com.chinaztt.mes.production.entity.ArtificialInformationRelation;
import com.chinaztt.mes.production.mapper.ArtificialInformationJoinMapper;
import com.chinaztt.mes.production.mapper.ArtificialInformationRelationMapper;
import com.chinaztt.mes.production.service.ArtificialInformationJoinService;
import com.chinaztt.ztt.admin.api.vo.TreeUtil;
import com.chinaztt.ztt.common.core.util.R;
import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;
 
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
 
/**
 * 人工类型分组
 *
 * @author sunxl
 * @date 2021-01-28 11:19:54
 */
@Service
@AllArgsConstructor
public class ArtificialInformationJoinServiceImpl extends ServiceImpl<ArtificialInformationJoinMapper, ArtificialInformationJoin> implements ArtificialInformationJoinService {
      private ArtificialInformationRelationMapper artificialInformationRelationMapper;
    @Override
    public List<ArtificialInformationJoinTreeNode> selectTree() {
        List<ArtificialInformationJoinTreeNode> collect = baseMapper.selectList(null)
                .stream().sorted(Comparator.comparing(ArtificialInformationJoin::getSort)).map(artificialInformationJoin -> {
                    ArtificialInformationJoinTreeNode node = new ArtificialInformationJoinTreeNode();
                    node.setId(artificialInformationJoin.getId().intValue());
                    node.setName(artificialInformationJoin.getName());
                    node.setParentId(artificialInformationJoin.getParentId().intValue());
                    return node;
                }).collect(Collectors.toList());
        return TreeUtil.build(collect, 0);
    }
 
    @Override
    public R<Boolean> fullSave(List<ArtificialInformationRelation> artificialInformationRelationList) {
        for (ArtificialInformationRelation artificialInformationRelation :artificialInformationRelationList){
            artificialInformationRelationMapper.insert(artificialInformationRelation);
        }
        return R.ok();
    }
 
    @Override
    public R<Boolean> fullDelete(Long id) {
        baseMapper.deleteById(id);
        artificialInformationRelationMapper.delete(Wrappers.<ArtificialInformationRelation>lambdaQuery().eq(ArtificialInformationRelation::getProductionHandymanJoinId,id));
        return R.ok();
    }
 
}