[FLINK-2982][cli] Fix to show streaming plans via info option.

This closes apache#1334.
PangZhi · Nov 9, 2015 · 4462f53 · 4462f53
1 parent ed73f66
commit 4462f53
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/flink-clients/src/main/java/org/apache/flink/client/CliFrontend.java b/flink-clients/src/main/java/org/apache/flink/client/CliFrontend.java
@@ -63,6 +63,7 @@
 import org.apache.flink.optimizer.costs.DefaultCostEstimator;
 import org.apache.flink.optimizer.plan.FlinkPlan;
 import org.apache.flink.optimizer.plan.OptimizedPlan;
+import org.apache.flink.optimizer.plan.StreamingPlan;
 import org.apache.flink.optimizer.plandump.PlanJSONDumpGenerator;
 import org.apache.flink.runtime.akka.AkkaUtils;
 import org.apache.flink.runtime.client.JobStatusMessage;
@@ -415,8 +416,12 @@ protected int info(String[] args) {
 				this.optimizedPlan = flinkPlan;
 				this.packagedProgram = program;
 			} else {
-				String jsonPlan = new PlanJSONDumpGenerator()
-						.getOptimizerPlanAsJSON((OptimizedPlan) flinkPlan);
+				String jsonPlan = null;
+				if (flinkPlan instanceof OptimizedPlan) {
+					jsonPlan = new PlanJSONDumpGenerator().getOptimizerPlanAsJSON((OptimizedPlan) flinkPlan);
+				} else if (flinkPlan instanceof StreamingPlan) {
+					jsonPlan = ((StreamingPlan) flinkPlan).getStreamingPlanAsJSON();
+				}
 
 				if (jsonPlan != null) {
 					System.out.println("----------------------- Execution Plan -----------------------");