59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
import socket
|
||
import subprocess
|
||
import time
|
||
from datetime import datetime
|
||
|
||
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
|
||
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", shell=True)
|
||
# 正确示例
|
||
# subprocess.run(["bash", "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 restart_zzbservice():
|
||
try:
|
||
subprocess.run("cd /root/pdf_parser/zzb_data_prod", shell=True)
|
||
subprocess.run("nohup python3 app.py > app.log 2>&1 &", shell=True)
|
||
print("zzb服务重启成功")
|
||
return True
|
||
except subprocess.CalledProcessError as e:
|
||
print(f"[{get_time()}] zzb服务重启失败: {str(e)}")
|
||
|
||
if __name__ == '__main__':
|
||
print(f"[{get_time()}] 启动Milvus监控服务")
|
||
port_ok = check_port("127.0.0.1", 19530)
|
||
if port_ok not in [0,True]:
|
||
print("检测到Milvus服务异常，尝试重启...")
|
||
restart_service()
|
||
|
||
print(f"[{get_time()}] 启动zzb监控服务")
|
||
port_ok = check_port("127.0.0.1", 8000)
|
||
|
||
if port_ok not in [0,True]:
|
||
print("检测到zzb服务异常，尝试重启...")
|
||
restart_zzbservice()
|
||
|
||
|