logstash-output-elasticsearch_java 1.0.0.beta1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3f3598838ddd3f3b31bb2fd185b947913d8abf21
+  data.tar.gz: b360525c05bd16bff468a8c0d93257e966926d36
+SHA512:
+  metadata.gz: dc56371b39b232bb78bc6369fc98e7976b9ec4f91a64ae5876b3207d58e29c941dd5e896693f801cfe30be5dfc136d105c8be26b410118f7b57eee0525c0d5f1
+  data.tar.gz: 24e90919491c6d754cc9c08f6d1d90bd178a5883d1692c516a0bcba2e276ccea1e61221f6871ce1847ec58dca23daa295498ceae08717512433085822a77831a

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+Gemfile.lock
+.bundle
+.idea
+*~

data/CHANGELOG.md

@@ -0,0 +1,2 @@
+## 1.0.0
+  - Initial Release. 

data/CONTRIBUTORS

@@ -0,0 +1,31 @@
+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)
+* Bob Corsaro (dokipen)
+* Colin Surprenant (colinsurprenant)
+* Dmitry Koprov (dkoprov)
+* Graham Bleach (bleach)
+* Hao Chen (haoch)
+* James Turnbull (jamtur01)
+* John E. Vincent (lusis)
+* Jordan Sissel (jordansissel)
+* João Duarte (jsvd)
+* Kurt Hurtado (kurtado)
+* Miah Johnson (miah)
+* Pere Urbón (purbon)
+* Pete Fritchman (fetep)
+* Pier-Hugues Pellerin (ph)
+* Raymond Feng (raymondfeng)
+* Richard Pijnenburg (electrical)
+* Spenser Jones (SpenserJ)
+* Suyog Rao (suyograo)
+* Tal Levy (talevy)
+* Tom Hodder (tolland)
+* jimmyjones2
+
+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,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/LICENSE

@@ -0,0 +1,13 @@
+Copyright (c) 2012–2015 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
+
+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 unit tests
+
+```sh
+bundle exec rspec
+```
+
+- Run integration tests
+
+Dependencies: [Docker](http://docker.com)
+
+Before the test suite is run, we will load and run an
+Elasticsearch instance within a docker container. This container 
+will be cleaned up when suite has finished.
+
+```sh
+bundle exec rspec --tag integration
+```
+
+### 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
+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
+bin/plugin install /your/local/plugin/logstash-filter-awesome.gem
+```
+- 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/Rakefile

@@ -0,0 +1 @@
+require "logstash/devutils/rake"

data/lib/logstash/outputs/elasticsearch_java/elasticsearch-template.json

@@ -0,0 +1,41 @@
+{
+  "template" : "logstash-*",
+  "settings" : {
+    "index.refresh_interval" : "5s"
+  },
+  "mappings" : {
+    "_default_" : {
+       "_all" : {"enabled" : true, "omit_norms" : true},
+       "dynamic_templates" : [ {
+         "message_field" : {
+           "match" : "message",
+           "match_mapping_type" : "string",
+           "mapping" : {
+             "type" : "string", "index" : "analyzed", "omit_norms" : true
+           }
+         }
+       }, {
+         "string_fields" : {
+           "match" : "*",
+           "match_mapping_type" : "string",
+           "mapping" : {
+             "type" : "string", "index" : "analyzed", "omit_norms" : true,
+               "fields" : {
+                 "raw" : {"type": "string", "index" : "not_analyzed", "ignore_above" : 256}
+               }
+           }
+         }
+       } ],
+       "properties" : {
+         "@version": { "type": "string", "index": "not_analyzed" },
+         "geoip"  : {
+           "type" : "object",
+             "dynamic": true,
+             "properties" : {
+               "location" : { "type" : "geo_point" }
+             }
+         }
+       }
+    }
+  }
+}

data/lib/logstash/outputs/elasticsearch_java/protocol.rb

@@ -0,0 +1,258 @@
+require "cabin"
+require "base64"
+require 'logstash-output-elasticsearch_java_jars.rb'
+require 'logstash/outputs/elasticsearch_java'
+
+module LogStash
+  module Outputs
+    module ElasticSearchJavaPlugins
+      module Protocols
+        class Base
+          private
+          def initialize(options={})
+            # host(s), port, cluster
+            @logger = Cabin::Channel.get
+          end
+
+          def client
+            return @client if @client
+            @client = build_client(@options)
+            return @client
+          end
+
+          def template_install(name, template, force=false)
+            if template_exists?(name) && !force
+              @logger.debug("Found existing Elasticsearch template. Skipping template management", :name => name)
+              return
+            end
+            template_put(name, template)
+          end
+
+          # Do a bulk request with the given actions.
+          #
+          # 'actions' is expected to be an array of bulk requests as string json
+          # values.
+          #
+          # Each 'action' becomes a single line in the bulk api call. For more
+          # details on the format of each.
+          def bulk(actions)
+            raise NotImplemented, "You must implement this yourself"
+            # bulk([
+            # '{ "index" : { "_index" : "test", "_type" : "type1", "_id" : "1" } }',
+            # '{ "field1" : "value1" }'
+            #])
+          end
+
+          public(:initialize, :template_install)
+        end
+
+        class NodeClient < Base
+          CLIENT_MUTEX = Mutex.new
+
+          def self.get_client(settings)
+            CLIENT_MUTEX.synchronize {
+              if @client
+                @client
+              else
+                nodebuilder = org.elasticsearch.node.NodeBuilder.nodeBuilder
+                @client = nodebuilder.settings(settings.build).node().client()
+              end
+            }
+          end
+
+          def self.clear_client()
+            CLIENT_MUTEX.synchronize {
+              @client = nil
+            }
+          end
+
+          private
+
+          DEFAULT_OPTIONS = {
+            :port => 9300,
+          }
+
+          def initialize(options={})
+            super
+            require "java"
+            @options = DEFAULT_OPTIONS.merge(options)
+            setup(@options)
+          end # def initialize
+
+          def settings
+            return @settings
+          end
+
+          def client
+            self.class.get_client(settings)
+          end
+
+          def setup(options={})
+            @settings = org.elasticsearch.common.settings.ImmutableSettings.settingsBuilder
+            if options[:hosts]
+              @settings.put("discovery.zen.ping.multicast.enabled", false)
+              @settings.put("discovery.zen.ping.unicast.hosts", NodeClient.hosts(options))
+            end
+
+            @settings.put("node.client", true)
+            @settings.put("http.enabled", false)
+
+            if options[:client_settings]
+              options[:client_settings].each do |key, value|
+                @settings.put(key, value)
+              end
+            end
+
+            return @settings
+          end
+
+          def self.hosts(options)
+            # http://www.elasticsearch.org/guide/reference/modules/discovery/zen/
+            result = Array.new
+            if options[:hosts].class == Array
+              options[:hosts].each do |host|
+                if host.to_s =~ /^.+:.+$/
+                  # For host in format: host:port, ignore options[:port]
+                  result << host
+                else
+                  if options[:port].to_s =~ /^\d+-\d+$/
+                    # port ranges are 'host[port1-port2]'
+                    result << Range.new(*options[:port].split("-")).collect { |p| "#{host}:#{p}" }
+                  else
+                    result << "#{host}:#{options[:port]}"
+                  end
+                end
+              end
+            else
+              if options[:hosts].to_s =~ /^.+:.+$/
+                # For host in format: host:port, ignore options[:port]
+                result << options[:hosts]
+              else
+                if options[:port].to_s =~ /^\d+-\d+$/
+                  # port ranges are 'host[port1-port2]' according to
+                  # http://www.elasticsearch.org/guide/reference/modules/discovery/zen/
+                  # However, it seems to only query the first port.
+                  # So generate our own list of unicast hosts to scan.
+                  range = Range.new(*options[:port].split("-"))
+                  result << range.collect { |p| "#{options[:hosts]}:#{p}" }
+                else
+                  result << "#{options[:hosts]}:#{options[:port]}"
+                end
+              end
+            end
+            result.flatten.join(",")
+          end
+
+          def self.normalize_bulk_response(bulk_response)
+            # TODO(talevy): parse item response objects to retrieve correct 200 (OK) or 201(created) status codes
+            if bulk_response.has_failures()
+              {"errors" => true,
+               "statuses" => bulk_response.map { |i| (i.is_failed && i.get_failure.get_status.get_status) || 200 }}
+            else
+              {"errors" => false}
+            end
+          end
+
+          def bulk(actions)
+            # Actions an array of [ action, action_metadata, source ]
+            prep = client.prepareBulk
+            actions.each do |action, args, source|
+              prep.add(build_request(action, args, source))
+            end
+            response = prep.execute.actionGet()
+
+            self.class.normalize_bulk_response(response)
+          end # def bulk
+
+          def build_request(action, args, source)
+            case action
+              when "index"
+                request = org.elasticsearch.action.index.IndexRequest.new(args[:_index])
+                request.id(args[:_id]) if args[:_id]
+                request.routing(args[:_routing]) if args[:_routing]
+                request.source(source)
+              when "delete"
+                request = org.elasticsearch.action.delete.DeleteRequest.new(args[:_index])
+                request.id(args[:_id])
+                request.routing(args[:_routing]) if args[:_routing]
+              when "create"
+                request = org.elasticsearch.action.index.IndexRequest.new(args[:_index])
+                request.id(args[:_id]) if args[:_id]
+                request.routing(args[:_routing]) if args[:_routing]
+                request.source(source)
+                request.opType("create")
+              when "create_unless_exists"
+                unless args[:_id].nil?
+                  request = org.elasticsearch.action.index.IndexRequest.new(args[:_index])
+                  request.id(args[:_id])
+                  request.routing(args[:_routing]) if args[:_routing]
+                  request.source(source)
+                  request.opType("create")
+                else
+                  raise(LogStash::ConfigurationError, "Specifying action => 'create_unless_exists' without a document '_id' is not supported.")
+                end
+              when "update"
+                unless args[:_id].nil?
+                  request = org.elasticsearch.action.update.UpdateRequest.new(args[:_index], args[:_type], args[:_id])
+                  request.routing(args[:_routing]) if args[:_routing]
+                  request.doc(source)
+                  if @options[:doc_as_upsert]
+                    request.docAsUpsert(true)
+                  else
+                    request.upsert(args[:_upsert]) if args[:_upsert]
+                  end
+                else
+                  raise(LogStash::ConfigurationError, "Specifying action => 'update' without a document '_id' is not supported.")
+                end
+              else
+                raise(LogStash::ConfigurationError, "action => '#{action_name}' is not currently supported.")
+            end # case action
+
+            request.type(args[:_type]) if args[:_type]
+            return request
+          end # def build_request
+
+          def template_exists?(name)
+            request = org.elasticsearch.action.admin.indices.template.get.GetIndexTemplatesRequestBuilder.new(client.admin.indices, name)
+            response = request.get
+            return !response.getIndexTemplates.isEmpty
+          end
+
+          def template_put(name, template)
+            request = org.elasticsearch.action.admin.indices.template.put.PutIndexTemplateRequestBuilder.new(client.admin.indices, name)
+            request.setSource(LogStash::Json.dump(template))
+
+            # execute the request and get the response, if it fails, we'll get an exception.
+            request.get
+          end
+
+          public(:initialize, :bulk)
+        end # class NodeClient
+
+        class TransportClient < NodeClient
+          private
+          def build_client(options)
+            client = org.elasticsearch.client.transport.TransportClient.
+              builder().
+              settings((settings.build)).
+              build()
+
+            options[:hosts].each do |host|
+              matches = host.match /(.+)(?:.*)/
+
+              inet_addr = java.net.InetAddress.getByName(matches[1])
+              port = (matches[2] || options[:port]).to_i
+              client.addTransportAddress(
+                org.elasticsearch.common.transport.InetSocketTransportAddress.new(
+                  inet_addr, port
+                )
+              )
+            end
+
+            return client
+          end
+        end
+      end
+    end
+  end
+end