forked from mckelvin/redis-ctl
/
daemon.py
42 lines (32 loc) · 1.03 KB
/
daemon.py
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
import gevent
import gevent.monkey
gevent.monkey.patch_all()
import config
import handlers.base
import models.base
import stats
from daemonutils import stats_models as _
from algalon_cli import AlgalonClient
def run(interval, algalon_client, app):
from daemonutils.node_polling import NodeStatCollector
from daemonutils.cluster_task import TaskPoller
daemons = [
TaskPoller(app, interval),
NodeStatCollector(app, interval, algalon_client),
]
for d in daemons:
d.start()
for d in daemons:
d.join()
def main():
config.init_logging()
app = handlers.base.app
app.config['SQLALCHEMY_DATABASE_URI'] = config.SQLALCHEMY_DATABASE_URI
models.base.init_db(app)
if config.OPEN_FALCON and config.OPEN_FALCON['host_query']:
stats.init(**config.OPEN_FALCON)
algalon_client = (AlgalonClient(**config.ALGALON)
if config.ALGALON and config.ALGALON['dsn'] else None)
run(config.POLL_INTERVAL, algalon_client, app)
if __name__ == '__main__':
main()