mirror of https://gitee.com/bigwinds/arangodb
96 lines
3.6 KiB
C++
96 lines
3.6 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2018 ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|
/// you may not use this file except in compliance with the License.
|
|
/// You may obtain a copy of the License at
|
|
///
|
|
/// http://www.apache.org/licenses/LICENSE-2.0
|
|
///
|
|
/// Unless required by applicable law or agreed to in writing, software
|
|
/// distributed under the License is distributed on an "AS IS" BASIS,
|
|
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
/// See the License for the specific language governing permissions and
|
|
/// limitations under the License.
|
|
///
|
|
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
|
|
///
|
|
/// @author Michael Hackstein
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "MultiDependencySingleRowFetcher.h"
|
|
|
|
#include "Aql/AqlItemBlock.h"
|
|
#include "Aql/DependencyProxy.h"
|
|
|
|
using namespace arangodb;
|
|
using namespace arangodb::aql;
|
|
|
|
MultiDependencySingleRowFetcher::DependencyInfo::DependencyInfo()
|
|
: _upstreamState{ExecutionState::HASMORE}, _rowIndex{0} {}
|
|
|
|
MultiDependencySingleRowFetcher::MultiDependencySingleRowFetcher(DependencyProxy<false>& executionBlock)
|
|
: _dependencyProxy(&executionBlock) {}
|
|
|
|
std::pair<ExecutionState, SharedAqlItemBlockPtr> MultiDependencySingleRowFetcher::fetchBlockForDependency(
|
|
size_t dependency, size_t atMost) {
|
|
TRI_ASSERT(!_dependencyInfos.empty());
|
|
atMost = (std::min)(atMost, ExecutionBlock::DefaultBatchSize());
|
|
TRI_ASSERT(dependency < _dependencyInfos.size());
|
|
|
|
auto& depInfo = _dependencyInfos[dependency];
|
|
TRI_ASSERT(depInfo._upstreamState != ExecutionState::DONE);
|
|
|
|
// There are still some blocks left that ask their parent even after they got
|
|
// DONE the last time, and I don't currently have time to track them down.
|
|
// Thus the following assert is commented out.
|
|
// TRI_ASSERT(_upstreamState != ExecutionState::DONE);
|
|
auto res = _dependencyProxy->fetchBlockForDependency(dependency, atMost);
|
|
depInfo._upstreamState = res.first;
|
|
|
|
return res;
|
|
}
|
|
|
|
std::pair<ExecutionState, size_t> MultiDependencySingleRowFetcher::skipSomeForDependency(
|
|
size_t const dependency, size_t const atMost) {
|
|
TRI_ASSERT(!_dependencyInfos.empty());
|
|
TRI_ASSERT(dependency < _dependencyInfos.size());
|
|
auto& depInfo = _dependencyInfos[dependency];
|
|
TRI_ASSERT(depInfo._upstreamState != ExecutionState::DONE);
|
|
|
|
// There are still some blocks left that ask their parent even after they got
|
|
// DONE the last time, and I don't currently have time to track them down.
|
|
// Thus the following assert is commented out.
|
|
// TRI_ASSERT(_upstreamState != ExecutionState::DONE);
|
|
auto res = _dependencyProxy->skipSomeForDependency(dependency, atMost);
|
|
depInfo._upstreamState = res.first;
|
|
|
|
return res;
|
|
}
|
|
|
|
MultiDependencySingleRowFetcher::MultiDependencySingleRowFetcher()
|
|
: _dependencyProxy(nullptr) {}
|
|
|
|
RegisterId MultiDependencySingleRowFetcher::getNrInputRegisters() const {
|
|
return _dependencyProxy->getNrInputRegisters();
|
|
}
|
|
|
|
void MultiDependencySingleRowFetcher::initDependencies() {
|
|
// Need to setup the dependencies, they are injected lazily.
|
|
TRI_ASSERT(_dependencyProxy->numberDependencies() > 0);
|
|
TRI_ASSERT(_dependencyInfos.empty());
|
|
_dependencyInfos.reserve(_dependencyProxy->numberDependencies());
|
|
for (size_t i = 0; i < _dependencyProxy->numberDependencies(); ++i) {
|
|
_dependencyInfos.emplace_back(DependencyInfo{});
|
|
}
|
|
}
|
|
|
|
size_t MultiDependencySingleRowFetcher::numberDependencies() {
|
|
if (_dependencyInfos.empty()) {
|
|
initDependencies();
|
|
}
|
|
return _dependencyInfos.size();
|
|
}
|