threading_demo.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# author:yanshushuang
# datetime:2021/9/28

import json
from concurrent.futures import ThreadPoolExecutor, wait, ALL_COMPLETED
from alarm.engine import Alarmer
from alarm.get_params import all_funcs
from handler import dingding_handler, mail_handler,fake_handler
from logger import Logger

if __name__ == '__main__':
    my_logger = Logger(stream=True).get_logger()
    with open("config.json", "r", encoding="utf-8") as f:
        configs = json.load(f)['databases']
    t = ThreadPoolExecutor(max_workers=5)
    tasks = []
    for config in configs:
        user = config.get("user")
        passwd = config.get("passwd")
        host = config.get("host")
        port = config.get("port")
        slaves = config.get("slaves")
        tags = config.get("tags")
        my_functions = ["disk", "thread_connect", "innodb_buffer_hint_precent",
                        "trans_time", "innodb_buffer_pool_wait_free", "lock_waits"]
        my_func_dic = {name: all_funcs.get(name) for name in my_functions}
        client = Alarmer(user=user, passwd=passwd, host=host, port=port, slaves=slaves,
                         alarm_handler=[dingding_handler], tags=tags, logger=my_logger,
                         funcs=my_func_dic
                         )
        tasks.append(t.submit(client.th_run))
    wait(tasks, return_when=ALL_COMPLETED)