blob: ef10337b1f6858411942b0c0c64a21cbda9eb694 (
plain)
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
55
56
57
58
|
#include <iostream>
#include "ocelot.h"
#include "config.h"
#include "db.h"
#include "worker.h"
#include "schedule.h"
schedule::schedule(config * conf, worker * worker_obj, mysql * db_obj, site_comm * sc_obj) : work(worker_obj), db(db_obj), sc(sc_obj) {
load_config(conf);
counter = 0;
last_opened_connections = 0;
last_request_count = 0;
next_reap_peers = reap_peers_interval;
}
void schedule::load_config(config * conf) {
reap_peers_interval = conf->get_uint("reap_peers_interval");
schedule_interval = conf->get_uint("schedule_interval");
}
void schedule::reload_config(config * conf) {
load_config(conf);
}
//---------- Schedule - gets called every schedule_interval seconds
void schedule::handle(ev::timer &watcher, int events_flags) {
unsigned int cur_schedule_interval = watcher.repeat;
stats.connection_rate = (stats.opened_connections - last_opened_connections) / cur_schedule_interval;
stats.request_rate = (stats.requests - last_request_count) / cur_schedule_interval;
if (counter % 20 == 0) {
std::cout << stats.open_connections << " open, "
<< stats.opened_connections << " connections (" << stats.connection_rate << "/s), "
<< stats.requests << " requests (" << stats.request_rate << "/s)" << std::endl;
}
if (work->get_status() == CLOSING && db->all_clear() && sc->all_clear()) {
std::cout << "all clear, shutting down" << std::endl;
exit(0);
}
last_opened_connections = stats.opened_connections;
last_request_count = stats.requests;
db->flush();
sc->flush_tokens();
next_reap_peers -= cur_schedule_interval;
if (next_reap_peers <= 0) {
work->start_reaper();
next_reap_peers = reap_peers_interval;
}
counter++;
if (schedule_interval != cur_schedule_interval) {
watcher.set(schedule_interval, schedule_interval);
}
}
|
