From 51a6e1fa160b94b0c2a1442c1a17fa9f6fb5d2ae Mon Sep 17 00:00:00 2001
From: Zhiyuan He <362583303@qq.com>
Date: Tue, 31 Mar 2020 10:10:40 +0800
Subject: [PATCH] Add a notice if user still wants to use yarn version (#4340)

* init

* fix

* fix
---
 .../paiLibrary/paiService/service_management_start.py     | 8 ++++++++
 tests/jenkins/stage_deploy.sh                             | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/deployment/paiLibrary/paiService/service_management_start.py b/deployment/paiLibrary/paiService/service_management_start.py
index 286ecf404..fdb5c95e9 100644
--- a/deployment/paiLibrary/paiService/service_management_start.py
+++ b/deployment/paiLibrary/paiService/service_management_start.py
@@ -44,6 +44,14 @@ class serivce_management_start:
             self.service_list = service_management_configuration.get_service_list(self.cluster_type)
         else:
             self.service_list = service_list
+        if self.cluster_type == 'yarn':
+            user_input = raw_input(
+                "Cluster type `yarn` is not well tested. We recommend you to use `k8s` version or " +
+                "stick to 0.14.0 if you prefer yarn version. " +
+                "If you still want to continue, please input Y. " +
+                "Other inputs will stop installation immediately: ")
+            if user_input != "Y":
+                sys.exit(1)
         self.logger.info("Get the service-list to manage : {0}".format(str(self.service_list)))
 
         self.retry_times = 5
diff --git a/tests/jenkins/stage_deploy.sh b/tests/jenkins/stage_deploy.sh
index 849ea8425..854b26e2f 100644
--- a/tests/jenkins/stage_deploy.sh
+++ b/tests/jenkins/stage_deploy.sh
@@ -49,5 +49,5 @@ echo "Y" | python paictl.py cluster k8s-bootup -p /cluster-configuration
 echo "pai" | python paictl.py config push -p /cluster-configuration
 
 # 3. start PAI services
-echo "pai" | python paictl.py service start
+echo -e "pai\nY\n" | python paictl.py service start
 EOF_DEV_BOX