test commit

main.py

@@ -0,0 +1,112 @@
+#!/usr/bin/env python3
+"""
+메인 실행 파일
+- 백그라운드 스레드: 센서 데이터 측정
+- 메인 스레드(비동기): FastAPI + Socket.IO 서버 실행
+"""
+
+import asyncio
+import threading
+import time
+import rapimeasure
+import apiserver
+import uvicorn
+
+
+# 측정 데이터를 저장할 공유 dict 기본값
+shared_data = {"volume_ml": 0
+               , "time": 0
+               , "status": "Initializing"
+               }
+
+# 이벤트 루프 참조 (Socket.IO emit용)
+main_loop = None
+
+def measurement_loop():
+    """
+    백그라운드 스레드에서 실행되는 측정 무한 루프
+    센서 데이터를 읽고 Socket.IO로 브로드캐스트
+    """
+    global main_loop
+    print("[measure Thread] start")
+
+    # 센서 초기화
+    rapimeasure.init_hx711()
+    shared_data["status"] = "Measuring"
+
+    measurement_count = 0
+
+    while True:
+        try:
+            # 일시정지 체크
+            if rapimeasure.is_paused:
+                time.sleep(0.1)
+                continue
+
+            # 센서 데이터 읽기
+            _, volume_ml = rapimeasure.get_weight()
+
+            # 공유 데이터 업데이트
+            shared_data["volume_ml"] = round(volume_ml, 2)
+            shared_data["time"] = measurement_count
+
+            # Socket.IO로 실시간 브로드캐스트
+            if main_loop is not None:
+                try:
+                    # Thread-safe하게 비동기 emit 함수 호출
+                    asyncio.run_coroutine_threadsafe(
+                        apiserver.emit_weight_data(shared_data.copy()), main_loop
+                    )
+                except Exception:
+                    pass  # emit 실패 시 무시 (연결 종료 등)
+
+            measurement_count += 1
+
+            # 200ms 간격으로 측정 (5Hz)
+            time.sleep(0.2)
+
+        except Exception as e:
+            print(f"[measure error] {e}")
+            shared_data["status"] = f"error: {e}"
+            time.sleep(1)
+
+
+async def main():
+    """비동기 메인 함수"""
+    global main_loop
+    main_loop = asyncio.get_running_loop()
+
+    print("=" * 50)
+    print("  LoadCell Dashboard + Socket.IO")
+    print("  access address:")
+    print("    - local: http://localhost:8080")
+    print("    - network: http://192.168.0.191:8080")
+    print("  Swagger document: http://localhost:8080/docs")
+    print("=" * 50)
+
+    # 1. 공유 데이터를 API 서버에 연결
+    apiserver.set_shared_data(shared_data)
+
+    # 2. 측정 스레드 시작
+    measure_thread = threading.Thread(target=measurement_loop, daemon=True)
+    measure_thread.start()
+
+    print("[main] measure thread started")
+    print("[main] FastAPI + Socket.IO server starting...")
+
+    # 3. uvicorn 서버 실행
+    config = uvicorn.Config(
+        apiserver.app,
+        host="0.0.0.0",
+        port=8080,
+        log_level="warning",  # info -> warning으로 줄여서 로그 깔끔하게
+    )
+    server = uvicorn.Server(config)
+    await server.serve()
+
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("\nProgram terminated.")