126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
|
import socket
|
|||
|
|
import subprocess
|
|||
|
|
import time
|
|||
|
|
from datetime import datetime
|
|||
|
|
import os
|
|||
|
|
import mysql.connector
|
|||
|
|
from zzb_data_prod.config import MYSQL_HOST,MYSQL_USER,MYSQL_PASSWORD,MYSQL_DB
|
|||
|
|
|
|||
|
|
def get_time():
|
|||
|
|
return datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|||
|
|
|
|||
|
|
|
|||
|
|
def check_port(host, port):
|
|||
|
|
try:
|
|||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|||
|
|
sock.settimeout(5)
|
|||
|
|
result = sock.connect_ex((host, port))
|
|||
|
|
sock.close()
|
|||
|
|
return result == 0 # 返回布尔值，表示端口是否可用
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[{get_time()}] 端口检测异常: {str(e)}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def restart_service():
|
|||
|
|
try:
|
|||
|
|
subprocess.run(["bash", "/root/docker/milvus/standalone_embed.sh", "restart"])
|
|||
|
|
print(f"[{get_time()}] milvus服务重启成功")
|
|||
|
|
return True
|
|||
|
|
except subprocess.CalledProcessError as e:
|
|||
|
|
print(f"[{get_time()}] 服务重启失败: {str(e)}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def start_application_process():
|
|||
|
|
"""启动8000端口对应的应用进程"""
|
|||
|
|
try:
|
|||
|
|
# 先尝试停止可能存在的旧进程
|
|||
|
|
time.sleep(2) # 给进程停止的时间
|
|||
|
|
# 进入应用目录
|
|||
|
|
# 启动新进程
|
|||
|
|
subprocess.run(
|
|||
|
|
["bash", "/root/pdf_parser/restart_app.sh"],
|
|||
|
|
)
|
|||
|
|
print(f"[{get_time()}] 应用进程(8000端口)已成功启动")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[{get_time()}] 启动应用进程失败: {str(e)}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_local_ip():
|
|||
|
|
try:
|
|||
|
|
# 创建一个 UDP 套接字
|
|||
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|||
|
|
# 连接到一个外部地址（这里使用 Google 的公共 DNS 服务器）
|
|||
|
|
s.connect(("8.8.8.8", 80))
|
|||
|
|
# 获取本地套接字的 IP 地址
|
|||
|
|
local_ip = s.getsockname()[0]
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[{get_time()}] 获取内网 IP 失败: {e}")
|
|||
|
|
local_ip = "127.0.0.1" # 如果失败，返回本地回环地址
|
|||
|
|
finally:
|
|||
|
|
s.close() # 关闭套接字
|
|||
|
|
return local_ip
|
|||
|
|
|
|||
|
|
def monitor_port_8000():
|
|||
|
|
"""监控8000端口，如果异常则启动应用进程"""
|
|||
|
|
print(f"[{get_time()}] 检查8000端口状态...")
|
|||
|
|
port_available = check_port("127.0.0.1", 8000)
|
|||
|
|
|
|||
|
|
if not port_available:
|
|||
|
|
print(f"[{get_time()}] 检测到8000端口异常，尝试启动应用进程...")
|
|||
|
|
success = start_application_process()
|
|||
|
|
|
|||
|
|
if success:
|
|||
|
|
# 启动后检查是否成功
|
|||
|
|
time.sleep(10) # 等待应用启动
|
|||
|
|
if check_port("127.0.0.1", 8000):
|
|||
|
|
print(f"[{get_time()}] 应用进程启动成功，8000端口已正常")
|
|||
|
|
# INSERT_YOUR_CODE
|
|||
|
|
# 检查并修改数据库字段
|
|||
|
|
try:
|
|||
|
|
|
|||
|
|
conn = mysql.connector.connect(
|
|||
|
|
host=MYSQL_HOST,
|
|||
|
|
user=MYSQL_USER,
|
|||
|
|
password=MYSQL_PASSWORD,
|
|||
|
|
database=MYSQL_DB
|
|||
|
|
)
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
local_ip = get_local_ip()
|
|||
|
|
sql = f"update model_ip set status = 0 where ip = '{local_ip}:8000';"
|
|||
|
|
print(f"[{get_time()}] 执行sql: {sql}")
|
|||
|
|
cursor.execute(sql)
|
|||
|
|
conn.commit()
|
|||
|
|
print(f"[{get_time()}] 数据库字段已成功修改")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[{get_time()}] 修改数据库字段失败: {str(e)}")
|
|||
|
|
finally:
|
|||
|
|
try:
|
|||
|
|
cursor.close()
|
|||
|
|
conn.close()
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
else:
|
|||
|
|
print(f"[{get_time()}] 应用进程启动后，8000端口仍未正常")
|
|||
|
|
else:
|
|||
|
|
print(f"[{get_time()}] 8000端口状态正常")
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
print(f"[{get_time()}] 启动Milvus监控服务")
|
|||
|
|
port_ok = check_port("127.0.0.1", 19530)
|
|||
|
|
if not port_ok:
|
|||
|
|
print("检测到Milvus服务异常，尝试重启...")
|
|||
|
|
restart_service()
|
|||
|
|
|
|||
|
|
print(f"[{get_time()}] 启动 8000 端口监控服务")
|
|||
|
|
# 开始监控8000端口，每60秒检查一次
|
|||
|
|
monitor_port_8000()
|
|||
|
|
|
|||
|
|
|