logstash-input-mongoprofile 0.1.14 → 0.1.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9f1791577f9d52eae124d6f9603624d9ff2db228
-  data.tar.gz: 0db3f8282ddf833154b8e1db77f78496da25e91b
+  metadata.gz: d4b22c8ef7d441e7d6d7130e51f18990fe34f272
+  data.tar.gz: 3534f85ef64368f518f18ac27cb259a64c862a46
 SHA512:
-  metadata.gz: 0bace47f999c477d705a29026390c5ef63b65a5c1d6a971c4a81cc415c121cc1cff9705b1d9c03a6f753809632ca4588869ca00655491c48c209fc7ea3d28fe7
-  data.tar.gz: 5067c562a892770f2d0c9b588bc60810d4e5e431b3ba063564fd0b669aa1e31ee66fe3e2d5a10d5aae95d3b29e397099261969a17336dc1f9e2a026751672d8c
+  metadata.gz: 40e4941008a948ab16952549645d78b6a4194d566024aba8bb1b38b66c2c9d83643228d141e42a76e2399acf9927917ab632adb20499608ca0aba456ddb847b6
+  data.tar.gz: f560de9428f0d222f5810218cc1782efe77bd74f9143d257621b7a2191f48d7b4b3db9e41c8ae54f39f5d963bb652364b5b8c88c683f74aa025d9c8e1b5e5307

data/lib/logstash/inputs/mongoprofile.rb

@@ -20,11 +20,12 @@ class LogStash::Inputs::Mongoprofile < LogStash::Inputs::Base
   config :url, :validate => :string, :required => true
   config :path, :validate => :string, :required => true
   config :client_host, :validate => :string, :default => '127.0.0.1'
+  config :generate_id, :validate => :boolean, :default => false
 
   public
   def register
     @host = Socket.gethostname
-    @controller = Controller.new(@host, @url, 'system.profile', 1000, @path, @client_host, @logger)
+    @controller = Controller.new(@host, @url, 'system.profile', 1000, @path, @client_host, @logger, @generate_id)
   end
 
   # def register
@@ -83,7 +84,8 @@ class MongoAccessor
 end
 
 class ProfileCollection
-  def initialize(documents, parser)
+  def initialize(documents, parser, generate_id)
+    @generate_id = generate_id
     @documents = []
 
     documents.each do |document|
@@ -95,7 +97,10 @@ class ProfileCollection
 
   def each
     @documents.each do |document|
-      document['_id'] = generate_id.to_s
+      if @generate_id
+        document['_id'] = generate_id.to_s
+      end
+
       yield @parser.parse(document)
     end
   end
@@ -154,10 +159,11 @@ class LastValueStore
 end
 
 class Controller
-  def initialize(event, url, collection, limit, path, client_host, logger)
+  def initialize(event, url, collection, limit, path, client_host, logger, generate_id)
     @mongo_accessor = MongoAccessor.new(url, collection, client_host)
     @last_value_store = LastValueStore.new(path, collection)
     @document_parser = DocumentParser.new(event, logger)
+    @generate_id = generate_id
     @limit = limit
     @logger = logger
   end
@@ -173,7 +179,7 @@ class Controller
       documents = @mongo_accessor.get_documents_by_ts(last_date_value, @limit)
     end
 
-    profile_collection = ProfileCollection.new(documents, @document_parser)
+    profile_collection = ProfileCollection.new(documents, @document_parser, @generate_id)
 
     if  profile_collection.get_last_document_date != nil
       @last_value_store.save_last_value(profile_collection.get_last_document_date)

data/logstash-input-mongoprofile.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name          = 'logstash-input-mongoprofile'
-  s.version       = '0.1.14'
+  s.version       = '0.1.15'
   s.licenses      = ['Apache License (2.0)']
   s.summary       = 'MongoDB system.profile input plugin'
   s.description   = 'MongoDB system.profile input plugin'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: logstash-input-mongoprofile
 version: !ruby/object:Gem::Version
-  version: 0.1.14
+  version: 0.1.15
 platform: ruby
 authors:
 - Artem Antonov