mirror of https://gitee.com/bigwinds/arangodb
98 lines
3.5 KiB
C++
98 lines
3.5 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2016 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 Simon Grätzer
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "Utils.h"
|
|
#include "Basics/StringUtils.h"
|
|
|
|
#include "Utils/SingleCollectionTransaction.h"
|
|
#include "Utils/StandaloneTransactionContext.h"
|
|
#include "VocBase/LogicalCollection.h"
|
|
#include "VocBase/vocbase.h"
|
|
#include "VocBase/vocbase.h"
|
|
|
|
using namespace arangodb::pregel;
|
|
|
|
std::string const Utils::apiPrefix = "/_api/pregel/";
|
|
|
|
std::string const Utils::startExecutionPath = "startExecution";
|
|
std::string const Utils::startGSSPath = "startGSS";
|
|
std::string const Utils::finishedGSSPath = "finishedGSS";
|
|
std::string const Utils::messagesPath = "messages";
|
|
std::string const Utils::finalizeExecutionPath = "finalizeExecution";
|
|
|
|
std::string const Utils::executionNumberKey = "exn";
|
|
std::string const Utils::totalVertexCount = "vertexCount";
|
|
std::string const Utils::totalEdgeCount = "edgeCount";
|
|
|
|
std::string const Utils::collectionPlanIdMapKey = "collectionPlanIdMap";
|
|
std::string const Utils::vertexShardsListKey = "vertexShards";
|
|
std::string const Utils::edgeShardsListKey = "edgeShards";
|
|
|
|
// std::string const Utils::resultShardKey = "resultShard";
|
|
|
|
std::string const Utils::coordinatorIdKey = "coordinatorId";
|
|
std::string const Utils::algorithmKey = "algorithm";
|
|
std::string const Utils::globalSuperstepKey = "gss";
|
|
std::string const Utils::messagesKey = "msgs";
|
|
std::string const Utils::senderKey = "sender";
|
|
std::string const Utils::doneKey = "done";
|
|
|
|
std::string const Utils::parameterMapKey = "params";
|
|
std::string const Utils::aggregatorValuesKey = "aggregators";
|
|
|
|
std::string const Utils::edgeShardingKey = "_vertex";
|
|
|
|
std::string Utils::baseUrl(std::string dbName) {
|
|
return "/_db/" + basics::StringUtils::urlEncode(dbName) + Utils::apiPrefix;
|
|
}
|
|
|
|
std::string Utils::collectionFromToValue(std::string const& graphKey) {
|
|
std::size_t pos = graphKey.find('/');
|
|
return graphKey.substr(0, pos + 1);
|
|
}
|
|
|
|
std::string Utils::vertexKeyFromToValue(std::string const& graphKey) {
|
|
std::size_t pos = graphKey.find('/');
|
|
return graphKey.substr(
|
|
pos + 1, graphKey.length() - pos -
|
|
1); // if pos == -1 (npos) the entire string is returned
|
|
}
|
|
|
|
int64_t Utils::countDocuments(TRI_vocbase_t* vocbase,
|
|
CollectionID const& collection) {
|
|
SingleCollectionTransaction trx(StandaloneTransactionContext::Create(vocbase),
|
|
collection, TRI_TRANSACTION_READ);
|
|
|
|
int res = trx.begin();
|
|
if (res != TRI_ERROR_NO_ERROR) {
|
|
THROW_ARANGO_EXCEPTION(res);
|
|
}
|
|
OperationResult opResult = trx.count(collection);
|
|
res = trx.finish(opResult.code);
|
|
if (res != TRI_ERROR_NO_ERROR) {
|
|
THROW_ARANGO_EXCEPTION(res);
|
|
}
|
|
VPackSlice s = opResult.slice();
|
|
TRI_ASSERT(s.isNumber());
|
|
return s.getInt();
|
|
}
|