Переглянути джерело

完善负载均衡器健康检测接口

locky 2 роки тому
батько
коміт
738002dba6
1 змінених файлів з 13 додано та 5 видалено
  1. 13 5
      Controller/InitController.py

+ 13 - 5
Controller/InitController.py

@@ -5,11 +5,13 @@ import logging
 
 from django.contrib.auth.hashers import check_password
 from django.db import transaction
+from django.http import HttpResponse
 from django.views import View
 
-from Model.models import Device_User, Device_Info, Order_Model, \
-    UidPushModel, UserExModel, UserOauth2Model, UidUserModel, GrantCodeModel, UserAppFrequencyModel, \
-    AppLogModel, UserFamily, FamilyMember, FamilyMemberJoin, SmartScene, DeviceSuperPassword
+from Model.models import Device_User, Device_Info, Order_Model, UidPushModel, UserExModel, UserOauth2Model, \
+    UidUserModel, GrantCodeModel, UserAppFrequencyModel, \
+    AppLogModel, UserFamily, FamilyMember, FamilyMemberJoin, SmartScene, DeviceSuperPassword, SceneLog
+from Object.RedisObject import RedisObject
 from Object.ResponseObject import ResponseObject
 
 
@@ -33,8 +35,14 @@ class InitView(View):
 
     @staticmethod
     def health_check(request_dict):
-        response = ResponseObject()
-        return response.json(0)
+        try:
+            redis_obj = RedisObject()
+            response = ResponseObject()
+            Device_Info.objects.filter().values('id').first()
+            SceneLog.objects.filter().values('id').first()
+            return response.json(0)
+        except Exception as e:
+            return HttpResponse(repr(e), status=500)
 
     @staticmethod
     def delete_account(request_dict):