add timer

personas_vector/match_api.py

@@ -33,8 +33,8 @@ def match_tractate_by_device(device_id, n=10):
             D, I = FAISS_TAGS_INDEX.search(np.array([np.average(vectors, axis=0)]).astype("float32"), n)
             res = I.tolist()[0]
         time_end = time.time() - time_begin
-        # if time_end > 0.03:
-        send_performance_msg_to_dingtalk("match_tractate_by_device cost {}ms".format(time_end))
+        if time_end > 0.04:
+            send_performance_msg_to_dingtalk("match_tractate_by_device cost {:.5f}s".format(time_end))
         return res
     except Exception as e:
         send_msg_to_dingtalk(str(traceback.format_exc()))

word_vector/api.py

@@ -91,8 +91,8 @@ def clicked_tractate_ids_item2vec_model(id, n=5):
         time_begin = time.time()
         res = TRACTATE_CLICK_IDS_MODEL.wv.most_similar(id, topn=n)
         time_end = time.time() - time_begin
-        # if time_end > 0.03:
-        send_performance_msg_to_dingtalk("clicked_tractate_ids_item2vec_model cost {}ms".format(time_end))
+        if time_end > 0.04:
+            send_performance_msg_to_dingtalk("clicked_tractate_ids_item2vec_model cost {:.5f}s".format(time_end))
         return res
     except KeyError as e:
         send_msg_to_dingtalk("tractate_item2vec: " + str(e))