forked from ChunelFeng/CGraph
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathGElementManager.cpp
More file actions
189 lines (141 loc) · 4.62 KB
/
GElementManager.cpp
File metadata and controls
189 lines (141 loc) · 4.62 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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
/***************************
@Author: Chunel
@Contact: chunel@foxmail.com
@File: GElementManager.cpp
@Time: 2021/6/2 10:33 下午
@Desc:
***************************/
#include "GElementManager.h"
#include "_GOptimizer/GOptimizerInclude.h"
CGRAPH_NAMESPACE_BEGIN
GElementManager::~GElementManager() {
/**
* manager中的节点，在析构的时候不需要释放。
* 所有的节点信息在GPipeLine类中统一申请和释放
*/
CGRAPH_DELETE_PTR(engine_)
}
CStatus GElementManager::init() {
CGRAPH_FUNCTION_BEGIN
for (auto element : manager_elements_) {
CGRAPH_ASSERT_NOT_NULL(element)
// 确保在任何情况下，任何切面中，都可以获取参数，出发事件信息
element->updateAspectInfo();
}
status = initEngine();
CGRAPH_FUNCTION_CHECK_STATUS
for (GElementPtr element : manager_elements_) {
status = element->fatProcessor(CFunctionType::INIT);
CGRAPH_FUNCTION_CHECK_STATUS
element->is_init_ = true;
}
CGRAPH_FUNCTION_END
}
CStatus GElementManager::destroy() {
CGRAPH_FUNCTION_BEGIN
for (GElementPtr element : manager_elements_) {
status = element->fatProcessor(CFunctionType::DESTROY);
CGRAPH_FUNCTION_CHECK_STATUS
element->is_init_ = false;
}
CGRAPH_DELETE_PTR(engine_)
CGRAPH_FUNCTION_END
}
CStatus GElementManager::run() {
CGRAPH_FUNCTION_BEGIN
status = engine_->run(); // 通过引擎来执行全部的逻辑
CGRAPH_FUNCTION_END
}
CStatus GElementManager::add(GElementPtr element) {
CGRAPH_FUNCTION_BEGIN
CGRAPH_ASSERT_NOT_NULL(element)
this->manager_elements_.emplace(element);
CGRAPH_FUNCTION_END
}
CBool GElementManager::find(GElementPtr element) const {
if (nullptr == element) {
return false;
}
return manager_elements_.end() != manager_elements_.find(element);
}
CStatus GElementManager::remove(GElementPtr element) {
CGRAPH_FUNCTION_BEGIN
CGRAPH_ASSERT_NOT_NULL(element)
manager_elements_.erase(element);
CGRAPH_FUNCTION_END
}
CStatus GElementManager::clear() {
CGRAPH_FUNCTION_BEGIN
for (auto* element : manager_elements_) {
CGRAPH_DELETE_PTR(element)
}
manager_elements_.clear();
CGRAPH_FUNCTION_END
}
GElementManagerPtr GElementManager::setEngineType(GEngineType engineType) {
engine_type_ = engineType;
return this;
}
CStatus GElementManager::initEngine() {
CGRAPH_FUNCTION_BEGIN
CGRAPH_DELETE_PTR(engine_)
switch (engine_type_) {
case GEngineType::DYNAMIC : engine_ = CGRAPH_SAFE_MALLOC_COBJECT(GDynamicEngine) break;
case GEngineType::TOPO: engine_ = CGRAPH_SAFE_MALLOC_COBJECT(GTopoEngine) break;
case GEngineType::STATIC: engine_ = CGRAPH_SAFE_MALLOC_COBJECT(GStaticEngine) break;
default: CGRAPH_RETURN_ERROR_STATUS("unknown engine type")
}
engine_->thread_pool_ = thread_pool_;
status = engine_->setup(manager_elements_);
CGRAPH_FUNCTION_END
}
GElementManagerPtr GElementManager::setThreadPool(UThreadPoolPtr ptr) {
CGRAPH_ASSERT_NOT_NULL_THROW_ERROR(ptr)
this->thread_pool_ = ptr;
return this;
}
CSize GElementManager::calcMaxParaSize() {
CGRAPH_THROW_EXCEPTION_BY_CONDITION(!GMaxParaOptimizer::match(manager_elements_),
"cannot calculate max parallel size within groups")
return GMaxParaOptimizer::getMaxParaSize(manager_elements_);
}
CBool GElementManager::checkSerializable() {
if (engine_type_ != GEngineType::DYNAMIC) {
return false; // 目前仅支持动态引擎的执行方式
}
/**
* 判定思路：
* 1. 内部的element，均为可串行执行的
* 2. 当前element，不超过1个前驱或者后继
* 3. 有且仅有一个起点，一个终点
* 4. 有超时逻辑
*/
int frontSize = 0, tailSize = 0;
for (auto& cur : manager_elements_) {
if (!cur->isSerializable()
|| cur->run_before_.size() > 1
|| cur->dependence_.size() > 1
|| cur->isAsync()) {
return false;
}
if (cur->dependence_.empty()) {
frontSize++;
}
if (cur->run_before_.empty()) {
tailSize++;
}
}
return (1 == frontSize) && (1 == tailSize);
}
CSize GElementManager::trim() {
return GTrimOptimizer::trim(manager_elements_);
}
CStatus GElementManager::process(const GSortedGElementPtrSet& elements) {
CGRAPH_FUNCTION_BEGIN
CGRAPH_ASSERT_NOT_NULL(engine_)
// 主要是给 mutable 使用
status += engine_->setup(elements);
status += run();
CGRAPH_FUNCTION_END
}
CGRAPH_NAMESPACE_END