add log

ad_predict.py

@@ -45,11 +45,10 @@ def thompson_process(creative_id):
     return thompson_res
 
 
-def get_creative_id_with_thompson(mid, creative_id_list):
+def get_creative_id_with_thompson(mid, creative_id_list, gevent_pool):
     """利用Thompson采样获取此次要展示的广告创意ID"""
     # 限制协程最大并发数：20
     st_time6 = time.time()
-    gevent_pool = pool.Pool(100)
     gevent_pool_time = int(time.time() - st_time6) * 1000
     st_time5 = time.time()
     tasks = [gevent_pool.spawn(thompson_process, creative_id) for creative_id in creative_id_list]

app.py

@@ -5,7 +5,7 @@ import json
 import time
 import traceback
 import ast
-from gevent import monkey
+from gevent import monkey, pool
 monkey.patch_all()
 
 from flask import Flask, request
@@ -16,6 +16,7 @@ from ad_predict import get_creative_id_with_thompson
 app = Flask(__name__)
 log_ = Log()
 config_ = set_config()
+gevent_pool = pool.Pool(100)
 
 # log_.info(f"server start...")
 
@@ -32,7 +33,7 @@ def get_creative_id():
         request_data = json.loads(request.get_data())
         mid = request_data.get('mid')
         creative_id_list = request_data.get('creativeIdList')
-        thompson_result = get_creative_id_with_thompson(mid=mid, creative_id_list=creative_id_list)
+        thompson_result = get_creative_id_with_thompson(mid=mid, creative_id_list=creative_id_list, gevent_pool=gevent_pool)
         result = {'code': 200, 'message': 'success', 'data': {'mid': mid, 'creativeId': thompson_result['creative_id']}}
         log_message = {
             'requestUri': '/ad/predict/getCreativeId',