[skip ci] Add new func of benchmark (#10890)

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
2021-10-29 12:10:55 +08:00 · 2021-10-29 12:10:55 +08:00 · 80c57df90d
parent 165be0445d
commit 80c57df90d
1 changed files with 11 additions and 0 deletions
--- a/tests/benchmark/milvus_benchmark/update.py
+++ b/tests/benchmark/milvus_benchmark/update.py
@ -10,6 +10,17 @@ import requests
 import json


+def get_token(url):
+    rep = requests.get(url)
+    data = json.loads(rep.text)
+    if 'token' in data:
+        token = data['token']
+    else:
+        token = ''
+        print("Can not get token.")
+    return token
+
+
 def parse_server_tag(server_tag):
    """ paser server tag from server config"""
    # tag format: "8c"/"8c16m"/"8c16m1g"