update

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
+from  utils import request_post, 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(request_url=config_.TF_SERVING_URL, request_data=request_data, timeout=(0.1, 1))
+    result = request_post_data(config_.TF_SERVING_URL,request_data_dict)
 
     # print("result:", result)
     if result is None:

utils.py

@@ -45,6 +45,7 @@ def request_post(request_url, request_data, timeout):
     try:
         headers = {"Connection": "close"}
         print(request_url)
+        print(headers)
         response = requests.post(url=request_url, json=request_data, timeout=timeout, headers=headers)
         print("response:", response)
         if response.status_code == 200:
@@ -58,6 +59,31 @@ def request_post(request_url, request_data, timeout):
         return None
 
 
+def request_post_data(request_url, request_data):
+    """
+    post 请求 HTTP接口
+    :param request_url: 接口URL
+    :param request_data: 请求参数
+    :param timeout: 超时时间，单位为秒，type-float or tuple(connect_timeout, read_timeout)
+    :return: res_data json格式
+    """
+    try:
+        headers = {"Connection": "close"}
+        print(request_url)
+        print(headers)
+        response = requests.post(url=request_url, data=request_data)
+        print("response:", response)
+        if response.status_code == 200:
+            res_data = json.loads(response.text)
+            return res_data
+        else:
+            return None
+    except Exception as e:
+        print(e)
+        log_.error('url: {}, exception: {}, traceback: {}'.format(request_url, e, traceback.format_exc()))
+        return None
+
+
 def request_get(request_url, timeout):
     """
     get 请求 HTTP接口