msg-generator: sending a limited number of messages

Signed-off-by: Terez Nemes <terez.noble@gmail.com>
syslog-ng · Feb 22, 2019 · 290eafc · 290eafc
1 parent e771c6b
commit 290eafc
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 1 deletion.
diff --git a/modules/examples/sources/msg-generator/msg-generator-grammar.ym b/modules/examples/sources/msg-generator/msg-generator-grammar.ym
@@ -58,6 +58,7 @@ MsgGeneratorSourceOptions *last_msg_generator_source_options;
 
 %token KW_MSG_GENERATOR
 %token KW_FREQ
+%token KW_NUM
 %token KW_TEMPLATE
 
 %type <ptr> source_msg_generator
@@ -92,6 +93,10 @@ source_msg_generator_option
     {
       msg_generator_source_options_set_freq(last_msg_generator_source_options, $3);
     }
+  | KW_NUM '(' nonnegative_integer ')'
+    {
+      msg_generator_source_options_set_num(last_msg_generator_source_options, $3);
+    }
   | KW_TEMPLATE '(' template_content ')'
     {
       msg_generator_source_options_set_template(last_msg_generator_source_options, $3);

diff --git a/modules/examples/sources/msg-generator/msg-generator-parser.c b/modules/examples/sources/msg-generator/msg-generator-parser.c
@@ -33,6 +33,7 @@ static CfgLexerKeyword msg_generator_keywords[] =
 {
   { "example_msg_generator", KW_MSG_GENERATOR },
   { "freq", KW_FREQ },
+  { "num", KW_NUM },
   { "template", KW_TEMPLATE },
   { NULL }
 };

diff --git a/modules/examples/sources/msg-generator/msg-generator-source-options.h b/modules/examples/sources/msg-generator/msg-generator-source-options.h
@@ -31,6 +31,7 @@ typedef struct
 {
   LogSourceOptions super;
   guint freq;
+  gint max_num;
   LogTemplate *template;
 } MsgGeneratorSourceOptions;
 
@@ -60,6 +61,12 @@ msg_generator_source_options_set_freq(MsgGeneratorSourceOptions *self, gdouble f
   self->freq = (guint) (freq * 1000);
 }
 
+static inline void
+msg_generator_source_options_set_num(MsgGeneratorSourceOptions *self, gint max_num)
+{
+  self->max_num = max_num;
+}
+
 static inline void
 msg_generator_source_options_set_template(MsgGeneratorSourceOptions *self, LogTemplate *template)
 {

diff --git a/modules/examples/sources/msg-generator/msg-generator-source.c b/modules/examples/sources/msg-generator/msg-generator-source.c
@@ -36,6 +36,7 @@ struct MsgGeneratorSource
   LogSource super;
   MsgGeneratorSourceOptions *options;
   struct iv_timer timer;
+  gint num;
 };
 
 static void
@@ -120,7 +121,18 @@ _timer_expired(void *cookie)
 
   _send_generated_message(self);
 
-  _start_timer(self);
+  if (self->options->max_num > 0)
+    {
+      self->num++;
+      if (self->num < self->options->max_num)
+        {
+          _start_timer(self);
+        }
+    }
+  else
+    {
+      _start_timer(self);
+    }
 }
 
 gboolean