variable number of metrics

dclim · Jun 22, 2016 · f2ea46b · f2ea46b
1 parent 667b7ad
commit f2ea46b
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 7 deletions.
diff --git a/src/main/java/io/imply/jackhammer/EventGenerator.java b/src/main/java/io/imply/jackhammer/EventGenerator.java
@@ -26,17 +26,26 @@ public class EventGenerator
   private static final Random RND = new Random();
   private static final String TIME_FORMAT = "yyyy-MM-dd'T'HH:mm:ss'Z'";
 
-  public static String generate(int numLowCardDims, int numHighCardDims, int lowCardRange, int highCardRange)
+  public static String generate(
+      int numLowCardDims,
+      int numHighCardDims,
+      int lowCardRange,
+      int highCardRange,
+      int numMetrics
+  )
   {
     StringBuilder builder = new StringBuilder(150);
     builder.append(
         String.format(
-            "{\"timestamp\":\"%s\", \"value\":%d",
-            new DateTime(DateTimeZone.UTC).toString(TIME_FORMAT),
-            RND.nextInt(1000000)
+            "{\"timestamp\":\"%s\"",
+            new DateTime(DateTimeZone.UTC).toString(TIME_FORMAT)
         )
     );
 
+    for (int i = 0; i < numMetrics; i++) {
+      builder.append(String.format(", \"met%d\":%d", i, RND.nextInt(1000000)));
+    }
+
     for (int i = 0; i < numLowCardDims; i++) {
       builder.append(String.format(", \"dim%d\":\"%s\"", i, RND.nextInt(lowCardRange)));
     }

diff --git a/src/main/java/io/imply/jackhammer/Main.java b/src/main/java/io/imply/jackhammer/Main.java
@@ -63,6 +63,9 @@ public class Main
   @Option(name = {"-rh"}, description = "High cardinality range (default: 1000000)")
   public int highCardRange = 1000000;
 
+  @Option(name = {"-m"}, description = "Number of metric columns (default: 1)")
+  public int numMetrics = 1;
+
 
   public static void main(String[] args) throws Exception
   {
@@ -98,7 +101,8 @@ public void run() throws InterruptedException
         numLowCardDims,
         numHighCardDims,
         lowCardRange,
-        highCardRange
+        highCardRange,
+        numMetrics
     );
     runner.run();
 

diff --git a/src/main/java/io/imply/jackhammer/Runner.java b/src/main/java/io/imply/jackhammer/Runner.java
@@ -38,6 +38,7 @@ public class Runner
   private final int numHighCardDims;
   private final int lowCardRange;
   private final int highCardRange;
+  private final int numMetrics;
   private final ExecutorService executorService;
   private final ScheduledExecutorService scheduledExecutorService = Executors.newSingleThreadScheduledExecutor();
 
@@ -54,7 +55,8 @@ public Runner(
       int numLowCardDims,
       int numHighCardDims,
       int lowCardRange,
-      int highCardRange
+      int highCardRange,
+      int numMetrics
   )
   {
     this.writer = writer;
@@ -66,6 +68,7 @@ public Runner(
     this.numHighCardDims = numHighCardDims;
     this.lowCardRange = lowCardRange;
     this.highCardRange = highCardRange;
+    this.numMetrics = numMetrics;
   }
 
   public void run() throws InterruptedException
@@ -148,7 +151,7 @@ public void run()
           break;
         }
 
-        writer.write(EventGenerator.generate(numLowCardDims, numHighCardDims, lowCardRange, highCardRange));
+        writer.write(EventGenerator.generate(numLowCardDims, numHighCardDims, lowCardRange, highCardRange, numMetrics));
         counter++;
 
         if (rate > 0) {