mirror of https://gitee.com/bigwinds/arangodb
92 lines
2.8 KiB
C++
92 lines
2.8 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 Kaveh Vahedipour
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "AgentActivator.h"
|
|
|
|
#include <chrono>
|
|
#include <thread>
|
|
|
|
#include "Agency/Agent.h"
|
|
#include "Agency/ActivationCallback.h"
|
|
#include "Basics/ConditionLocker.h"
|
|
|
|
using namespace arangodb::consensus;
|
|
using namespace std::chrono;
|
|
|
|
AgentActivator::AgentActivator() : Thread("AgentActivator"), _agent(nullptr) {}
|
|
|
|
AgentActivator::AgentActivator(Agent* agent, std::string const& failed,
|
|
std::string const& replacement)
|
|
: Thread("AgentActivator"),
|
|
_agent(agent),
|
|
_failed(failed),
|
|
_replacement(replacement) {}
|
|
|
|
// Shutdown if not already
|
|
AgentActivator::~AgentActivator() {
|
|
LOG_TOPIC(DEBUG, Logger::AGENCY) << "Done activating " << _replacement;
|
|
shutdown();
|
|
}
|
|
|
|
void AgentActivator::run() {
|
|
|
|
LOG_TOPIC(DEBUG, Logger::AGENCY) << "Starting activation of " << _replacement;
|
|
|
|
std::string const path = privApiPrefix + "activate";
|
|
auto const started = system_clock::now();
|
|
auto const timeout = seconds(60);
|
|
auto const endpoint = _agent->config().pool().at(_replacement);
|
|
|
|
CONDITION_LOCKER(guard, _cv);
|
|
|
|
while (!this->isStopping()) {
|
|
|
|
// All snapshots and all logs
|
|
query_t allLogs = _agent->allLogs();
|
|
|
|
auto headerFields =
|
|
std::make_unique<std::unordered_map<std::string, std::string>>();
|
|
arangodb::ClusterComm::instance()->asyncRequest(
|
|
"1", 1, endpoint, rest::RequestType::POST, path,
|
|
std::make_shared<std::string>(allLogs->toJson()), headerFields,
|
|
std::make_shared<ActivationCallback>(_agent, _failed, _replacement),
|
|
5.0, true, 1.0);
|
|
|
|
_cv.wait(10000000); // 10 sec
|
|
|
|
if ((std::chrono::system_clock::now() - started) > timeout) {
|
|
_agent->failedActivation(_failed, _replacement);
|
|
LOG_TOPIC(WARN, Logger::AGENCY)
|
|
<< "Timed out while activating agent " << _replacement;
|
|
break;
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
void AgentActivator::beginShutdown() {
|
|
Thread::beginShutdown();
|
|
CONDITION_LOCKER(guard, _cv);
|
|
guard.broadcast();
|
|
}
|