mirror of https://gitee.com/bigwinds/arangodb
95 lines
3.6 KiB
C++
95 lines
3.6 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 Dr. Frank Celler
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "Basics/Common.h"
|
|
#include "Basics/directories.h"
|
|
|
|
#include "ApplicationFeatures/BasicPhase.h"
|
|
#include "ApplicationFeatures/CommunicationPhase.h"
|
|
#include "ApplicationFeatures/ConfigFeature.h"
|
|
#include "ApplicationFeatures/GreetingsPhase.h"
|
|
#include "ApplicationFeatures/ShellColorsFeature.h"
|
|
#include "ApplicationFeatures/ShutdownFeature.h"
|
|
#include "ApplicationFeatures/TempFeature.h"
|
|
#include "ApplicationFeatures/VersionFeature.h"
|
|
#include "Basics/ArangoGlobalContext.h"
|
|
#include "Benchmark/BenchFeature.h"
|
|
#include "Logger/Logger.h"
|
|
#include "Logger/LoggerFeature.h"
|
|
#include "ProgramOptions/ProgramOptions.h"
|
|
#include "Random/RandomFeature.h"
|
|
#include "Shell/ClientFeature.h"
|
|
#include "Ssl/SslFeature.h"
|
|
|
|
using namespace arangodb;
|
|
using namespace arangodb::application_features;
|
|
using namespace arangodb::basics;
|
|
|
|
int main(int argc, char* argv[]) {
|
|
TRI_GET_ARGV(argc, argv);
|
|
return ClientFeature::runMain(argc, argv, [&](int argc, char* argv[]) -> int {
|
|
ArangoGlobalContext context(argc, argv, BIN_DIRECTORY);
|
|
context.installHup();
|
|
|
|
std::shared_ptr<options::ProgramOptions> options(
|
|
new options::ProgramOptions(argv[0], "Usage: arangobench [<options>]",
|
|
"For more information use:", BIN_DIRECTORY));
|
|
ApplicationServer server(options, BIN_DIRECTORY);
|
|
int ret;
|
|
|
|
server.addFeature(new application_features::CommunicationFeaturePhase(server));
|
|
server.addFeature(new application_features::BasicFeaturePhase(server, true));
|
|
server.addFeature(new application_features::GreetingsFeaturePhase(server, true));
|
|
|
|
server.addFeature(new BenchFeature(server, &ret));
|
|
server.addFeature(new ClientFeature(server, false));
|
|
server.addFeature(new ConfigFeature(server, "arangobench"));
|
|
server.addFeature(new LoggerFeature(server, false));
|
|
server.addFeature(new RandomFeature(server));
|
|
server.addFeature(new ShellColorsFeature(server));
|
|
server.addFeature(new ShutdownFeature(server, {"Bench"}));
|
|
server.addFeature(new SslFeature(server));
|
|
server.addFeature(new TempFeature(server, "arangobench"));
|
|
server.addFeature(new VersionFeature(server));
|
|
|
|
try {
|
|
server.run(argc, argv);
|
|
if (server.helpShown()) {
|
|
// --help was displayed
|
|
ret = EXIT_SUCCESS;
|
|
}
|
|
} catch (std::exception const& ex) {
|
|
LOG_TOPIC("0a1a9", ERR, arangodb::Logger::FIXME)
|
|
<< "arangobench terminated because of an unhandled exception: " << ex.what();
|
|
ret = EXIT_FAILURE;
|
|
} catch (...) {
|
|
LOG_TOPIC("61697", ERR, arangodb::Logger::FIXME)
|
|
<< "arangobench terminated because of an unhandled exception of "
|
|
"unknown type";
|
|
ret = EXIT_FAILURE;
|
|
}
|
|
|
|
return context.exit(ret);
|
|
});
|
|
}
|