logstash-input-faker 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 161f82cb48552d214e675d4abc8c10630f0b449f
-  data.tar.gz: b52dda64bd66647284769b5ffc91a33b4bad76e1
+  metadata.gz: 3b6abd516ff7f97b9ef94f1fd8d50ba02b63d739
+  data.tar.gz: 999846e5ad3c765fd3f09156f013ab133062cfdf
 SHA512:
-  metadata.gz: 330785d305642a000971147d623139bc9f6604fd0556e651b3b32b29ba22923edfbe0a97ec24671d19fc2951613359370cfe00063c5fb991185b4acc648b00e5
-  data.tar.gz: f3484dbd524e0ddbfcf8f556023d873d5172e0e6e79b5662c2044de573bfc8b673c81e5fd5637729c01d71f2e1a39373933d4d050b697491b79f17c5e3103c48
+  metadata.gz: 66432796a801cec3072c82e7554095fb3c2053d3619ecba20855dfaa7dbe99ee9c7b0b8ac269bf1e7a0b2198ba52b1a444404241216d3c6234a5875497ca7034
+  data.tar.gz: dc5794b860d8ee70694e7da1d84cb247044dfdef383e289633a1109ed1510052f638642301494edf798f35d2179a97f8bb87c13c9917d1a24601da80d91007ac

data/lib/logstash/inputs/faker.rb

@@ -46,6 +46,10 @@ class LogStash::Inputs::Faker < LogStash::Inputs::Base
 
   config :add_foreign_keys_to_events, :validate => :boolean, :default => false
 
+  config :multiply_field, :validate => :hash, :default => {}
+
+  config :multiply_splitable_field, :validate => :hash, :default => {}
+
   # Similar to LogStash::Inputs::Base.add_field define a hash
   # where the value is a Faker module and method call
   # ex: Name.first_name
@@ -91,13 +95,13 @@ class LogStash::Inputs::Faker < LogStash::Inputs::Base
     set_foreign_keys  if @add_foreign_keys_to_events
     while !stop? && (@count <= 0 || number < @count)
       event = LogStash::Event.new({})
-      if @add_primary_key_to_events
-        LogStash::Util::Decorators.add_fields(@primary_key, event,"inputs/#{self.class.name}")
-      end
       add_faker_fields(event)
       if @splitable_field
         add_splitable_fields(event)
       end
+      if @add_primary_key_to_events
+        LogStash::Util::Decorators.add_fields(@primary_key, event,"inputs/#{self.class.name}")
+      end
       decorate(event)
       event.set("host", @host)
       queue << event
@@ -105,12 +109,27 @@ class LogStash::Inputs::Faker < LogStash::Inputs::Base
     end #end loop
   end # end run
 
+  protected
+  def generate_multiplied_values(value, count, is_faker=false)
+    num = (count == 0 ? rand(1000) : count)
+    if is_faker
+      return num.times.map{ Faker.class_eval(value) }
+    else
+      return num.times.map{ value }
+    end
+  end
+
   protected
   def add_faker_fields(event)
     new_fields = {}
     @add_faker_field.each do |field, faker_string|
-      event.remove(field) if @overwrite_fields
-      new_fields[field] = Faker.class_eval(faker_string)
+      if @multiply_field[field]
+        event.remove(field) if @overwrite_fields
+        new_fields[field] = generate_multiplied_values(faker_string, @multiply_field[field], true)
+      else
+        event.remove(field) if @overwrite_fields
+        new_fields[field] = Faker.class_eval(faker_string)
+      end
     end
     LogStash::Util::Decorators.add_fields(new_fields, event,"inputs/#{self.class.name}")
   end
@@ -125,11 +144,19 @@ class LogStash::Inputs::Faker < LogStash::Inputs::Base
     end
     @splitable_field_count.times do
       new_event = LogStash::Event.new()
-      @add_splitable_faker_field.each do |field, faker_string|
-        new_event.set(field, Faker.class_eval(event.sprintf(faker_string)))
-      end
       @add_splitable_field.each do |field, value|
-        new_event.set(field, event.sprintf(value))
+        if @multiply_splitable_field[field]
+          new_event.set(field, generate_multiplied_values(faker_string, @multiply_splitable_field[field], false))
+        else
+          new_event.set(field, event.sprintf(value))
+        end
+      end
+      @add_splitable_faker_field.each do |field, faker_string|
+        if @multiply_splitable_field[field]
+          new_event.set(field, generate_multiplied_values(faker_string, @multiply_splitable_field[field], true))
+        else
+          new_event.set(field, event.sprintf(Faker.class_eval(faker_string)))
+        end
       end
       if @add_foreign_keys_to_events
         key = (@foreign_keys - used_ids).sample

data/logstash-input-faker.gemspec

@@ -1,7 +1,7 @@
 Gem::Specification.new do |s|
 
   s.name            = 'logstash-input-faker'
-  s.version         = '0.3.0'
+  s.version         = '0.4.0'
   s.licenses        = ['Apache License (2.0)']
   s.summary         = "Generate log events with fake data"
   s.description     = "This gem is a Logstash plugin required to be installed on top of the Logstash core pipeline using $LS_HOME/bin/logstash-plugin install gemname. This gem is not a stand-alone program"

data/spec/inputs/faker_spec.rb

@@ -19,8 +19,10 @@ describe LogStash::Inputs::Faker do
                                 "test" => "field"
                               }
     plg.add_splitable_faker_field = {
-                                      "[name][testing]" => "Internet.user_name('%{first_name} %{last_name}', ['_'])"
+                                      "[name][testing]" => "Internet.user_name()",
+                                      "[follower_id]" => "Number.number(10)"
                                     }
+    plg.multiply_splitable_field = { "[follower_id]" => 10 }
     plg.splitable_field_count = 10
     plg.add_foreign_keys_to_events = true
     plg.foreign_key_field = "[id]"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: logstash-input-faker
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Jsericks
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2017-05-31 00:00:00.000000000 Z
+date: 2017-06-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement