Method to run the random event generator

diff --git a/logic/src/main/scala/gr/grnet/aquarium/util/RandomEventGenerator.scala b/logic/src/main/scala/gr/grnet/aquarium/util/RandomEventGenerator.scala
index 294eb69..5d3316a 100644 (file)
--- a/logic/src/main/scala/gr/grnet/aquarium/util/RandomEventGenerator.scala
+++ b/logic/src/main/scala/gr/grnet/aquarium/util/RandomEventGenerator.scala
@@ -93,4 +93,19 @@ trait RandomEventGenerator extends AkkaAMQP {
         "event.%d.%s".format(event.cliendId, event.resource))
     }
   }
-}
\ No newline at end of file
+}
+
+object RandomEventGenerator extends RandomEventGenerator {
+
+  def main(args: Array[String]): Unit = {
+
+    var num = 100
+
+    if (args.size > 0)
+      num = args.head.toInt
+
+    println("Publishing %d messages. Hit Ctrl+c to stop".format(num))
+
+    genPublish(num)
+  }
+}