#
vincentlu
2025-05-13 ebd2f4397a92c6a5096de1b86d59154363344720
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
package com.zy.acs.manager.manager.service.impl;
 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zy.acs.framework.exception.CoolException;
import com.zy.acs.manager.manager.entity.Segment;
import com.zy.acs.manager.manager.entity.Task;
import com.zy.acs.manager.manager.enums.ActionStsType;
import com.zy.acs.manager.manager.enums.SegmentStateType;
import com.zy.acs.manager.manager.enums.TaskStsType;
import com.zy.acs.manager.manager.mapper.SegmentMapper;
import com.zy.acs.manager.manager.service.*;
import com.zy.acs.framework.common.Cools;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import java.util.ArrayList;
import java.util.Comparator;
import java.util.Date;
import java.util.List;
 
@Slf4j
@Service("segmentService")
public class SegmentServiceImpl extends ServiceImpl<SegmentMapper, Segment> implements SegmentService {
 
    @Autowired
    private AgvService agvService;
    @Autowired
    private TravelService travelService;
    @Autowired
    private TaskService taskService;
    @Autowired
    private ActionService actionService;
 
    @Override
    public void processNext(List<Segment> segmentList) {
 
        segmentList.stream().max(Comparator.comparingInt(Segment::getSerial)).ifPresent(segment -> {
            Segment nextSegment = this.getNextStepOfInit(segment.getTravelId(), segment.getSerial());
 
            if (null != nextSegment) {
 
                nextSegment.setState(SegmentStateType.WAITING.toString());
                nextSegment.setUpdateTime(new Date());
                if (!this.updateById(nextSegment)) {
                    log.error("Segment [{}] failed to update !!!", nextSegment.getGroupId() + " - " + nextSegment.getSerial());
                }
 
            } else {
 
                travelService.checkFinish(segment.getTravelId());
            }
        });
    }
 
    @Override
    public Segment getNextStepOfInit(Long travelId, Integer serial) {
        return this.getOne(new LambdaQueryWrapper<Segment>()
                .eq(Segment::getTravelId, travelId)
                .eq(Segment::getSerial, serial + 1)
                .eq(Segment::getState, SegmentStateType.INIT.toString())
        );
    }
 
    @Override
    public Segment getPreviousStepOfFinish(Long travelId, Integer serial, String memo) {
        return this.getOne(new LambdaQueryWrapper<Segment>()
                .eq(Segment::getTravelId, travelId)
                .eq(Segment::getSerial, serial - 1)
//                .eq(Segment::getState, SegmentStateType.FINISH.toString())
                .eq(Segment::getMemo, memo)
        );
    }
 
    @Override
    public List<Segment> getJustWaitingSeg(Long agvId) {
        if (null == agvId) {
            return null;
        }
        List<Segment> waitingSegList = this.getByAgvAndState(agvId, SegmentStateType.WAITING.toString());
        if (Cools.isEmpty(waitingSegList)) {
            return null;
        } else {
            if (waitingSegList.size() > 1) {
                log.error("{}号车辆存在多个等待中的Segment!!!", agvService.getById(agvId).getUuid());
            }
            return waitingSegList;
        }
    }
 
    @Override
    public List<Segment> getSamePos(List<Segment> segmentList) {
        return null;
    }
 
    @Override
    public List<Segment> getByAgvAndState(Long agvId, String state) {
        return this.list(new LambdaQueryWrapper<Segment>()
                .eq(Segment::getAgvId, agvId)
                .eq(Segment::getState, state)
                .orderByAsc(Segment::getSerial)
        );
    }
 
    @Override
    public List<String> getGroupNo(SegmentStateType state, Long agvId, String groupNo) {
        return this.baseMapper.getGroupNo(state.toString(), agvId, groupNo);
    }
 
    @Override
    public List<Segment> getRunningByEndCode(Long codeId) {
        if (null == codeId) {
            return new ArrayList<>();
        }
        return this.list(new LambdaQueryWrapper<Segment>()
                .eq(Segment::getEndNode, codeId)
                .in(Segment::getState, SegmentStateType.WAITING.toString(), SegmentStateType.RUNNING.toString())
        );
    }
 
    @Override
    @Transactional
    public Boolean cancel(Long segmentId, Long userId) {
        Date now = new Date();
        Segment segment = this.getById(segmentId);
        String groupId = segment.getGroupId();
        // update segment list
        List<Segment> segmentList = this.list(new LambdaQueryWrapper<Segment>().eq(Segment::getGroupId, groupId));
        for (Segment seg : segmentList) {
            Task task = taskService.getById(segment.getTaskId());
            if (task != null &&
                    (!task.getTaskSts().equals(TaskStsType.COMPLETE.val()) && !task.getTaskSts().equals(TaskStsType.CANCEL.val()))) {
                throw new CoolException("the task is not finish");
            }
            seg.setState(SegmentStateType.FINISH.toString());
            seg.setUpdateTime(now);
            if (!this.updateById(seg)) {
                throw new CoolException("failed to cancel segment");
            }
        }
        this.processNext(segmentList);
        // update action list
        actionService.updateStsByGroupId(groupId, ActionStsType.EXPIRED.val());
        return Boolean.TRUE;
    }
 
}