Write plan to a file inside the output dir

psemenyuk · Jun 20, 2013 · aa96bbf · aa96bbf
1 parent 9ee4ebe
commit aa96bbf
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/java/voldemort/client/rebalance/RebalancePlan.java b/src/java/voldemort/client/rebalance/RebalancePlan.java
@@ -126,6 +126,8 @@ public RebalancePlan(final Cluster currentCluster,
         zoneMoveMap = new MoveMap(interimCluster.getZoneIds());
 
         plan();
+        // write plan to file
+        RebalanceUtils.dumpPlanToFile(this.outputDir, this);
     }
 
     public RebalancePlan(final Cluster currentCluster,

diff --git a/src/java/voldemort/utils/RebalanceUtils.java b/src/java/voldemort/utils/RebalanceUtils.java
@@ -43,6 +43,7 @@
 import voldemort.client.protocol.admin.AdminClient;
 import voldemort.client.protocol.admin.AdminClientConfig;
 import voldemort.client.rebalance.RebalancePartitionsInfo;
+import voldemort.client.rebalance.RebalancePlan;
 import voldemort.cluster.Cluster;
 import voldemort.cluster.Node;
 import voldemort.cluster.Zone;
@@ -963,6 +964,26 @@ public static void dumpAnalysisToFile(String outputDirName,
         }
     }
 
+    /**
+     * Prints the plan to a file.
+     * 
+     * @param outputDirName
+     * @param plan
+     */
+    public static void dumpPlanToFile(String outputDirName, RebalancePlan plan) {
+        if(outputDirName != null) {
+            File outputDir = new File(outputDirName);
+            if(!outputDir.exists()) {
+                Utils.mkdirs(outputDir);
+            }
+            try {
+                FileUtils.writeStringToFile(new File(outputDirName, "plan.out"), plan.toString());
+            } catch(IOException e) {
+                logger.error("IOException during dumpPlanToFile: " + e);
+            }
+        }
+    }
+
     /**
      * Given a list of tuples of [replica_type, partition], flattens it and
      * generates a map of replica_type to partition mapping