From 58a31cfe22341d28b4e7d183291683b034e7156b Mon Sep 17 00:00:00 2001
From: "peng.xu" <peng.xu@zilliz.com>
Date: Mon, 4 Nov 2019 11:15:07 +0800
Subject: [PATCH] (shards): update makefile

---
 shards/Makefile                 | 18 +++++++++++++++++-
 shards/all_in_one/probe_test.py | 25 +++++++++++++++++++++++++
 2 files changed, 42 insertions(+), 1 deletion(-)
 create mode 100644 shards/all_in_one/probe_test.py

diff --git a/shards/Makefile b/shards/Makefile
index 7ad724ec4c..b1cdecdce2 100644
--- a/shards/Makefile
+++ b/shards/Makefile
@@ -1,13 +1,29 @@
+HOST=$(or $(host),127.0.0.1)
+PORT=$(or $(port),19530)
+
 build:
 	docker build --network=host -t milvusdb/mishards .
 push:
 	docker push milvusdb/mishards
 pull:
 	docker pull milvusdb/mishards
-deploy:
+deploy: clean_deploy
 	cd all_in_one && docker-compose -f all_in_one.yml up -d && cd -
 clean_deploy:
 	cd all_in_one && docker-compose -f all_in_one.yml down && cd -
+probe_deploy:
+	docker run --rm --name probe --net=host milvusdb/mishards /bin/bash -c "python all_in_one/probe_test.py"
+cluster:
+	cd kubernetes_demo;./start.sh baseup;./start.sh appup;cd -
+clean_cluster:
+	cd kubernetes_demo;./start.sh cleanup;cd -
+cluster_status:
+	kubectl get pods -n milvus -o wide
+probe_cluster:
+	@echo
+	$(shell kubectl get service -n milvus | grep milvus-proxy-servers | awk {'print $$4,$$5'} | awk -F"[: ]" {'print "docker run --rm --name probe --net=host milvusdb/mishards /bin/bash -c \"python all_in_one/probe_test.py --port="$$2" --host="$$1"\""'})
+probe:
+	docker run --rm --name probe --net=host milvusdb/mishards /bin/bash -c "python all_in_one/probe_test.py --port=${PORT} --host=${HOST}"
 clean_coverage:
 	rm -rf cov_html
 clean: clean_coverage clean_deploy
diff --git a/shards/all_in_one/probe_test.py b/shards/all_in_one/probe_test.py
new file mode 100644
index 0000000000..6250465910
--- /dev/null
+++ b/shards/all_in_one/probe_test.py
@@ -0,0 +1,25 @@
+from milvus import Milvus
+
+RED = '\033[0;31m'
+GREEN = '\033[0;32m'
+ENDC = ''
+
+
+def test(host='127.0.0.1', port=19531):
+    client = Milvus()
+    try:
+        status = client.connect(host=host, port=port)
+        if status.OK():
+            print('{}Pass: Connected{}'.format(GREEN, ENDC))
+            return 0
+        else:
+            print('{}Error: {}{}'.format(RED, status, ENDC))
+            return 1
+    except Exception as exc:
+        print('{}Error: {}{}'.format(RED, exc, ENDC))
+        return 1
+
+
+if __name__ == '__main__':
+    import fire
+    fire.Fire(test)