1

templates/crond_udp.py.tpl

@@ -0,0 +1,85 @@
+#!/usr/bin/env python3
+# -*- coding: UTF-8 -*-
+# Auto-generated from template. DO NOT EDIT DIRECTLY!
+# Device: {device_name} ({device_id}) - UDP Protocol
+
+import time
+import logging
+import os
+import socket
+import sys
+import subprocess
+
+# ====== 配置参数（从配置文件读取）======
+TARGET_IP = '{tms_server_ip}'
+TARGET_PORT = {listen_port}
+LOG_FILE = '{base_dir}/log/crond-{device_id}.log'
+RESTART_COMMAND = "{python_path} {base_dir}/bin/{device_id}.py &"
+DEVICE_ID = '{device_id}'
+# =========================
+
+ISOTIMEFORMAT = '%Y-%m-%d %X'
+
+logging.basicConfig(filename=LOG_FILE, filemode="a", level=logging.DEBUG,
+                    format='%(asctime)s - %(levelname)s - %(message)s', datefmt=ISOTIMEFORMAT)
+
+
+def is_udp_port_open(ip, port):
+    """
+    检查UDP端口是否可用。
+    UDP是无连接的，这里通过检查端口是否被占用或发送测试数据来判断。
+    """
+    try:
+        # 尝试绑定到该端口，如果成功说明端口未被占用（服务未启动）
+        # 如果失败说明端口已被占用（服务正在运行）
+        test_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        test_sock.settimeout(1)
+        test_sock.bind((ip, int(port)))
+        test_sock.close()
+        
+        # 如果绑定成功，说明服务未启动
+        print(f"[{{DEVICE_ID}}] UDP端口 {{port}} 未被占用，服务可能未运行")
+        logging.warning(f"UDP端口 {{port}} 未被占用，服务可能未运行，尝试重启程序")
+        logging.info(f"执行重启命令: {{RESTART_COMMAND}}")
+        os.system(RESTART_COMMAND)
+        return False
+    except OSError as e:
+        # 绑定失败，说明端口已被占用，服务正在运行
+        if "Address already in use" in str(e) or "98" in str(e):
+            print(f"[{{DEVICE_ID}}] UDP端口 {{port}} is open (已被占用，服务运行中)")
+            logging.info(f"UDP端口 {{port}} is open (服务运行中)")
+            return True
+        else:
+            print(f"[{{DEVICE_ID}}] 检查UDP端口 {{port}} 时发生错误: {{e}}")
+            logging.error(f"检查UDP端口 {{port}} 时发生错误: {{e}}")
+            return False
+
+
+def check_process_running():
+    """通过进程名检查服务是否正在运行"""
+    try:
+        # 检查是否有对应设备的Python进程在运行
+        cmd = f"ps aux | grep '{{DEVICE_ID}}.py' | grep -v grep"
+        result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
+        if result.returncode == 0 and result.stdout.strip():
+            print(f"[{{DEVICE_ID}}] 进程检查: 服务正在运行")
+            logging.info(f"进程检查: 服务正在运行")
+            return True
+        else:
+            print(f"[{{DEVICE_ID}}] 进程检查: 服务未运行")
+            logging.warning(f"进程检查: 服务未运行，尝试重启")
+            logging.info(f"执行重启命令: {{RESTART_COMMAND}}")
+            os.system(RESTART_COMMAND)
+            return False
+    except Exception as e:
+        print(f"[{{DEVICE_ID}}] 进程检查失败: {{e}}")
+        logging.error(f"进程检查失败: {{e}}")
+        return False
+
+
+if __name__ == '__main__':
+    logging.info(f"[{{DEVICE_ID}}] 启动UDP端口检查程序，检查 {{TARGET_IP}}:{{TARGET_PORT}}")
+    print(f"[{{DEVICE_ID}}] 启动UDP端口检查程序，检查 {{TARGET_IP}}:{{TARGET_PORT}}")
+    
+    # 对于UDP，优先使用进程检查方式
+    check_process_running()