logstash-output-mongodb_custom 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 155093e771411be5b38edfaa8128bd1b2c1aa5e939df0502eae047bfe27ac02b
+  data.tar.gz: b453945bc7057acc243607535a610fb5978f0d702cc24e35fdffbc5f5b6a9ec5
+SHA512:
+  metadata.gz: 9335d8a8ee2ded792a6ee87b2edb9898afc0a275690a436559e049bdd05ae9daf669fc690e4b25130d9b7a229ca7e3c4c8e3f0e1b195d5431c9d9c50acdb6f88
+  data.tar.gz: f7b74ff18f50992f9b13cdeeaabc79528ca896fc2d7639ff8c11af49181c33be0ee15581cf5413908c9a0e668905c8a3f5458d71f293f2e1dfbd13b7bb8377fd

data/CHANGELOG.md

@@ -0,0 +1,45 @@
+## 3.1.5
+  - Fixed @timestamp handling, BSON::ObjectId generation, close method [#57](https://github.com/logstash-plugins/logstash-output-mongodb/pull/57)
+
+## 3.1.4
+  - Docs: Set the default_codec doc attribute.
+
+## 3.1.3
+  - Update gemspec summary
+
+## 3.1.2
+  - Fix some documentation issues
+
+## 3.1.0
+ - Add support for bulk inserts to improve performance.
+
+## 3.0.1
+ - Docs: Fix doc generation issue by removing extraneous comments and adding a short description of the plugin
+
+## 3.0.0
+ - Breaking: Updated plugin to use new Java Event APIs
+ - relax logstash-core-plugin-api constrains
+ - update .travis.yml
+
+## 2.0.5
+  - Depend on logstash-core-plugin-api instead of logstash-core, removing the need to mass update plugins on major releases of logstash
+
+## 2.0.4
+  - New dependency requirements for logstash-core for the 5.0 release
+
+## 2.0.3
+ - Patch Timestamp and BigDecimal with to_bson method and register with BSON.
+
+## 2.0.0
+ - Plugins were updated to follow the new shutdown semantic, this mainly allows Logstash to instruct input plugins to terminate gracefully,
+   instead of using Thread.raise on the plugins' threads. Ref: https://github.com/elastic/logstash/pull/3895
+ - Dependency on logstash-core update to 2.0
+
+## 1.0.0
+ - Fixes the plugin to be in the 2.0 series
+ - Add integration and unit test to the project
+ - Adapt the codebase to be 2.0 compatible
+ - Make the internal logger in mongo to report to LS logger
+
+## 0.2.0
+ - Add basic registration test to the project

data/CONTRIBUTORS

@@ -0,0 +1,23 @@
+The following is a list of people who have contributed ideas, code, bug
+reports, or in general have helped logstash along its way.
+
+Contributors:
+* Aaron Mildenstein (untergeek)
+* Graham Bleach (bleach)
+* John E. Vincent (lusis)
+* Jordan Sissel (jordansissel)
+* Kevin Amorin (kamorin)
+* Kevin O'Connor (kjoconnor)
+* Kurt Hurtado (kurtado)
+* Mathias Gug (zimathias)
+* Pete Fritchman (fetep)
+* Pier-Hugues Pellerin (ph)
+* Richard Pijnenburg (electrical)
+* bitsofinfo (bitsofinfo)
+* Guy Boertje (guyboertje)
+* Colin Surprenant (colinsurprenant)
+
+Note: If you've sent us patches, bug reports, or otherwise contributed to
+Logstash, and you aren't on the list above and want to be, please let us know
+and we'll make sure you're here. Contributions from folks like you are what make
+open source awesome.

data/Gemfile

@@ -0,0 +1,11 @@
+source 'https://rubygems.org'
+
+gemspec
+
+logstash_path = ENV["LOGSTASH_PATH"] || "../../logstash"
+use_logstash_source = ENV["LOGSTASH_SOURCE"] && ENV["LOGSTASH_SOURCE"].to_s == "1"
+
+if Dir.exist?(logstash_path) && use_logstash_source
+  gem 'logstash-core', :path => "#{logstash_path}/logstash-core"
+  gem 'logstash-core-plugin-api', :path => "#{logstash_path}/logstash-core-plugin-api"
+end

data/LICENSE

@@ -0,0 +1,13 @@
+Copyright (c) 2012-2018 Elasticsearch <http://www.elastic.co>
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

data/NOTICE.TXT

@@ -0,0 +1,5 @@
+Elasticsearch
+Copyright 2012-2015 Elasticsearch
+
+This product includes software developed by The Apache Software
+Foundation (http://www.apache.org/).

data/README.md

@@ -0,0 +1,98 @@
+# Logstash Plugin
+
+[![Travis Build Status](https://travis-ci.org/logstash-plugins/logstash-output-mongodb.svg)](https://travis-ci.org/logstash-plugins/logstash-output-mongodb)
+
+This is a plugin for [Logstash](https://github.com/elastic/logstash).
+
+It is fully free and fully open source. The license is Apache 2.0, meaning you are pretty much free to use it however you want in whatever way.
+
+## Documentation
+
+Logstash provides infrastructure to automatically generate documentation for this plugin. We use the asciidoc format to write documentation so any comments in the source code will be first converted into asciidoc and then into html. All plugin documentation are placed under one [central location](http://www.elastic.co/guide/en/logstash/current/).
+
+- For formatting code or config example, you can use the asciidoc `[source,ruby]` directive
+- For more asciidoc formatting tips, see the excellent reference here https://github.com/elastic/docs#asciidoc-guide
+
+## Need Help?
+
+Need help? Try #logstash on freenode IRC or the https://discuss.elastic.co/c/logstash discussion forum.
+
+## Developing
+
+### 1. Plugin Developement and Testing
+
+#### Code
+- To get started, you'll need JRuby with the Bundler gem installed.
+
+- Create a new plugin or clone and existing from the GitHub [logstash-plugins](https://github.com/logstash-plugins) organization. We also provide [example plugins](https://github.com/logstash-plugins?query=example).
+
+- Install dependencies
+```sh
+bundle install
+```
+
+#### Test
+
+- Update your dependencies
+
+```sh
+bundle install
+```
+
+- Run tests
+
+```sh
+bundle exec rspec
+```
+
+### 2. Running your unpublished Plugin in Logstash
+
+#### 2.1 Run in a local Logstash clone
+
+- Edit Logstash `Gemfile` and add the local plugin path, for example:
+```ruby
+gem "logstash-filter-awesome", :path => "/your/local/logstash-filter-awesome"
+```
+- Install plugin
+```sh
+# Logstash 2.3 and higher
+bin/logstash-plugin install --no-verify
+
+# Prior to Logstash 2.3
+bin/plugin install --no-verify
+
+```
+- Run Logstash with your plugin
+```sh
+bin/logstash -e 'filter {awesome {}}'
+```
+At this point any modifications to the plugin code will be applied to this local Logstash setup. After modifying the plugin, simply rerun Logstash.
+
+#### 2.2 Run in an installed Logstash
+
+You can use the same **2.1** method to run your plugin in an installed Logstash by editing its `Gemfile` and pointing the `:path` to your local plugin development directory or you can build the gem and install it using:
+
+- Build your plugin gem
+```sh
+gem build logstash-filter-awesome.gemspec
+```
+- Install the plugin from the Logstash home
+```sh
+# Logstash 2.3 and higher
+bin/logstash-plugin install --no-verify
+
+# Prior to Logstash 2.3
+bin/plugin install --no-verify
+
+```
+- Start Logstash and proceed to test the plugin
+
+## Contributing
+
+All contributions are welcome: ideas, patches, documentation, bug reports, complaints, and even something you drew up on a napkin.
+
+Programming is not a required skill. Whatever you've seen about open source and maintainers or community members  saying "send patches or die" - you will not see that here.
+
+It is more important to the community that you are able to contribute.
+
+For more information about contributing, see the [CONTRIBUTING](https://github.com/elastic/logstash/blob/master/CONTRIBUTING.md) file.

data/docs/index.asciidoc

@@ -0,0 +1,137 @@
+:plugin: mongodb
+:type: output
+:default_codec: plain
+
+///////////////////////////////////////////
+START - GENERATED VARIABLES, DO NOT EDIT!
+///////////////////////////////////////////
+:version: %VERSION%
+:release_date: %RELEASE_DATE%
+:changelog_url: %CHANGELOG_URL%
+:include_path: ../../../../logstash/docs/include
+///////////////////////////////////////////
+END - GENERATED VARIABLES, DO NOT EDIT!
+///////////////////////////////////////////
+
+[id="plugins-{type}s-{plugin}"]
+
+=== Mongodb output plugin
+
+include::{include_path}/plugin_header.asciidoc[]
+
+==== Description
+
+This output writes events to MongoDB.
+
+[id="plugins-{type}s-{plugin}-options"]
+==== Mongodb Output Configuration Options
+
+This plugin supports the following configuration options plus the <<plugins-{type}s-{plugin}-common-options>> described later.
+
+[cols="<,<,<",options="header",]
+|=======================================================================
+|Setting |Input type|Required
+| <<plugins-{type}s-{plugin}-bulk>> |<<boolean,boolean>>|No
+| <<plugins-{type}s-{plugin}-bulk_interval>> |<<number,number>>|No
+| <<plugins-{type}s-{plugin}-bulk_size>> |<<number,number>>|No
+| <<plugins-{type}s-{plugin}-collection>> |<<string,string>>|Yes
+| <<plugins-{type}s-{plugin}-database>> |<<string,string>>|Yes
+| <<plugins-{type}s-{plugin}-generateId>> |<<boolean,boolean>>|No
+| <<plugins-{type}s-{plugin}-isodate>> |<<boolean,boolean>>|No
+| <<plugins-{type}s-{plugin}-retry_delay>> |<<number,number>>|No
+| <<plugins-{type}s-{plugin}-uri>> |<<string,string>>|Yes
+|=======================================================================
+
+Also see <<plugins-{type}s-{plugin}-common-options>> for a list of options supported by all
+output plugins.
+
+&nbsp;
+
+[id="plugins-{type}s-{plugin}-bulk"]
+===== `bulk` 
+
+  * Value type is <<boolean,boolean>>
+  * Default value is `false`
+
+Bulk insert flag, set to true to allow bulk insertion, else it will insert events one by one.
+
+[id="plugins-{type}s-{plugin}-bulk_interval"]
+===== `bulk_interval` 
+
+  * Value type is <<number,number>>
+  * Default value is `2`
+
+Bulk interval, Used to insert events periodically if the "bulk" flag is activated.
+
+[id="plugins-{type}s-{plugin}-bulk_size"]
+===== `bulk_size` 
+
+  * Value type is <<number,number>>
+  * Default value is `900`
+
+Bulk events number, if the number of events to insert into a collection raise that limit, it will be bulk inserted
+whatever the bulk interval value (mongodb hard limit is 1000).
+
+[id="plugins-{type}s-{plugin}-collection"]
+===== `collection` 
+
+  * This is a required setting.
+  * Value type is <<string,string>>
+  * There is no default value for this setting.
+
+The collection to use. This value can use `%{foo}` values to dynamically
+select a collection based on data in the event.
+
+[id="plugins-{type}s-{plugin}-database"]
+===== `database` 
+
+  * This is a required setting.
+  * Value type is <<string,string>>
+  * There is no default value for this setting.
+
+The database to use.
+
+[id="plugins-{type}s-{plugin}-generateId"]
+===== `generateId` 
+
+  * Value type is <<boolean,boolean>>
+  * Default value is `false`
+
+If true, an "_id" field will be added to the document before insertion.
+The "_id" field will use the timestamp of the event and overwrite an existing
+"_id" field in the event.
+
+[id="plugins-{type}s-{plugin}-isodate"]
+===== `isodate` 
+
+  * Value type is <<boolean,boolean>>
+  * Default value is `false`
+
+If true, store the @timestamp field in MongoDB as an ISODate type instead
+of an ISO8601 string.  For more information about this, see
+http://www.mongodb.org/display/DOCS/Dates.
+
+[id="plugins-{type}s-{plugin}-retry_delay"]
+===== `retry_delay` 
+
+  * Value type is <<number,number>>
+  * Default value is `3`
+
+The number of seconds to wait after failure before retrying.
+
+[id="plugins-{type}s-{plugin}-uri"]
+===== `uri` 
+
+  * This is a required setting.
+  * Value type is <<string,string>>
+  * There is no default value for this setting.
+
+A MongoDB URI to connect to.
+See http://docs.mongodb.org/manual/reference/connection-string/.
+
+
+
+[id="plugins-{type}s-{plugin}-common-options"]
+include::{include_path}/{type}.asciidoc[]
+
+:default_codec!:

data/lib/logstash/outputs/bson/big_decimal.rb

@@ -0,0 +1,66 @@
+# Copyright (C) 2009-2014 MongoDB Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Modified 2015 Elastic
+
+module BSON
+
+  # Injects behaviour for encoding and decoding BigDecimal values
+  # to and from # raw bytes as specified by the BSON spec.
+  #
+  # @see http://bsonspec.org/#/specification
+  module BigDecimal
+
+    # A floating point is type 0x01 in the BSON spec.
+    BSON_TYPE = 1.chr.force_encoding(BINARY).freeze
+
+    # The pack directive is for 8 byte floating points.
+    PACK = "E".freeze
+
+    # Get the floating point as encoded BSON.
+    # @example Get the floating point as encoded BSON.
+    #   1.221311.to_bson
+    # @return [ String ] The encoded string.
+    # @see http://bsonspec.org/#/specification
+    def to_bson(encoded = ''.force_encoding(BINARY))
+      encoded << [ self ].pack(PACK)
+    end
+
+    module ClassMethods
+
+      # Deserialize an instance of a BigDecimal from a BSON double.
+      # @param [ BSON ] bson object from Mongo.
+      # @return [ BigDecimal ] The decoded BigDecimal.
+      # @see http://bsonspec.org/#/specification
+      def from_bson(bson)
+        from_bson_double(bson.read(8))
+      end
+
+      private
+
+      def from_bson_double(double)
+        new(double.unpack(PACK).first.to_s)
+      end
+    end
+
+    # Register this type when the module is loaded.
+    Registry.register(BSON_TYPE, ::BigDecimal)
+  end
+
+  # Enrich the core BigDecimal class with this module.
+  #
+  # @since 2.0.0
+  ::BigDecimal.send(:include, BigDecimal)
+  ::BigDecimal.send(:extend, BigDecimal::ClassMethods)
+end

data/lib/logstash/outputs/bson/logstash_event.rb

@@ -0,0 +1,76 @@
+# Copyright (C) 2009-2014 MongoDB Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Modified 2015 Elastic
+
+module BSON
+
+  # Injects behaviour for encoding and decoding time values to
+  # and from raw bytes as specified by the BSON spec.
+  #
+  # @see http://bsonspec.org/#/specification
+  module LogStashEvent
+
+    # An Event is an embedded document is type 0x03 in the BSON spec..
+    BSON_TYPE = 3.chr.force_encoding(BINARY).freeze
+
+    # Get the event as encoded BSON.
+    # @example Get the hash as encoded BSON.
+    #   Event.new("field" => "value").to_bson
+    # @return [ String ] The encoded string.
+    # @see http://bsonspec.org/#/specification
+     def to_bson(buffer = ByteBuffer.new)
+      position = buffer.length
+      buffer.put_int32(0)
+      to_hash.each do |field, value|
+        buffer.put_byte(value.bson_type)
+        buffer.put_cstring(field.to_bson_key)
+        value.to_bson(buffer)
+      end
+      buffer.put_byte(NULL_BYTE)
+      buffer.replace_int32(position, buffer.length - position)
+    end
+
+    # Converts the event to a normalized value in a BSON document.
+    # @example Convert the event to a normalized value.
+    #   event.to_bson_normalized_value
+    # @return [ BSON::Document ] The normalized event.
+    def to_bson_normalized_value
+      Document.new(self)
+    end
+
+    module ClassMethods
+      # Deserialize the Event from BSON.
+      # @param [ ByteBuffer ] buffer The byte buffer.
+      # @return [ Event ] The decoded bson document.
+      # @see http://bsonspec.org/#/specification
+      def from_bson(buffer)
+        hash = Hash.new
+        buffer.get_int32 # Throw away the size.
+        while (type = buffer.get_byte) != NULL_BYTE
+          field = buffer.get_cstring
+          hash.store(field, BSON::Registry.get(type).from_bson(buffer))
+        end
+        new(hash)
+      end
+    end
+
+    # Register this type when the module is loaded.
+    Registry.register(BSON_TYPE, ::LogStash::Event)
+  end
+
+  # Enrich the core LogStash::Event class with this module.
+  ::LogStash::Event.send(:include, ::LogStashEvent)
+  ::LogStash::Event.send(:extend, ::LogStashEvent::ClassMethods)
+end

data/lib/logstash/outputs/bson/logstash_timestamp.rb

@@ -0,0 +1,50 @@
+# Copyright (C) 2009-2014 MongoDB Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Modified 2015 Elastic
+
+module BSON
+
+  # Injects behaviour for encoding and decoding time values to
+  # and from raw bytes as specified by the BSON spec.
+  #
+  # @see http://bsonspec.org/#/specification
+  module LogStashTimestamp
+
+    # A time is type 0x09 in the BSON spec.
+    BSON_TYPE = 9.chr.force_encoding(BINARY).freeze
+
+    def to_bson(encoded = ''.force_encoding(BINARY))
+      time.to_bson(encoded)
+    end
+
+    module ClassMethods
+      # Deserialize UTC time from BSON.
+      # @param [ BSON ] bson encoded time.
+      # @return [ ::LogStash::Timestamp ] The decoded UTC time as a ::LogStash::Timestamp.
+      # @see http://bsonspec.org/#/specification
+      def from_bson(bson)
+        seconds, fragment = BSON::Int64.from_bson(bson).divmod(1000)
+        new(::Time.at(seconds, fragment * 1000).utc)
+      end
+    end
+
+    # Register this type when the module is loaded.
+    Registry.register(BSON_TYPE, ::LogStash::Timestamp)
+  end
+
+  # Enrich the core LogStash::Timestamp class with this module.
+  ::LogStash::Timestamp.send(:include, LogStashTimestamp)
+  ::LogStash::Timestamp.send(:extend, LogStashTimestamp::ClassMethods)
+end

data/lib/logstash/outputs/mongodb_custom.rb

@@ -0,0 +1,145 @@
+# encoding: utf-8
+require "logstash/outputs/base"
+require "logstash/namespace"
+require "mongo"
+require_relative "bson/big_decimal"
+require_relative "bson/logstash_timestamp"
+
+# This output writes events to MongoDB.
+class LogStash::Outputs::MongodbCustom < LogStash::Outputs::Base
+
+  config_name "mongodb_custom"
+
+  # A MongoDB URI to connect to.
+  # See http://docs.mongodb.org/manual/reference/connection-string/.
+  config :uri, :validate => :string, :required => true
+
+  # The database to use.
+  config :database, :validate => :string, :required => true
+
+  # The collection to use. This value can use `%{foo}` values to dynamically
+  # select a collection based on data in the event.
+  config :collection, :validate => :string, :required => true
+
+  # If true, store the @timestamp field in MongoDB as an ISODate type instead
+  # of an ISO8601 string.  For more information about this, see
+  # http://www.mongodb.org/display/DOCS/Dates.
+  config :isodate, :validate => :boolean, :default => false
+
+  # The number of seconds to wait after failure before retrying.
+  config :retry_delay, :validate => :number, :default => 3, :required => false
+
+  # If true, an "_id" field will be added to the document before insertion.
+  # The "_id" field will use the timestamp of the event and overwrite an existing
+  # "_id" field in the event.
+  config :generateId, :validate => :boolean, :default => false
+
+
+  # Bulk insert flag, set to true to allow bulk insertion, else it will insert events one by one.
+  config :bulk, :validate => :boolean, :default => false
+  # Bulk interval, Used to insert events periodically if the "bulk" flag is activated.
+  config :bulk_interval, :validate => :number, :default => 2
+  # Bulk events number, if the number of events to insert into a collection raise that limit, it will be bulk inserted
+  # whatever the bulk interval value (mongodb hard limit is 1000).
+  config :bulk_size, :validate => :number, :default => 900, :maximum => 999, :min => 2
+
+  config :date_keys, :validate => :string, :default => nil
+
+  # Mutex used to synchronize access to 'documents'
+  @@mutex = Mutex.new
+
+  def register
+    if @bulk_size > 1000
+      raise LogStash::ConfigurationError, "Bulk size must be lower than '1000', currently '#{@bulk_size}'"
+    end
+
+    Mongo::Logger.logger = @logger
+    conn = Mongo::Client.new(@uri)
+    @db = conn.use(@database)
+
+    @closed = Concurrent::AtomicBoolean.new(false)
+    @documents = {}
+
+    @bulk_thread = Thread.new(@bulk_interval) do |bulk_interval|
+      while @closed.false? do
+        sleep(bulk_interval)
+
+        @@mutex.synchronize do
+          @documents.each do |collection, values|
+            if values.length > 0
+              @db[collection].insert_many(values)
+              @documents.delete(collection)
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def receive(event)
+    begin
+      # Our timestamp object now has a to_bson method, using it here
+      # {}.merge(other) so we don't taint the event hash innards
+      document = {}.merge(event.to_hash)
+
+      if !@isodate
+        timestamp = event.timestamp
+        if timestamp
+          # not using timestamp.to_bson
+          document["@timestamp"] = timestamp.to_json
+        else
+          @logger.warn("Cannot set MongoDB document `@timestamp` field because it does not exist in the event", :event => event)
+        end
+      end
+
+      if @date_keys
+        keys = date_keys.to_s.split(",")
+        document.each do |key, value|
+          if keys.index key
+            document[key] = LogStash::Timestamp.new(value)
+          end
+        end
+      end
+
+      if @generateId
+        document["_id"] = BSON::ObjectId.new
+      end
+
+      if @bulk
+        collection = event.sprintf(@collection)
+        @@mutex.synchronize do
+          if(!@documents[collection])
+            @documents[collection] = []
+          end
+          @documents[collection].push(document)
+
+          if(@documents[collection].length >= @bulk_size)
+            @db[collection].insert_many(@documents[collection])
+            @documents.delete(collection)
+          end
+        end
+      else
+        @db[event.sprintf(@collection)].insert_one(document)
+      end
+    rescue => e
+      if e.message =~ /^E11000/
+        # On a duplicate key error, skip the insert.
+        # We could check if the duplicate key err is the _id key
+        # and generate a new primary key.
+        # If the duplicate key error is on another field, we have no way
+        # to fix the issue.
+        @logger.warn("Skipping insert because of a duplicate key error", :event => event, :exception => e)
+      else
+        @logger.warn("Failed to send event to MongoDB, retrying in #{@retry_delay.to_s} seconds", :event => event, :exception => e)
+        sleep(@retry_delay)
+        retry
+      end
+    end
+  end
+
+  def close
+    @closed.make_true
+    @bulk_thread.wakeup
+    @bulk_thread.join
+  end
+end

data/logstash-output-mongodb_custom.gemspec

@@ -0,0 +1,28 @@
+Gem::Specification.new do |s|
+  s.name            = 'logstash-output-mongodb_custom'
+  s.version         = '0.1.0'
+  s.licenses        = ['Apache-2.0']
+  s.summary         = "Writes events to MongoDB"
+  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"
+  s.authors         = ["Ravi Shah"]
+  s.email           = 'ravi.shah@softwareag.com'
+  s.homepage        = "http://www.elastic.co/guide/en/logstash/current/index.html"
+  s.require_paths = ["lib"]
+
+  # Files
+  s.files = Dir["lib/**/*","spec/**/*","*.gemspec","*.md","CONTRIBUTORS","Gemfile","LICENSE","NOTICE.TXT", "vendor/jar-dependencies/**/*.jar", "vendor/jar-dependencies/**/*.rb", "VERSION", "docs/**/*"]
+
+  # Tests
+  s.test_files = s.files.grep(%r{^(test|spec|features)/})
+
+  # Special flag to let us know this is actually a logstash plugin
+  s.metadata = { "logstash_plugin" => "true", "logstash_group" => "output" }
+
+  # Gem dependencies
+  s.add_runtime_dependency "logstash-core-plugin-api", ">= 1.60", "<= 2.99"
+  s.add_runtime_dependency 'logstash-codec-plain'
+  s.add_runtime_dependency 'mongo', '~> 2.0.6'
+
+  s.add_development_dependency 'logstash-devutils'
+end
+

data/spec/bson/big_decimal_spec.rb

@@ -0,0 +1,30 @@
+# encoding: utf-8
+require 'bigdecimal'
+require_relative "../spec_helper"
+require 'stringio'
+
+describe ::BigDecimal do
+  let(:a_number) { "4321.1234" }
+  let(:bson_number) { 4321.1234.to_bson }
+
+  subject { described_class.new(a_number) }
+
+  it "responds to to_bson" do
+    expect(subject).to respond_to(:to_bson)
+  end
+
+  it "to_bson returns a binary encoded number" do
+    expect(subject.to_bson).to eq(4321.1234.to_bson)
+  end
+
+  it "bson_type returns a binary encoded 1" do
+    expect(subject.bson_type).to eq(12.34.bson_type)
+  end
+
+  describe "class methods" do
+    it "builds a new BigDecimal from BSON" do
+      decoded = described_class.from_bson(StringIO.new(4321.1234.to_bson))
+      expect(decoded).to eql(BigDecimal.new(a_number))
+    end
+  end
+end

data/spec/bson/logstash_timestamp_spec.rb

@@ -0,0 +1,31 @@
+# encoding: utf-8
+require_relative "../spec_helper"
+require 'stringio'
+
+describe ::LogStash::Timestamp do
+  let(:time_array) { [1918,11,11,11,0,0, "+00:00"] }
+  let(:a_time) { Time.utc(*time_array) }
+  let(:bson_time) { Time.utc(*time_array).to_bson }
+
+  subject(:timestamp) { described_class.new(a_time) }
+
+  it "responds to to_bson" do
+    expect(subject).to respond_to(:to_bson)
+  end
+
+  it "to_bson returns a binary encoded timestamp" do
+    expect(timestamp.to_bson).to eq(bson_time)
+  end
+
+  it "bson_type returns a binary encoded 9" do
+    expect(subject.bson_type).to eq(a_time.bson_type)
+  end
+
+  describe "class methods" do
+    it "builds a new Timestamp from BSON" do
+      expected = ::LogStash::Timestamp.new(a_time)
+      decoded = ::LogStash::Timestamp.from_bson(StringIO.new(bson_time))
+      expect(decoded <=> expected).to eq(0)
+    end
+  end
+end

data/spec/integration/mongodb_spec.rb

@@ -0,0 +1,35 @@
+# encoding: utf-8
+require_relative "../spec_helper"
+
+describe LogStash::Outputs::Mongodb, :integration => true do
+
+  let(:uri)        { 'mongodb://localhost:27017' }
+  let(:database)   { 'logstash' }
+  let(:collection) { 'logs' }
+  let(:uuid)       { SecureRandom.uuid }
+
+  let(:config) do
+    { "uri" => uri, "database" => database,
+      "collection" => collection, "isodate" => true }
+  end
+
+  describe "#send" do
+
+    subject { LogStash::Outputs::Mongodb.new(config) }
+
+    let(:properties) { { "message" => "This is a message!",
+                         "uuid" => uuid, "number" => BigDecimal.new("4321.1234"),
+                         "utf8" => "żółć", "int" => 42,
+                         "arry" => [42, "string", 4321.1234]} }
+    let(:event)      { LogStash::Event.new(properties) }
+
+    before(:each) do
+      subject.register
+    end
+
+    it "should send the event to the database" do
+      subject.receive(event)
+      expect(subject).to have_received(event)
+    end
+  end
+end

data/spec/outputs/mongodb_spec.rb

@@ -0,0 +1,86 @@
+# encoding: utf-8
+require_relative "../spec_helper"
+require "logstash/plugin"
+
+describe LogStash::Outputs::Mongodb do
+
+  let(:uri)        { 'mongodb://localhost:27017' }
+  let(:database)   { 'logstash' }
+  let(:collection) { 'logs' }
+
+  let(:config) {{
+    "uri" => uri,
+    "database" => database,
+    "collection" => collection
+  }}
+
+  it "should register and close" do
+    plugin = LogStash::Plugin.lookup("output", "mongodb").new(config)
+    expect {plugin.register}.to_not raise_error
+    plugin.close
+  end
+
+  describe "receive" do
+    subject! { LogStash::Outputs::Mongodb.new(config) }
+
+    let(:event)      { LogStash::Event.new(properties) }
+    let(:connection) { double("connection") }
+    let(:client)     { double("client") }
+    let(:collection) { double("collection") }
+
+    before(:each) do
+      allow(Mongo::Client).to receive(:new).and_return(connection)
+      allow(connection).to receive(:use).and_return(client)
+      allow(client).to receive(:[]).and_return(collection)
+      allow(collection).to receive(:insert_one)
+      subject.register
+    end
+
+    after(:each) do
+      subject.close
+    end
+
+    describe "#send" do
+      let(:properties) {{
+        "message" => "This is a message!",
+        "uuid" => SecureRandom.uuid,
+        "number" => BigDecimal.new("4321.1234"),
+        "utf8" => "żółć"
+      }}
+
+      it "should send the event to the database" do
+        expect(collection).to receive(:insert_one)
+        subject.receive(event)
+      end
+    end
+
+    describe "no event @timestamp" do
+      let(:properties) { { "message" => "foo" } }
+
+      it "should not contain a @timestamp field in the mongo document" do
+        expect(event).to receive(:timestamp).and_return(nil)
+        expect(event).to receive(:to_hash).and_return(properties)
+        expect(collection).to receive(:insert_one).with(properties)
+        subject.receive(event)
+      end
+    end
+
+    describe "generateId" do
+      let(:properties) { { "message" => "foo" } }
+      let(:config) {{
+          "uri" => uri,
+          "database" => database,
+          "collection" => collection,
+          "generateId" => true
+      }}
+
+      it "should contain a BSON::ObjectId as _id" do
+        expect(BSON::ObjectId).to receive(:new).and_return("BSON::ObjectId")
+        expect(event).to receive(:timestamp).and_return(nil)
+        expect(event).to receive(:to_hash).and_return(properties)
+        expect(collection).to receive(:insert_one).with(properties.merge("_id" => "BSON::ObjectId"))
+        subject.receive(event)
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,11 @@
+# encoding: utf-8
+require "logstash/devutils/rspec/spec_helper"
+require "logstash/outputs/mongodb"
+
+RSpec::Matchers.define :have_received do |event|
+  match do |subject|
+    client     = subject.instance_variable_get("@db")
+    collection = subject.instance_variable_get("@collection")
+    client["#{collection}"].find("uuid" => event["uuid"]).count > 0
+  end
+end

metadata

@@ -0,0 +1,131 @@
+--- !ruby/object:Gem::Specification
+name: logstash-output-mongodb_custom
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Ravi Shah
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2020-02-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.60'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '2.99'
+  name: logstash-core-plugin-api
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.60'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '2.99'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: logstash-codec-plain
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.6
+  name: mongo
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.6
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: logstash-devutils
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+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
+email: ravi.shah@softwareag.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- CONTRIBUTORS
+- Gemfile
+- LICENSE
+- NOTICE.TXT
+- README.md
+- docs/index.asciidoc
+- lib/logstash/outputs/bson/big_decimal.rb
+- lib/logstash/outputs/bson/logstash_event.rb
+- lib/logstash/outputs/bson/logstash_timestamp.rb
+- lib/logstash/outputs/mongodb_custom.rb
+- logstash-output-mongodb_custom.gemspec
+- spec/bson/big_decimal_spec.rb
+- spec/bson/logstash_timestamp_spec.rb
+- spec/integration/mongodb_spec.rb
+- spec/outputs/mongodb_spec.rb
+- spec/spec_helper.rb
+homepage: http://www.elastic.co/guide/en/logstash/current/index.html
+licenses:
+- Apache-2.0
+metadata:
+  logstash_plugin: 'true'
+  logstash_group: output
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.7.6
+signing_key:
+specification_version: 4
+summary: Writes events to MongoDB
+test_files:
+- spec/bson/big_decimal_spec.rb
+- spec/bson/logstash_timestamp_spec.rb
+- spec/integration/mongodb_spec.rb
+- spec/outputs/mongodb_spec.rb
+- spec/spec_helper.rb