mirror of https://gitee.com/bigwinds/arangodb
129 lines
3.8 KiB
C++
129 lines
3.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 "Inception.h"
|
|
|
|
#include "Agency/Agent.h"
|
|
#include "Agency/GossipCallback.h"
|
|
#include "Basics/ConditionLocker.h"
|
|
|
|
#include <chrono>
|
|
#include <thread>
|
|
|
|
using namespace arangodb::consensus;
|
|
|
|
Inception::Inception()
|
|
: Thread("Inception"), _agent(nullptr) {}
|
|
|
|
Inception::Inception(Agent* agent)
|
|
: Thread("Inception"), _agent(agent) {}
|
|
|
|
// Shutdown if not already
|
|
Inception::~Inception() { shutdown(); }
|
|
|
|
void Inception::beginShutdown() {
|
|
Thread::beginShutdown();
|
|
}
|
|
|
|
bool Inception::start() { return Thread::start(); }
|
|
|
|
/// Gossip to others
|
|
/// - Get snapshot of gossip peers and agent pool
|
|
/// - Create outgoing gossip.
|
|
/// - Send to all peers
|
|
void Inception::run() {
|
|
|
|
TRI_ASSERT(_agent != nullptr);
|
|
|
|
auto s = std::chrono::system_clock::now();
|
|
std::chrono::seconds timeout(120);
|
|
size_t i = 0;
|
|
//bool cs = false;
|
|
while (!this->isStopping()) {
|
|
|
|
config_t config = _agent->config(); // get a copy of conf
|
|
|
|
query_t out = std::make_shared<Builder>();
|
|
out->openObject();
|
|
out->add("endpoint", VPackValue(config.endpoint()));
|
|
out->add("id", VPackValue(config.id()));
|
|
out->add("pool", VPackValue(VPackValueType::Object));
|
|
for (auto const& i : config.pool()) {
|
|
out->add(i.first,VPackValue(i.second));
|
|
}
|
|
out->close();
|
|
out->close();
|
|
|
|
std::string path = "/_api/agency_priv/gossip";
|
|
|
|
for (auto const& p : config.gossipPeers()) { // gossip peers
|
|
if (p != config.endpoint()) {
|
|
std::string clientid = config.id() + std::to_string(i++);
|
|
auto hf =
|
|
std::make_unique<std::unordered_map<std::string, std::string>>();
|
|
arangodb::ClusterComm::instance()->asyncRequest(
|
|
clientid, 1, p, GeneralRequest::RequestType::POST, path,
|
|
std::make_shared<std::string>(out->toJson()), hf,
|
|
std::make_shared<GossipCallback>(_agent), 1.0, true);
|
|
}
|
|
}
|
|
|
|
for (auto const& pair : config.pool()) { // pool entries
|
|
if (pair.second != config.endpoint()) {
|
|
std::string clientid = config.id() + std::to_string(i++);
|
|
auto hf =
|
|
std::make_unique<std::unordered_map<std::string, std::string>>();
|
|
arangodb::ClusterComm::instance()->asyncRequest(
|
|
clientid, 1, pair.second, GeneralRequest::RequestType::POST, path,
|
|
std::make_shared<std::string>(out->toJson()), hf,
|
|
std::make_shared<GossipCallback>(_agent), 1.0, true);
|
|
}
|
|
}
|
|
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(250));
|
|
|
|
if ((std::chrono::system_clock::now()-s) > timeout) {
|
|
if (config.poolComplete()) {
|
|
LOG_TOPIC(DEBUG, Logger::AGENCY) << "Stopping active gossipping!";
|
|
} else {
|
|
LOG_TOPIC(ERR, Logger::AGENCY) <<
|
|
"Failed to find complete pool of agents. Giving up!";
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (config.poolComplete()) {
|
|
//if(!cs) {
|
|
_agent->startConstituent();
|
|
break;
|
|
//cs = true;
|
|
//}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|