database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[pipelineX](sort) Support sort operator (#23322)

Browse Source
This commit is contained in:
Gabriel
2023-08-22 19:36:50 +08:00
committed by GitHub
parent 6124eea55e
commit 1609b6cbf2
22 changed files with 452 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
be/src/runtime/runtime_state.cpp
View File

@ -417,21 +417,25 @@ int64_t RuntimeState::get_load_mem_limit() {
void RuntimeState::emplace_local_state(
int id, std::shared_ptr<doris::pipeline::PipelineXLocalState> state) {
std::unique_lock<std::mutex> l(_local_state_lock);
_op_id_to_local_state.emplace(id, state);
}
std::shared_ptr<doris::pipeline::PipelineXLocalState> RuntimeState::get_local_state(int id) {
std::unique_lock<std::mutex> l(_local_state_lock);
DCHECK(_op_id_to_local_state.find(id) != _op_id_to_local_state.end());
return _op_id_to_local_state[id];
}
void RuntimeState::emplace_sink_local_state(
int id, std::shared_ptr<doris::pipeline::PipelineXSinkLocalState> state) {
std::unique_lock<std::mutex> l(_local_sink_state_lock);
_op_id_to_sink_local_state.emplace(id, state);
}
std::shared_ptr<doris::pipeline::PipelineXSinkLocalState> RuntimeState::get_sink_local_state(
int id) {
std::unique_lock<std::mutex> l(_local_sink_state_lock);
DCHECK(_op_id_to_sink_local_state.find(id) != _op_id_to_sink_local_state.end());
return _op_id_to_sink_local_state[id];
}