Redis + Websocket

app.py

@@ -24,8 +24,16 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Initialize Redis client
-redis_client = redis.Redis(host='localhost', port=6379, db=0)
+# Sửa phần khởi tạo Redis client và thêm biến để kiểm tra Redis status
+redis_available = False
+try:
+    redis_client = redis.Redis(host='localhost', port=6379, db=0)
+    redis_client.ping()  # Kiểm tra kết nối
+    redis_available = True
+except redis.ConnectionError:
+    print("Redis không khả dụng - hệ thống sẽ chạy không có cache")
+    redis_client = None
+
 CACHE_EXPIRE_TIME = 300  # 5 minutes
 
 # MongoDB connection
@@ -127,25 +135,28 @@ async def vote_by_voter(vote_request: VoteRequest):
     if not voter:
         raise HTTPException(status_code=404, detail="Voter not found")
 
-    # Kiểm tra số vote còn lại
-    remaining_votes = voter['number_of_votes']  # Số vote còn lại
+    remaining_votes = voter['number_of_votes']
     if vote_request.vote_count > remaining_votes:
         raise HTTPException(status_code=400, detail=f"Not enough votes remaining. You have {remaining_votes} votes left")
 
     try:
-        # Cập nhật MongoDB - giảm số vote còn lại
+        # Cập nhật MongoDB
         voter_collection.update_one(
             {'_id': ObjectId(vote_request.voter_id)}, 
-            {'$inc': {'number_of_votes': -vote_request.vote_count}}  # Giảm số vote còn lại
+            {'$inc': {'number_of_votes': -vote_request.vote_count}}
         )
         pdf_collection.update_one(
             {'_id': ObjectId(vote_request.file_id)}, 
             {'$inc': {'votes': vote_request.vote_count}}
         )
         
-        # Clear cache
-        redis_client.delete('all_files')
-        redis_client.delete(f'voter:{vote_request.voter_id}')
+        # Chỉ xóa cache nếu Redis khả dụng
+        if redis_available:
+            try:
+                redis_client.delete('all_files')
+                redis_client.delete(f'voter:{vote_request.voter_id}')
+            except redis.RedisError:
+                print("Không thể xóa cache Redis")
         
         # Emit socket event
         updated_file = pdf_collection.find_one({'_id': ObjectId(vote_request.file_id)})
@@ -161,14 +172,14 @@ async def vote_by_voter(vote_request: VoteRequest):
 @app.get("/get-voter")
 async def get_voter(id: str):
     try:
-        # Check cache
-        try:
-            cached_voter = redis_client.get(f'voter:{id}')
-            if cached_voter:
-                return json.loads(cached_voter)
-        except RedisError as redis_err:
-            print(f"Redis error: {redis_err}")
-            pass
+        # Chỉ check cache nếu Redis khả dụng
+        if redis_available:
+            try:
+                cached_voter = redis_client.get(f'voter:{id}')
+                if cached_voter:
+                    return json.loads(cached_voter)
+            except redis.RedisError:
+                print("Không thể đọc cache Redis")
 
         voter = voter_collection.find_one({'_id': ObjectId(id)})
         if not voter:
@@ -179,14 +190,15 @@ async def get_voter(id: str):
             'name': voter['name'],
             'group': voter['group'],
             'role': voter['role'],
-            'number_of_votes': voter['number_of_votes']  # Số này giờ là số vote còn lại
+            'number_of_votes': voter['number_of_votes']
         }
         
-        try:
-            redis_client.setex(f'voter:{id}', CACHE_EXPIRE_TIME, json.dumps(voter_data))
-        except RedisError as redis_err:
-            print(f"Redis caching error: {redis_err}")
-            pass
+        # Chỉ lưu cache nếu Redis khả dụng
+        if redis_available:
+            try:
+                redis_client.setex(f'voter:{id}', CACHE_EXPIRE_TIME, json.dumps(voter_data))
+            except redis.RedisError:
+                print("Không thể lưu cache Redis")
 
         return voter_data
     except Exception as e:
@@ -209,8 +221,9 @@ async def index():
 async def shutdown_event():
     # Close MongoDB connection
     client.close()
-    # Close Redis connection
-    redis_client.close()
+    # Close Redis connection if available
+    if redis_available and redis_client:
+        redis_client.close()
 
 if __name__ == "__main__":
     import uvicorn