1
0
Fork 0
arangodb/arangod/Aql/DocumentProducingNode.cpp

84 lines
2.7 KiB
C++

////////////////////////////////////////////////////////////////////////////////
/// DISCLAIMER
///
/// Copyright 2014-2016 ArangoDB GmbH, Cologne, Germany
/// Copyright 2004-2014 triAGENS 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 Jan Steemann
////////////////////////////////////////////////////////////////////////////////
#include "DocumentProducingNode.h"
#include "Aql/ExecutionNode.h"
#include "Aql/ExecutionPlan.h"
#include "Aql/Variable.h"
#include <velocypack/Builder.h>
#include <velocypack/Iterator.h>
#include <velocypack/Value.h>
#include <velocypack/ValueType.h>
#include <velocypack/velocypack-aliases.h>
using namespace arangodb::aql;
DocumentProducingNode::DocumentProducingNode(Variable const* outVariable)
: _outVariable(outVariable) {
TRI_ASSERT(_outVariable != nullptr);
}
DocumentProducingNode::DocumentProducingNode(ExecutionPlan* plan,
arangodb::velocypack::Slice slice)
: _outVariable(
Variable::varFromVPack(plan->getAst(), slice, "outVariable")) {
TRI_ASSERT(_outVariable != nullptr);
if (slice.hasKey("projection")) {
// old format
VPackSlice p = slice.get("projection");
if (p.isArray()) {
for (auto const& it : VPackArrayIterator(p)) {
_projections.emplace_back(it.copyString());
break; // stop after first sub-attribute!
}
}
} else if (slice.hasKey("projections")) {
// new format
VPackSlice p = slice.get("projections");
if (p.isArray()) {
for (auto const& it : VPackArrayIterator(p)) {
if (it.isString()) {
_projections.emplace_back(it.copyString());
}
}
}
}
}
void DocumentProducingNode::toVelocyPack(arangodb::velocypack::Builder& builder) const {
builder.add(VPackValue("outVariable"));
_outVariable->toVelocyPack(builder);
// export in new format
builder.add("projections", VPackValue(VPackValueType::Array));
for (auto const& it : _projections) {
builder.add(VPackValue(it));
}
builder.close();
builder.add("producesResult",
VPackValue(dynamic_cast<ExecutionNode const*>(this)->isVarUsedLater(_outVariable)));
}