init

.env.example

@@ -0,0 +1,5 @@
+DB_HOST=127.0.0.1
+DB_PORT=5432
+DB_DATABASE=postgres
+DB_USERNAME=postgres
+DB_PASSWORD=password

.gitignore

@@ -0,0 +1,2 @@
+.env
+healthbeat.log

Healthbeat.py

@@ -0,0 +1,66 @@
+# coding=utf-8
+import os
+import cv2
+import logging 
+import pymysql
+import threading
+import json
+from dotenv import load_dotenv
+
+class Camera:
+    def __init__(self, id, rtsp, status):
+        self.id = id
+        self.rtsp = rtsp
+        self.status = status
+
+    def healthbeat(self):
+        cap = cv2.VideoCapture(self.rtsp)
+        ret, _ = cap.read()
+        cap.release()
+
+        # 摄像头状态: 1 在线, 2 离线
+        status = 1 if ret == True else 2
+
+        if status == 1:
+            logging.info("摄像头ID: %s, 状态: 在线" % (self.id))
+        else:
+            logging.warning("摄像头ID: %s, 状态: 离线" % (self.id)) 
+
+        if (status == 1 and self.status == 2) or (status == 2 and self.status == 1):
+            conn = get_connection()
+
+            with conn:
+                with conn.cursor() as cursor:
+                    cursor.execute("update devices set state = %s where id = %s", (status, self.id))
+
+def init_logger():
+    logging.basicConfig(filename='healthbeat.log', level=logging.INFO, format='[%(asctime)s] %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
+
+def get_connection():
+    return pymysql.connect(
+        host=os.environ.get("DB_HOST"),
+        port=int(os.environ.get("DB_PORT")),
+        user=os.environ.get("DB_USERNAME"),
+        password=os.environ.get("DB_PASSWORD"),
+        database=os.environ.get("DB_DATABASE"),
+        charset='utf8mb4'
+    )
+
+if __name__ == '__main__':
+    load_dotenv()
+    # 初始化日志格式
+    init_logger()
+
+    conn = get_connection()
+    
+    with conn:
+        with conn.cursor() as cursor:
+            cursor.execute("SELECT id, state, extends FROM devices WHERE type = 1 AND state in (1, 2) ORDER BY id ASC")
+            rows = cursor.fetchall()
+
+            for row in rows:
+                extends = json.loads(row[2])
+                c = Camera(row[0], extends["rtsp_url"], row[1])
+                t = threading.Thread(target=c.healthbeat)
+                t.start()
+    

requirements.txt

@@ -0,0 +1,4 @@
+numpy==1.23.5
+opencv-python==4.6.0.66
+PyMySQL==1.1.0
+python-dotenv==0.21.0