mirror of https://gitee.com/bigwinds/arangodb
131 lines
3.9 KiB
C++
131 lines
3.9 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
/// DISCLAIMER
|
|
///
|
|
/// Copyright 2019 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 Kaveh Vahedipour
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "HotBackup.h"
|
|
|
|
#include "Cluster/ClusterFeature.h"
|
|
#include "Cluster/ClusterMethods.h"
|
|
#include "Cluster/ServerState.h"
|
|
#include "StorageEngine/EngineSelectorFeature.h"
|
|
|
|
#ifdef USE_ENTERPRISE
|
|
#include "Enterprise/RocksDBEngine/RocksDBHotBackup.h"
|
|
#include "Enterprise/StorageEngine/HotBackupFeature.h"
|
|
#endif
|
|
|
|
namespace arangodb {
|
|
|
|
HotBackup::HotBackup(application_features::ApplicationServer& server)
|
|
#ifdef USE_ENTERPRISE
|
|
: _server(server) {
|
|
#else
|
|
{
|
|
#endif
|
|
if (ServerState::instance()->isCoordinator()) {
|
|
_engine = BACKUP_ENGINE::CLUSTER;
|
|
} else if (EngineSelectorFeature::isRocksDB()) {
|
|
_engine = BACKUP_ENGINE::ROCKSDB;
|
|
} else {
|
|
_engine = BACKUP_ENGINE::MMFILES;
|
|
}
|
|
}
|
|
|
|
|
|
arangodb::Result HotBackup::execute (
|
|
std::string const& command, VPackSlice const payload, VPackBuilder& report) {
|
|
|
|
switch (_engine) {
|
|
case BACKUP_ENGINE::ROCKSDB:
|
|
return executeRocksDB(command, payload, report);
|
|
case BACKUP_ENGINE::MMFILES:
|
|
return executeMMFiles(command, payload, report);
|
|
case BACKUP_ENGINE::CLUSTER:
|
|
return executeCoordinator(command, payload, report);
|
|
}
|
|
|
|
return arangodb::Result();
|
|
|
|
}
|
|
|
|
|
|
arangodb::Result HotBackup::executeRocksDB(
|
|
std::string const& command, VPackSlice const payload, VPackBuilder& report) {
|
|
|
|
#ifdef USE_ENTERPRISE
|
|
std::shared_ptr<RocksDBHotBackup> operation;
|
|
auto& feature = _server.getFeature<HotBackupFeature>();
|
|
operation = RocksDBHotBackup::operationFactory(feature, command, payload, report);
|
|
|
|
if (operation->valid()) {
|
|
operation->execute();
|
|
} // if
|
|
|
|
// if !valid() then !success() already set
|
|
if (!operation->success()) {
|
|
return arangodb::Result(
|
|
operation->restResponseError(), operation->errorMessage());
|
|
} // if
|
|
#endif
|
|
|
|
return arangodb::Result();
|
|
|
|
}
|
|
|
|
|
|
arangodb::Result HotBackup::executeCoordinator(
|
|
std::string const& command, VPackSlice const payload, VPackBuilder& report) {
|
|
#ifdef USE_ENTERPRISE
|
|
auto& feature = _server.getFeature<ClusterFeature>();
|
|
if (command == "create") {
|
|
return hotBackupCoordinator(feature, payload, report);
|
|
} else if (command == "lock") {
|
|
return arangodb::Result(
|
|
TRI_ERROR_NOT_IMPLEMENTED, "backup locks not implemented on coordinators");
|
|
} else if (command == "restore") {
|
|
return hotRestoreCoordinator(feature, payload, report);
|
|
} else if (command == "delete") {
|
|
return deleteHotBackupsOnCoordinator(feature, payload, report);
|
|
} else if (command == "list") {
|
|
return listHotBackupsOnCoordinator(feature, payload, report);
|
|
} else if (command == "upload") {
|
|
return uploadBackupsOnCoordinator(feature, payload, report);
|
|
} else if (command == "download") {
|
|
return downloadBackupsOnCoordinator(feature, payload, report);
|
|
} else {
|
|
return arangodb::Result(
|
|
TRI_ERROR_NOT_IMPLEMENTED, command + " is not implemented on coordinators");
|
|
}
|
|
#endif
|
|
|
|
// We'll never get here
|
|
return arangodb::Result();
|
|
|
|
}
|
|
|
|
arangodb::Result HotBackup::executeMMFiles(
|
|
std::string const& command, VPackSlice const payload, VPackBuilder& report) {
|
|
return arangodb::Result(
|
|
TRI_ERROR_NOT_IMPLEMENTED, "hot backup not implemented on MMFiles");
|
|
}
|
|
|
|
}
|