update rank

rank_service.py

@@ -1,7 +1,7 @@
 #coding utf-8
 from db_helper import RedisHelper
 from config import set_config
-from  utils import request_post, request_post_data
+from  utils import  request_post_data
 import json
 from log import Log
 log_ = Log()
@@ -184,7 +184,7 @@ def get_tf_serving_sores(feature_dict):
     request_data= json.dumps(request_data_dict)
     print(request_data)
     # 调用http接口
-    result = request_post_data(config_.TF_SERVING_URL,request_data_dict)
+    result = request_post_data(config_.TF_SERVING_URL,request_data_dict, timeout=(0.1, 1))
 
     # print("result:", result)
     if result is None:

utils.py

@@ -59,7 +59,7 @@ def request_post(request_url, request_data, timeout):
         return None
 
 
-def request_post_data(request_url, request_data):
+def request_post_data(request_url, request_data, timeout):
     """
     post 请求 HTTP接口
     :param request_url: 接口URL
@@ -68,13 +68,14 @@ def request_post_data(request_url, request_data):
     :return: res_data json格式
     """
     try:
-        headers = {"Connection": "close"}
+        headers = {'content-type': 'application/json'}
         print(request_url)
         print(headers)
-        response = requests.post(url=request_url, data=request_data)
+        response = requests.post(url=request_url, data=request_data, timeout=timeout, headers=headers)
         print("response:", response)
         if response.status_code == 200:
             res_data = json.loads(response.text)
+            print(res_data)
             return res_data
         else:
             return None