Increase the number of threads in migrate partitions tool

voldemort · Jan 12, 2011 · bfc773f · bfc773f
1 parent 5adf557
commit bfc773f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/java/voldemort/client/rebalance/MigratePartitions.java b/src/java/voldemort/client/rebalance/MigratePartitions.java
@@ -321,11 +321,12 @@ public static void main(String[] args) throws IOException {
         try {
             VoldemortConfig voldemortConfig = createTempVoldemortConfig();
             Cluster currentCluster = new ClusterMapper().readCluster(new BufferedReader(new FileReader(currentClusterFile)));
+            Cluster targetCluster = new ClusterMapper().readCluster(new BufferedReader(new FileReader(targetClusterFile)));
             adminClient = RebalanceUtils.createTempAdminClient(voldemortConfig,
                                                                currentCluster,
-                                                               1,
+                                                               targetCluster.getNumberOfNodes(),
                                                                1);
-            Cluster targetCluster = new ClusterMapper().readCluster(new BufferedReader(new FileReader(targetClusterFile)));
+
             List<StoreDefinition> currentStoreDefs = new StoreDefinitionsMapper().readStoreList(new BufferedReader(new FileReader(currentStoresFile)));
             List<StoreDefinition> targetStoreDefs = new StoreDefinitionsMapper().readStoreList(new BufferedReader(new FileReader(targetStoresFile)));