From 80c57df90d56f29aeb5867ce84de4d9c2f39f825 Mon Sep 17 00:00:00 2001
From: wt <ting.wang@zilliz.com>
Date: Fri, 29 Oct 2021 12:10:55 +0800
Subject: [PATCH] [skip ci] Add new func of benchmark (#10890)

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
---
 tests/benchmark/milvus_benchmark/update.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/tests/benchmark/milvus_benchmark/update.py b/tests/benchmark/milvus_benchmark/update.py
index c060105c3f..a596e0cb4c 100644
--- 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"