(shards): update makefile

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

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)