Fix #1459, Fix stop cmdlet error (#1460)

Intel-bigdata · Dec 1, 2017 · 2cc104e · 2cc104e
1 parent 6c01665
commit 2cc104e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/smart-engine/src/main/java/org/smartdata/server/engine/cmdlet/agent/AgentMaster.java b/smart-engine/src/main/java/org/smartdata/server/engine/cmdlet/agent/AgentMaster.java
@@ -278,8 +278,12 @@ private boolean handleClientMessage(Object message) {
         return true;
       } else if (message instanceof StopCmdlet) {
         long cmdletId = ((StopCmdlet) message).getCmdletId();
-        dispatches.get(cmdletId).tell(message, getSelf());
-        getSender().tell("Succeed", getSelf());
+        if (dispatches.containsKey(cmdletId)) {
+          dispatches.get(cmdletId).tell(message, getSelf());
+          getSender().tell("Succeed", getSelf());
+        } else {
+          getSender().tell("NotFound", getSelf());
+        }
         return true;
       } else {
         return false;