mirror of https://github.com/Aidaho12/haproxy-wi
63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
from app.modules.db.db_model import KeepaliveRestart, Server
|
|
from app.modules.db.common import out_error
|
|
|
|
|
|
def select_keep_alive():
|
|
query = Server.select(Server.ip, Server.groups, Server.server_id).where(Server.active == 1)
|
|
try:
|
|
query_res = query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
else:
|
|
return query_res
|
|
|
|
|
|
def select_nginx_keep_alive():
|
|
query = Server.select(Server.ip, Server.groups, Server.server_id).where(Server.nginx_active == 1)
|
|
try:
|
|
query_res = query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
else:
|
|
return query_res
|
|
|
|
|
|
def select_apache_keep_alive():
|
|
query = Server.select(Server.ip, Server.groups, Server.server_id).where(Server.apache_active == 1)
|
|
try:
|
|
query_res = query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
else:
|
|
return query_res
|
|
|
|
|
|
def select_keepalived_keep_alive():
|
|
query = Server.select(Server.ip, Server.port, Server.groups, Server.server_id).where(Server.keepalived_active == 1)
|
|
try:
|
|
query_res = query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
else:
|
|
return query_res
|
|
|
|
|
|
def select_update_keep_alive_restart(server_id: int, service: str) -> int:
|
|
try:
|
|
restarted = KeepaliveRestart.get(
|
|
(KeepaliveRestart.server_id == server_id) &
|
|
(KeepaliveRestart.service == service)
|
|
).restarted
|
|
except Exception as e:
|
|
out_error(e)
|
|
else:
|
|
return restarted or 0
|
|
|
|
|
|
def update_keep_alive_restart(server_id: int, service: str, restarted: int) -> None:
|
|
query = KeepaliveRestart.insert(server_id=server_id, service=service, restarted=restarted).on_conflict('replace')
|
|
try:
|
|
query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|