mirror of https://gitee.com/bigwinds/arangodb
315 lines
13 KiB
C++
315 lines
13 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_ROCKSDB_ROCKSDB_REST_REPLICATION_HANDLER_H
|
|
#define ARANGOD_ROCKSDB_ROCKSDB_REST_REPLICATION_HANDLER_H 1
|
|
|
|
#include "RestHandler/RestReplicationHandler.h"
|
|
|
|
#include "RocksDBEngine/RocksDBReplicationManager.h"
|
|
|
|
namespace arangodb {
|
|
|
|
class SingleCollectionTransaction;
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief replication request handler
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
class RocksDBRestReplicationHandler : public RestReplicationHandler {
|
|
public:
|
|
RocksDBRestReplicationHandler(GeneralRequest*, GeneralResponse*);
|
|
~RocksDBRestReplicationHandler();
|
|
|
|
public:
|
|
RestStatus execute() override;
|
|
char const* name() const override final {
|
|
return "RocksDBRestReplicationHandler";
|
|
}
|
|
|
|
private:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @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() override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @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 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 handle a follow command for the replication log
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandLoggerFollow() override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle the command to determine the transactions that were open
|
|
/// at a certain point in time
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandDetermineOpenTransactions() override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a batch command
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandBatch() override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief return the inventory (current replication and collection state)
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandInventory() override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a restore command for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandRestoreCollection() override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief produce list of keys for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandCreateKeys();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief returns a key range
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandGetKeys();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief returns date for a key range
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandFetchKeys();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief remove a list of keys for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandRemoveKeys();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a dump command for a specific collection
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandDump();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a sync command
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandSync();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief handle a server-id command
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void handleCommandServerId();
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @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 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 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 restores the structure of a collection TODO MOVE
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
int processRestoreCollection(VPackSlice const&, bool, bool, bool,
|
|
std::string&);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the structure of a collection, coordinator case
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
int processRestoreCollectionCoordinator(VPackSlice const&, bool, bool, bool,
|
|
uint64_t, std::string&, uint64_t,
|
|
bool);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief creates a collection, based on the VelocyPack provided TODO: MOVE
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
int createCollection(VPackSlice, arangodb::LogicalCollection**, bool);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the indexes of a collection TODO MOVE
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
int processRestoreIndexes(VPackSlice const&, bool, std::string&) override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the indexes of a collection, coordinator case
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
int processRestoreIndexesCoordinator(VPackSlice const&, bool, std::string&) override;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief restores the data of a collection TODO MOVE
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
int processRestoreData(std::string const&, bool, bool, std::string&);
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief determine chunk size from request
|
|
/// Reads chunkSize attribute from request
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
uint64_t determineChunkSize() const;
|
|
|
|
private:
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief minimum chunk size
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static uint64_t const _defaultChunkSize;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief maximum chunk size
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static uint64_t const _maxChunkSize;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief condition locker to wake up holdReadLockCollection jobs
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static arangodb::basics::ConditionVariable _condVar;
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
/// @brief global set of ids of holdReadLockCollection jobs, an
|
|
/// id mapping to false here indicates that a request to get the
|
|
/// read lock has been started, the mapped value holds the
|
|
/// actual transaction. if the read lock is not yet acquired, the
|
|
/// shared_ptr will contain a nullptr still.
|
|
/// To cancel the read lock, remove the entry here (under the protection
|
|
/// of the mutex of condVar) and send a broadcast to the condition variable,
|
|
/// the job with that id is terminated. If it timeouts, then
|
|
/// the read lock is released automatically and the entry here
|
|
/// is deleted.
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
static std::unordered_map<std::string, std::shared_ptr<SingleCollectionTransaction>> _holdReadLockJobs;
|
|
|
|
RocksDBReplicationManager* _manager;
|
|
};
|
|
}
|
|
|
|
#endif
|