BLU:168 added validation on setting cluster name from shell
Project: http://git-wip-us.apache.org/repos/asf/incubator-blur/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-blur/commit/c7bb4ce6 Tree: http://git-wip-us.apache.org/repos/asf/incubator-blur/tree/c7bb4ce6 Diff: http://git-wip-us.apache.org/repos/asf/incubator-blur/diff/c7bb4ce6 Branch: refs/heads/0.2.0-newtypesystem Commit: c7bb4ce6e245d753df013279d79508574bf09f0d Parents: 823b0d8 Author: Gagan <[email protected]> Authored: Mon Jul 22 16:24:09 2013 +0530 Committer: Gagan <[email protected]> Committed: Mon Jul 22 16:24:09 2013 +0530 ---------------------------------------------------------------------- .../main/java/org/apache/blur/shell/Main.java | 21 +++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-blur/blob/c7bb4ce6/blur-shell/src/main/java/org/apache/blur/shell/Main.java ---------------------------------------------------------------------- diff --git a/blur-shell/src/main/java/org/apache/blur/shell/Main.java b/blur-shell/src/main/java/org/apache/blur/shell/Main.java index 7b53552..16c887e 100644 --- a/blur-shell/src/main/java/org/apache/blur/shell/Main.java +++ b/blur-shell/src/main/java/org/apache/blur/shell/Main.java @@ -133,11 +133,26 @@ public class Main { if (args.length != 2) { throw new CommandException("Invalid args: " + help()); } - cluster = args[1]; - out.println("cluster is now " + cluster); + String clusterNamePassed = args[1]; + if(validateClusterName(client, clusterNamePassed)) { + cluster = clusterNamePassed; + out.println("cluster is now " + cluster); + }else{ + out.println("[ " + clusterNamePassed + " ]"+" is not a valid cluster name."); + } } - @Override + private boolean validateClusterName(Iface client, String clusterName) throws BlurException, TException { + List<String> clusterNamesList = client.shardClusterList(); + if(clusterNamesList != null && !clusterNamesList.isEmpty()){ + if(clusterNamesList.contains(clusterName)){ + return true; + } + } + return false; + } + + @Override public String help() { return "set the cluster in use, args; clustername"; }
