mirror of https://gitee.com/bigwinds/arangodb
518 lines
22 KiB
C++
518 lines
22 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
|
|
/// @author Jan Christoph Uhde
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef ARANGOD_REST_HANDLER_REST_REPLICATION_HANDLER_H
|
|
#define ARANGOD_REST_HANDLER_REST_REPLICATION_HANDLER_H 1
|
|
|
|
#include "Aql/types.h"
|
|
#include "Basics/Common.h"
|
|
#include "Basics/Result.h"
|
|
#include "Cluster/ResultT.h"
|
|
#include "Replication/Syncer.h"
|
|
#include "Replication/common-defines.h"
|
|
#include "RestHandler/RestVocbaseBaseHandler.h"
|
|
|
|
namespace arangodb {
|
|
class ClusterInfo;
|
|
class CollectionNameResolver;
|
|
class LogicalCollection;
|
|
class ReplicationApplier;
|
|
class SingleCollectionTransaction;
|
|
|
|
namespace transaction {
|
|
class Methods;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief replication request handler
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
class RestReplicationHandler : public RestVocbaseBaseHandler {
|
|
public:
|
|
RequestLane lane() const override final {
|
|
auto const& suffixes = _request->suffixes();
|
|
|
|
size_t const len = suffixes.size();
|
|
if (len >= 1) {
|
|
std::string const& command = suffixes[0];
|
|
if (command == AddFollower || command == HoldReadLockCollection ||
|
|
command == RemoveFollower || command == LoggerFollow) {
|
|
return RequestLane::SERVER_REPLICATION_CATCHUP;
|
|
}
|
|
}
|
|
return RequestLane::SERVER_REPLICATION;
|
|
}
|
|
|
|
RestStatus execute() override;
|
|
|
|
// Never instantiate this.
|
|
// Only specific implementations allowed
|
|
protected:
|
|
RestReplicationHandler(GeneralRequest*, GeneralResponse*);
|
|
~RestReplicationHandler();
|
|
|
|
protected:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief list of available commands
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
static std::string const LoggerState;
|
|
static std::string const LoggerTickRanges;
|
|
static std::string const LoggerFirstTick;
|
|
static std::string const LoggerFollow;
|
|
static std::string const OpenTransactions;
|
|
static std::string const Batch;
|
|
static std::string const Barrier;
|
|
static std::string const Inventory;
|
|
static std::string const Keys;
|
|
static std::string const Dump;
|
|
static std::string const RestoreCollection;
|
|
static std::string const RestoreIndexes;
|
|
static std::string const RestoreData;
|
|
static std::string const RestoreView;
|
|
static std::string const Sync;
|
|
static std::string const MakeSlave;
|
|
static std::string const ServerId;
|
|
static std::string const ApplierConfig;
|
|
static std::string const ApplierStart;
|
|
static std::string const ApplierStop;
|
|
static std::string const ApplierState;
|
|
static std::string const ApplierStateAll;
|
|
static std::string const ClusterInventory;
|
|
static std::string const AddFollower;
|
|
static std::string const RemoveFollower;
|
|
static std::string const SetTheLeader;
|
|
static std::string const HoldReadLockCollection;
|
|
|
|
protected:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief creates an error if called on a coordinator server
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
bool isCoordinatorError();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief turn the server into a slave of another
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandMakeSlave();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief forward a command in the coordinator case
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleTrampolineCoordinator();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief returns the cluster inventory, only on coordinator
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandClusterInventory();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a restore command for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandRestoreCollection();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a restore command for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandRestoreIndexes();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a restore command for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandRestoreData();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a restore of all views for this collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandRestoreView();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a server-id command
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandServerId();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a sync command
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandSync();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the configuration of the the replication applier
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandApplierGetConfig();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief configure the replication applier
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandApplierSetConfig();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief start the replication applier
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandApplierStart();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief stop the replication applier
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandApplierStop();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the state of the replication applier
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandApplierGetState();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the state of the all replication applier
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandApplierGetStateAll();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief delete the replication applier state
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandApplierDeleteState();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief add a follower of a shard to the list of followers
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandAddFollower();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief remove a follower of a shard from the list of followers
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandRemoveFollower();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief update the leader of a shard
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandSetTheLeader();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief hold a read lock on a collection to stop writes temporarily
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandHoldReadLockCollection();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief check if we are holding a read lock on a collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandCheckHoldReadLockCollection();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief cancel holding a read lock on a collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandCancelHoldReadLockCollection();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief get an ID for a hold read lock job
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandGetIdForReadLockCollection();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the state of the replication logger
|
|
/// @route GET logger-state
|
|
/// @caller Syncer::getMasterState
|
|
/// @response VPackObject describing the ServerState in a certain point
|
|
/// * state (server state)
|
|
/// * server (version / id)
|
|
/// * clients (list of followers)
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandLoggerState();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the first tick available in a logfile
|
|
/// @route GET logger-first-tick
|
|
/// @caller js/client/modules/@arangodb/replication.js
|
|
/// @response VPackObject with minTick of LogfileManager->ranges()
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandLoggerFirstTick();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the available logfile range
|
|
/// @route GET logger-tick-ranges
|
|
/// @caller js/client/modules/@arangodb/replication.js
|
|
/// @response VPackArray, containing info about each datafile
|
|
/// * filename
|
|
/// * status
|
|
/// * tickMin - tickMax
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandLoggerTickRanges();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief determine chunk size from request
|
|
/// Reads chunkSize attribute from request
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
uint64_t determineChunkSize() const;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief Grant temporary restore rights
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
void grantTemporaryRights();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief Get correct replication applier, based on global paramerter
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
ReplicationApplier* getApplier(bool& global);
|
|
|
|
private:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the structure of a collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result processRestoreCollection(VPackSlice const&, bool overwrite, bool force);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the structure of a collection, coordinator case
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result processRestoreCollectionCoordinator(VPackSlice const&, bool overwrite,
|
|
bool force, uint64_t numberOfShards,
|
|
uint64_t replicationFactor,
|
|
uint64_t minReplicationFactor,
|
|
bool ignoreDistributeShardsLikeErrors);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the data of the _users collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result processRestoreUsersBatch(std::string const& colName);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the data of a collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result processRestoreDataBatch(transaction::Methods& trx, std::string const& colName);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the indexes of a collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result processRestoreIndexes(VPackSlice const&, bool);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the indexes of a collection, coordinator case
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result processRestoreIndexesCoordinator(VPackSlice const&, bool);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the data of a collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result processRestoreData(std::string const& colName);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief parse an input batch
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
Result parseBatch(std::string const& collectionName,
|
|
std::unordered_map<std::string, VPackValueLength>& latest,
|
|
VPackBuilder& allMarkers);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief creates a collection, based on the VelocyPack provided
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
int createCollection(VPackSlice, arangodb::LogicalCollection**);
|
|
|
|
private:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief minimum chunk size
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static uint64_t const _defaultChunkSize;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief maximum chunk size
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static uint64_t const _maxChunkSize;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief timeout for tombstones
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static std::chrono::hours const _tombstoneTimeout;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief lock for the tombstone list
|
|
/// I do not think that this will ever be a bottleneck,
|
|
/// if it is we can easily make one lock per vocbase by
|
|
/// modifying the tombstones map.
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static basics::ReadWriteLock _tombLock;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief tombstones, should only be used, if a lock is cancelled
|
|
/// before it was actually registered and therefor only seldomly
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static std::unordered_map<std::string, std::chrono::time_point<std::chrono::steady_clock>> _tombstones;
|
|
|
|
protected:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// SECTION:
|
|
/// Functions to be implemented by specialization
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a follow command for the replication log
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandLoggerFollow() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle the command to determine the transactions that were open
|
|
/// at a certain point in time
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandDetermineOpenTransactions() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a batch command
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandBatch() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief add or remove a WAL logfile barrier
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandBarrier() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the inventory (current replication and collection state)
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandInventory() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief produce list of keys for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandCreateKeys() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief returns a key range
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandGetKeys() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief returns date for a key range
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandFetchKeys() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief remove a list of keys for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandRemoveKeys() = 0;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a dump command for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
virtual void handleCommandDump() = 0;
|
|
|
|
private:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief Cleanup tombstones that have expired
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
void timeoutTombstones() const;
|
|
|
|
bool isTombstoned(aql::QueryId id) const;
|
|
|
|
void registerTombstone(aql::QueryId id) const;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief Create a blocking transaction for the given collectionName,
|
|
/// It will be registered with the given id, and it will have
|
|
/// the given time to live.
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
Result createBlockingTransaction(aql::QueryId id, LogicalCollection& col,
|
|
double ttl, AccessMode::Type access) const;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief Test if we already have the read-lock
|
|
/// Will return true, if we have it and can use it
|
|
/// Will return false, if we are still in the process of getting it.
|
|
/// Will return error, if the lock has expired.
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
ResultT<bool> isLockHeld(aql::QueryId id) const;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief compute a local checksum for the given collection
|
|
/// Will return error if the lock has expired.
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
ResultT<std::string> computeCollectionChecksum(aql::QueryId readLockId,
|
|
LogicalCollection* col) const;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief Cacnel the lock with the given id
|
|
/// Will return true, if we did have the lock
|
|
/// Will return false, if we were still in the process of getting it.
|
|
/// Will return error if the lock has expired or is not found.
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
ResultT<bool> cancelBlockingTransaction(aql::QueryId id) const;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief Validate that the requesting user has access rights to this route
|
|
/// Will return TRI_ERROR_NO_ERROR if user has access
|
|
/// Will return error code otherwise.
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
Result testPermissions();
|
|
};
|
|
} // namespace arangodb
|
|
#endif
|