| import os.path |
| from typing import Callable |
|
|
| import psutil |
|
|
| from mod.project.node.dbutil import HttpNode, LoadSite, TcpNode |
| from mod.project.node.nodeutil import ServerNode, LocalNode |
| from .nginx_utils import NginxUtils |
| from .request_check import RequestChecker |
|
|
|
|
| def check_http_node(node:HttpNode, main_domain, log_call:Callable[[str], None]) -> str: |
| srv = ServerNode.new_by_id(node.node_id) |
| if not srv: |
| return "无法连接到目标节点的机器" |
| if srv.is_local: |
| return _check_local_http_node(node, main_domain, log_call) |
| if node.node_site_id != -1: |
| if not node.node_site_id: |
| log_call("负载服务节点：{}不存在，即将在目标机器创建网站！".format(node.node_site_name)) |
| site_id, err = srv.create_php_site(node.node_site_name, node.port) |
| if err: |
| return "无法在目标节点{}的机器创建网站：{}".format(srv.show_name(), err) |
|
|
| node.node_site_id = site_id |
| log_call("成功创建网站：{}".format(node.node_site_name)) |
|
|
| if srv and node.node_site_name != main_domain and main_domain!="" and not srv.has_domain(node.node_site_id, main_domain): |
| log_call("负载服务节点：{}的域名与主域名不一致，即将把主域名添加至目标机器".format(node.node_site_name)) |
| flag, msg = srv.add_domain(node.node_site_id, node.node_site_name, main_domain, node.port) |
| if not flag: |
| return "无法在目标节点{}的机器更新域名".format(srv.show_name()) |
| log_call("成功更新域名：{}".format(node.node_site_name)) |
|
|
| log_call("开始检查负载服务节点：{}访问情况".format(node.node_site_name)) |
| if not RequestChecker.check_http_node(node): |
| if node.port in (80, 443): |
| return "无法访问负载服务节点：{}".format(node.node_site_name) |
| log_call("负载服务节点：{}访问失败，将尝试在目标机器放行防火墙端口".format(node.node_site_name)) |
| flag, msg = srv.set_firewall_open(node.port, "tcp") |
| if not flag: |
| return "无法在目标节点{}的机器放行端口：{}".format(srv.show_name(), node.port) |
| if not RequestChecker.check_http_node(node): |
| return "无法访问负载服务节点：{}".format(node.node_site_name) |
|
|
| log_call("负载服务节点：{}访问成功".format(node.node_site_name)) |
| return "" |
|
|
|
|
| def _check_local_http_node(node:HttpNode, main_domain, log_call:Callable[[str], None]) -> str: |
| srv = LocalNode() |
| if main_domain != "" and main_domain == node.node_site_name: |
| return "使用本机节点作为负载服务节点时，网站域名不能与主域名一致" |
| if node.port in (80, 443): |
| return "使用本机节点作为负载服务节点时，端口不能为80或443" |
| if not node.node_site_id: |
| log_call("负载服务节点：{}不存在，即将在本机创建网站！".format(node.node_site_name)) |
| site_id, err = srv.create_php_site(node.node_site_name, node.port, ps="负载均衡：{}【子站点】".format(main_domain)) |
| if err: |
| return "无法在目标节点{}的机器创建网站：{}".format(srv.show_name(), err) |
|
|
| node.node_site_id = site_id |
| log_call("成功创建网站：{}".format(node.node_site_name)) |
| else: |
| srv.add_domain(node.node_site_id, node.node_site_name, node.node_site_name, node.port) |
| return "" |
|
|
|
|
| def check_http_load_data(load:LoadSite, log_call:Callable[[str], None]): |
| if not load.site_id: |
| log_call("主节点网站：{}不存在，即将在本机创建网站！".format(load.site_name)) |
| site_id, err = LocalNode().create_php_site(site_name=load.site_name, ps="负载均衡：{}【主站点】".format(load.ps)) |
| if err: |
| return "无法在本机创建网站：{}".format(err) |
|
|
| load.site_id = site_id |
|
|
| if LocalNode.site_proxy_list(load.site_name): |
| return "当前选择的主节点网站存在反向代理配置，无法为其设置负载均衡" |
| if load.http_config["http_alg"] == "sticky_cookie": |
| if not NginxUtils.has_sticky_module(): |
| return "负载均衡的cookie会话跟随，需要Nginx安装sticky-module模块，您可以尝试在【软件商店】更新、重装或切换Nginx版本" |
|
|
| def check_tcp_load_data(load: LoadSite, log_call: Callable[[str], None]) -> str: |
| |
| for conn in psutil.net_connections(): |
| if conn.status == 'LISTEN' and conn.laddr.port == load.tcp_config["port"]: |
| pid = conn.pid |
| if psutil.Process(pid).exe().find("nginx") == -1: |
| return "端口{}被占用，请注意是否影响nginx启动".format(load.tcp_config["port"]) |
| return "" |
|
|
| def check_tcp_node(tcp_node: TcpNode, log_call: Callable[[str], None]): |
| log_call("开始检查负载服务节点：{}:{}访问情况".format(tcp_node.host, tcp_node.port)) |
| if not RequestChecker.check_tcp_node(tcp_node): |
| return "无法访问负载服务节点：{}:{}".format(tcp_node.host, tcp_node.port) |
| return "" |
|
|
|
|
