From 62ae0c645bdc53516e8c540f4a51381889bef829 Mon Sep 17 00:00:00 2001
From: xiajun <xiajun@xiaomi.com>
Date: Wed, 3 Dec 2014 16:48:19 +0800
Subject: [PATCH] fix bug: startSupervisors not work after stopSupervisors

Summary: Ref T2565

Test Plan: N/A

Reviewers: peng.zhang, yongxing, yehangjun

Subscribers: infra-dev

Maniphest Tasks: T2565

Differential Revision: https://phabricator.d.xiaomi.net/D8001
---
 .../com/yahoo/storm/yarn/StormMasterServerHandler.java | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/yahoo/storm/yarn/StormMasterServerHandler.java b/src/main/java/com/yahoo/storm/yarn/StormMasterServerHandler.java
index 37b07d6..3eae371 100644
--- a/src/main/java/com/yahoo/storm/yarn/StormMasterServerHandler.java
+++ b/src/main/java/com/yahoo/storm/yarn/StormMasterServerHandler.java
@@ -21,7 +21,9 @@
 import java.net.UnknownHostException;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.atomic.AtomicInteger;
 
+import backtype.storm.utils.Utils;
 import org.apache.thrift7.TException;
 import org.json.simple.JSONValue;
 import org.slf4j.Logger;
@@ -38,12 +40,14 @@ public class StormMasterServerHandler implements StormMaster.Iface {
     Map _storm_conf;
     StormAMRMClient _client;
     MasterServer _masterServer;
+    AtomicInteger curNumNeedSupervisors;
     
     StormMasterServerHandler(@SuppressWarnings("rawtypes") Map storm_conf, StormAMRMClient client) {
         _storm_conf = storm_conf;
         setStormHostConf();
         Util.rmNulls(_storm_conf);
         _client = client;
+        curNumNeedSupervisors = new AtomicInteger(0);
     }
 
     void init(MasterServer masterServer) {
@@ -102,6 +106,7 @@ public void setStormConf(String storm_conf) throws TException {
     @Override
     public void addSupervisors(int number) throws TException {
         LOG.info("adding "+number+" supervisors...");
+        curNumNeedSupervisors.addAndGet(number);
         _client.addSupervisors(number);
     }
 
@@ -215,7 +220,10 @@ public void stopUI() throws TException {
     @Override
     public void startSupervisors() throws TException {
         LOG.info("starting supervisors...");
-        _client.startAllSupervisors();
+        if (!_client.supervisorsAreToRun()) {
+            _client.addSupervisors(curNumNeedSupervisors.get());
+            _client.startAllSupervisors();
+        }
     }
 
     @Override