From 056c636dad3be037bca1d7944f053073cec2209b Mon Sep 17 00:00:00 2001 From: wt Date: Mon, 1 Nov 2021 17:17:56 +0800 Subject: [PATCH] [skip ci] Add new func of benchmark (#11012) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/update.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/tests/benchmark/milvus_benchmark/update.py b/tests/benchmark/milvus_benchmark/update.py index a596e0cb4c..e551b0e830 100644 --- a/tests/benchmark/milvus_benchmark/update.py +++ b/tests/benchmark/milvus_benchmark/update.py @@ -21,6 +21,26 @@ def get_token(url): return token +def get_tags(url, token): + headers = {'Content-type': "application/json", + "charset": "UTF-8", + "Accept": "application/vnd.docker.distribution.manifest.v2+json", + "Authorization": "Bearer %s" % token} + try: + rep = requests.get(url, headers=headers) + data = json.loads(rep.text) + + tags = [] + if 'tags' in data: + tags = data["tags"] + else: + print("Can not get the tag list") + return tags + except: + print("Can not get the tag list") + return [] + + def parse_server_tag(server_tag): """ paser server tag from server config""" # tag format: "8c"/"8c16m"/"8c16m1g"