fix update number of bigtable nodes

spotify · Oct 23, 2016 · 282b725 · 282b725
1 parent 17ddfc2
commit 282b725
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/scio-bigtable/src/main/java/com/spotify/scio/bigtable/BigtableUtil.java b/scio-bigtable/src/main/java/com/spotify/scio/bigtable/BigtableUtil.java
@@ -54,8 +54,14 @@ public static void updateNumberOfBigtableNodes(final CloudBigtableConfiguration
                                              cloudBigtableConfiguration.getInstanceId());
 
     for (Cluster cluster : clusterUtilities.getClusters().getClustersList()) {
-      clusterUtilities.setClusterSize(cluster.getName(), cluster.getLocation(), numberOfNodes);
+      final String clusterId = extractAfterSlash(cluster.getName());
+      final String zoneId = extractAfterSlash(cluster.getLocation());
+      clusterUtilities.setClusterSize(clusterId, zoneId, numberOfNodes);
     }
     Thread.sleep(sleepDuration.getMillis());
   }
+
+  private static String extractAfterSlash(final String path) {
+    return path.substring(path.lastIndexOf("/") + 1);
+  }
 }