1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
#include <event2/thread.h>
#include <gflags/gflags.h>
#include <glog/logging.h>
#include <atomic>
#include "util/etcd.h"
#include "util/libevent_wrapper.h"
#include "util/masterelection.h"
#include "util/thread_pool.h"
namespace libevent = cert_trans::libevent;
using cert_trans::EtcdClient;
using cert_trans::MasterElection;
using cert_trans::ThreadPool;
using cert_trans::UrlFetcher;
using std::make_shared;
using std::shared_ptr;
DEFINE_string(etcd, "127.0.0.1", "etcd server address");
DEFINE_int32(etcd_port, 4001, "etcd server port");
DEFINE_string(proposal_dir, "/master", "path to watch");
DEFINE_string(node_id, "", "unique node id.");
int main(int argc, char* argv[]) {
google::ParseCommandLineFlags(&argc, &argv, true);
google::InitGoogleLogging(argv[0]);
evthread_use_pthreads();
CHECK(!FLAGS_node_id.empty()) << "Must set --node_id";
const shared_ptr<libevent::Base> event_base(make_shared<libevent::Base>());
ThreadPool pool;
UrlFetcher fetcher(event_base.get(), &pool);
EtcdClient etcd(&pool, &fetcher, FLAGS_etcd, FLAGS_etcd_port);
MasterElection election(event_base, &etcd, FLAGS_proposal_dir,
FLAGS_node_id);
election.StartElection();
libevent::EventPumpThread pump(event_base);
LOG(INFO) << "Waiting to become master...";
election.WaitToBecomeMaster();
LOG(INFO) << "I'm the boss!";
sleep(10);
LOG(INFO) << "Giving it all up and going fishing instead...";
election.StopElection();
LOG(INFO) << "Gone fishin'.";
return 0;
}
|
