Add some simple command line overrides for ZK and statsd hosts

Fixes #5

build.gradle

@@ -22,10 +22,6 @@ test {
         events "passed", "skipped", "failed", "standardOut", "standardError"
     }
 }
-
-run {
-    args zookeeperHosts
-}
 ////////////////////////////////////////////////////////////////////////////////
 
 

gradle.properties

@@ -1,4 +1,3 @@
 org.gradle.daemon=true
 bintrayUser=
 bintrayKey=
-zookeeperHosts=localhost:2181

src/main/groovy/com/github/lookout/verspaetung/Main.groovy

@@ -9,6 +9,7 @@ import groovy.transform.TypeChecked
 import com.timgroup.statsd.StatsDClient
 import com.timgroup.statsd.NonBlockingDogStatsDClient
 
+import org.apache.commons.cli.*
 import org.apache.curator.retry.ExponentialBackoffRetry
 import org.apache.curator.framework.CuratorFrameworkFactory
 import org.apache.curator.framework.CuratorFramework
@@ -16,18 +17,41 @@ import org.apache.curator.framework.recipes.cache.TreeCache
 import org.slf4j.Logger
 import org.slf4j.LoggerFactory
 
-//@TypeChecked
 class Main {
-    private static final StatsDClient statsd = new NonBlockingDogStatsDClient('verspaetung', '10.32.2.211', 8125)
-    private static final Logger logger = LoggerFactory.getLogger(Main.class)
+    private static final String METRICS_PREFIX = 'verspaetung'
+
+    private static StatsDClient statsd
+    private static Logger logger
 
     static void main(String[] args) {
+        String zookeeperHosts = 'localhost:2181'
+        String statsdHost = 'localhost'
+        Integer statsdPort = 8125
+
+        CommandLine cli = parseCommandLine(args)
+
+        if (cli.hasOption('z')) {
+            zookeeperHosts = cli.getOptionValue('z')
+        }
+
+        if (cli.hasOption('H')) {
+            statsdHost = cli.getOptionValue('H')
+        }
+
+        if (cli.hasOption('p')) {
+            statsdPort = cli.getOptionValue('p')
+        }
+
+        logger = LoggerFactory.getLogger(Main.class)
         logger.info("Running with: ${args}")
+        logger.warn("Using: zookeepers=${zookeeperHosts} statsd=${statsdHost}:${statsdPort}")
 
         ExponentialBackoffRetry retry = new ExponentialBackoffRetry(1000, 3)
-        CuratorFramework client = CuratorFrameworkFactory.newClient(args[0], retry)
+        CuratorFramework client = CuratorFrameworkFactory.newClient(zookeeperHosts, retry)
         ConcurrentHashMap<TopicPartition, List<zk.ConsumerOffset>> consumers = new ConcurrentHashMap()
 
+        statsd = new NonBlockingDogStatsDClient(METRICS_PREFIX, statsdHost, statsdPort)
+
         client.start()
 
         TreeCache cache = new TreeCache(client, '/consumers')
@@ -66,4 +90,50 @@ class Main {
         poller.join()
         return
     }
+
+    static Options createCLI() {
+        Options options = new Options()
+
+        Option zookeeper = OptionBuilder.withArgName('HOSTS')
+                                        .hasArg()
+                                        .withDescription('Comma separated list of Zookeeper hosts (e.g. localhost:2181)')
+                                        .withLongOpt('zookeeper')
+                                        .withValueSeparator(',' as char)
+                                        .create('z')
+
+        Option statsd_host = OptionBuilder.withArgName('STATSD')
+                                        .hasArg()
+                                        .withType(String)
+                                        .withDescription('Hostname for a statsd instance (defaults to localhost)')
+                                        .withLongOpt('statsd-host')
+                                        .create('H')
+
+        Option statsd_port = OptionBuilder.withArgName('PORT')
+                                          .hasArg()
+                                          .withType(Integer)
+                                          .withDescription('Port for the statsd instance (defaults to 8125)')
+                                          .withLongOpt('statsd-port')
+                                          .create('p')
+
+        options.addOption(zookeeper)
+        options.addOption(statsd_host)
+        options.addOption(statsd_port)
+
+        return options
+    }
+
+    static CommandLine parseCommandLine(String[] args) {
+        Options options = createCLI()
+        PosixParser parser = new PosixParser()
+
+        try {
+            return parser.parse(options, args)
+        }
+        catch (MissingOptionException|UnrecognizedOptionException ex) {
+            HelpFormatter formatter = new HelpFormatter()
+            println ex.message
+            formatter.printHelp('verspaetung', options)
+            System.exit(1)
+        }
+    }
 }