forked from ChunelFeng/CGraph
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathGElementRepository.cpp
More file actions
138 lines (109 loc) · 3.75 KB
/
GElementRepository.cpp
File metadata and controls
138 lines (109 loc) · 3.75 KB
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
/***************************
@Author: Chunel
@Contact: chunel@foxmail.com
@File: GElementRepository.cpp
@Time: 2023/6/24 17:51
@Desc:
***************************/
#include "GElementRepository.h"
#include "GNode/GNodeInclude.h"
CGRAPH_NAMESPACE_BEGIN
CVoid GElementRepository::insert(GElementPtr ptr) {
CGRAPH_ASSERT_NOT_NULL_THROW_ERROR(ptr)
elements_.insert(ptr);
}
CBool GElementRepository::find(GElementPtr ptr) const {
CGRAPH_ASSERT_NOT_NULL_THROW_ERROR(ptr)
return elements_.find(ptr) != elements_.end();
}
GElementRepositoryPtr GElementRepository::setThreadPool(UThreadPoolPtr ptr) {
CGRAPH_ASSERT_NOT_NULL_THROW_ERROR(ptr)
for (auto& cur : this->elements_) {
cur->setThreadPool(ptr);
}
return this;
}
CStatus GElementRepository::setup() {
CGRAPH_FUNCTION_BEGIN
// 一旦执行，全部设置为 normal状态
status = pushAllState(GElementState::NORMAL);
CGRAPH_FUNCTION_END
}
CStatus GElementRepository::reset() {
CGRAPH_FUNCTION_BEGIN
for (auto& cur : async_elements_) {
if (GElementTimeoutStrategy::HOLD_BY_PIPELINE == cur->timeout_strategy_) {
// 强烈建议，在这里等待执行完成
status += cur->getAsyncResult();
}
// 无论如何，写回非 timeout状态，确保不会迭代到下一轮
if (cur->isTimeout()) {
cur->cur_state_.store(GElementState::NORMAL, std::memory_order_release);
}
}
CGRAPH_FUNCTION_END
}
CStatus GElementRepository::pushAllState(const GElementState& state) {
CGRAPH_FUNCTION_BEGIN
if (cur_state_ == state) {
return status; // 避免重复赋值
}
for (auto& cur : elements_) {
cur->cur_state_.store(state, std::memory_order_release);
if (GElementState::YIELD != state) {
// 目前仅非yield状态，需要切换的。如果一直处于 yield状态，是不需要被通知的
cur->yield_cv_.notify_one();
}
}
cur_state_ = state; // 记录当前的状态信息
CGRAPH_FUNCTION_END
}
CBool GElementRepository::isCancelState() const {
/**
* 因为每次执行的时候，都需要判断一下这个状态是否为 cancel
* 且理论上不会出现多线程问题
* 故这一层的 cur_state_ 就不设置为atomic类型的了
*/
return GElementState::CANCEL == cur_state_;
}
CStatus GElementRepository::init() {
CGRAPH_FUNCTION_BEGIN
async_elements_.clear(); // 每次记得清空这里。因为每次init之后，都可能不一样
for (auto& element : elements_) {
/**
* 1. 查验element是否为空
* 2. 查验配置信息是否正确
* 3. 记录异步节点的信息
*/
CGRAPH_ASSERT_NOT_NULL(element)
status = element->checkSuitable();
CGRAPH_FUNCTION_CHECK_STATUS
if (element->isAsync()) {
async_elements_.emplace(element);
}
}
// 每次初始化之后，所有的状态，都可以恢复了
status = pushAllState(GElementState::NORMAL);
CGRAPH_FUNCTION_END
}
CStatus GElementRepository::destroy() {
CGRAPH_FUNCTION_BEGIN
/**
* destroy的时候，恢复create的状态，确保再次轮训的时候正常
* 理论上，cancel状态仅会出现在极短的时间段之内，
* 当程序 cancel完成之后，就会重新恢复 CREATE的状态
* 问题详见: https://github.com/ChunelFeng/CGraph/issues/153
*/
status = pushAllState(GElementState::NORMAL);
CGRAPH_FUNCTION_END
}
CStatus GElementRepository::run() {
CGRAPH_NO_SUPPORT
}
GElementRepository::~GElementRepository() {
// 删除所有内部的element信息
for (GElementPtr element : elements_) {
CGRAPH_DELETE_PTR(element)
}
}
CGRAPH_NAMESPACE_END