embulk-output-elasticsearch_using_url 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 685d2edcad17b21ddc3ed1513b21d0eb60a65fbd
+  data.tar.gz: b2f159f1f335d2b702505a5ebec892fc3f3462fc
+SHA512:
+  metadata.gz: f951209582d56ebbc82cf13e92e9bd52b49abf17cafd7bdea36d64fdb7f58254e103d69665f40a2f6a8282faadc7084cf1a56cc81024578fda37d49f19c47237
+  data.tar.gz: 0461770d6eedf3ab7f00204f7e31d06288768eb47a2453c20fb539695d2ff1c82fd387c476e27460ae56c5280bcabffa10453b7b43050cc47023f796de4c91aa

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org/'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,28 @@
+# Elaasticsearch Using Url output plugin for Embulk
+This plugin is forked from https://github.com/ipros-team/embulk-output-elasticsearch_ruby
+
+## Overview
+
+* **Plugin type**: output
+* **Load all or nothing**: no
+* **Resume supported**: no
+* **Cleanup supported**: no
+
+## Configuration
+
+## Example
+
+```yaml
+out:
+  type: elaasticsearch_using_url
+  mode: normal
+  nodes:
+    - url: "http://localhost:9200"
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

@@ -0,0 +1,3 @@
+require "bundler/gem_tasks"
+
+task default: :build

data/embulk-output-elasticsearch_using_url.gemspec

@@ -0,0 +1,21 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-output-elasticsearch_using_url"
+  spec.version       = "0.1.0"
+  spec.authors       = ["Yuma Murata"]
+  spec.summary       = "Elasticsearch Using Url output plugin for Embulk"
+  spec.description   = "Dumps records to Elasticsearch Using Url."
+  spec.email         = ["murata@ebisol.co.jp"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/ymurata/embulk-output-elasticsearch_using_url"
+
+  spec.files         = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'elasticsearch'
+  spec.add_dependency 'excon'
+  spec.add_development_dependency 'embulk', ['>= 0.8.16']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/output/elasticsearch_using_url.rb

@@ -0,0 +1,148 @@
+module Embulk
+  module Output
+
+    class ElasticsearchUsingUrl < OutputPlugin
+      Plugin.register_output("elasticsearch_using_url", self)
+      ENABLE_MODE = %w[normal update]
+
+      def self.transaction(config, schema, count, &control)
+        task = {
+          "nodes" => config.param("nodes", :array, default: [{ 'url' => 'http://localhost:9200' }]),
+          "request_timeout" => config.param("request_timeout", :integer, default: 60),
+          "index" => config.param("index", :string, default: 'logstash-%Y.%m.%d'),
+          "mode" => config.param("mode", :string, default: 'normal'),
+          "reload_connections" => config.param("reload_connections", :bool, default: true),
+          "reload_on_failure" => config.param("reload_on_failure", :bool, default: false),
+          "delete_old_index" => config.param("delete_old_index", :bool, default: false),
+          "index_type" => config.param("index_type", :string),
+          "id_keys" => config.param("id_keys", :array, default: nil),
+          "id_format" => config.param("id_format", :string, default: nil),
+          "array_columns" => config.param("array_columns", :array, default: nil),
+          "bulk_actions" => config.param("bulk_actions", :integer, default: 1000),
+          "retry_on_failure" => config.param("retry_on_failure", :integer, default: 5),
+          "before_template_name" => config.param("before_template_name", :string, default: nil),
+          "before_template" => config.param("before_template", :hash, default: nil),
+        }
+        task['time_value'] = Time.now.strftime('%Y.%m.%d.%H.%M.%S')
+        task['index'] = Time.now.strftime(task['index'])
+
+        unless ENABLE_MODE.include?(task['mode'])
+          raise ConfigError.new "`mode` must be one of #{ENABLE_MODE.join(', ')}"
+        end
+        Embulk.logger.info("mode => #{task['mode']}")
+
+        if task['before_template_name'] && task['before_template']
+          client = create_client(task)
+          Embulk.logger.info("put template => #{task['before_template_name']}")
+          client.indices.put_template name: task['before_template_name'], body: task['before_template']
+        end
+
+        return self.resume(task, schema, count, &control)
+      end
+
+      def self.resume(task, schema, count, &control)
+        task_reports = yield(task)
+        next_config_diff = {}
+        return next_config_diff
+      end
+
+      def init
+        @nodes = task["nodes"]
+        @index_type = task["index_type"]
+        @id_keys = task["id_keys"]
+        @id_format = task["id_format"]
+        @bulk_actions = task["bulk_actions"]
+        @array_columns = task["array_columns"]
+        @retry_on_failure = task["retry_on_failure"]
+        @mode = task["mode"]
+        @index = task['index']
+
+        @client = create_client(task)
+        @bulk_message = []
+      end
+
+      def close
+      end
+
+      def add(page)
+        page.each do |record|
+          hash = Hash[schema.names.zip(record)]
+          action = (@mode == 'update') ? :update : :index
+          meta = {}
+          meta[action] = { _index: @index, _type: @index_type }
+          meta[action][:_id] = generate_id(@id_format, hash, @id_keys) unless @id_keys.nil?
+          source = generate_array(hash)
+          @bulk_message << meta
+          @bulk_message << source
+          if @bulk_actions * 2 <= @bulk_message.size
+            send
+          end
+        end
+      end
+
+      def finish
+        if @bulk_message.size > 0
+          send
+        end
+      end
+
+      def abort
+      end
+
+      def commit
+        task_report = {}
+        return task_report
+      end
+
+      private
+
+      def create_client(task)
+        return ::Elasticsearch::Client.new urls: task['nodes'].map{|v| v['url']}.join(','),
+          reload_connections: task['reload_connections'],
+          reload_on_failure: task['reload_on_failure'],
+          retry_on_failure: task['retry_on_failure'],
+          transport_options: {
+            request: { timeout: task['request_timeout'] }
+          }
+      end
+
+      def generate_array(record)
+        result = {}
+
+        record.each { |key, value|
+          result[key] = value
+          next if (value.nil? || !@array_columns)
+          @array_columns.each do |array_column|
+            if array_column['name'] == key
+              array_value = value.split(array_column['delimiter']).reject(&:empty?)
+              array_value = array_value.map(&:to_i) if array_column['is_integer']
+              result[key] = array_value
+            end
+          end
+        }
+        (@mode == 'update') ? {doc: result} : result
+      end
+
+      def generate_id(template, record, id_keys)
+        template % id_keys.map { |key| record[key] }
+      end
+
+      def send
+        retries = 0
+        begin
+          @client.bulk body: @bulk_message
+          Embulk.logger.info "bulk: #{@bulk_message.size/2} success."
+        rescue => e
+          if retries < @retry_on_failure
+            retries += 1
+            Embulk.logger.warn "Could not push logs to Elasticsearch, resetting connection and trying again. #{e.message}"
+            sleep 2**retries
+            retry
+          end
+          raise "Could not push logs to Elasticsearch after #{retries} retries. #{e.message}"
+        end
+        @bulk_message.clear
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,120 @@
+--- !ruby/object:Gem::Specification
+name: embulk-output-elasticsearch_using_url
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Yuma Murata
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-02-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: elasticsearch
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: excon
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: embulk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.16
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.16
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Dumps records to Elasticsearch Using Url.
+email:
+- murata@ebisol.co.jp
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-output-elasticsearch_using_url.gemspec
+- lib/embulk/output/elasticsearch_using_url.rb
+homepage: https://github.com/ymurata/embulk-output-elasticsearch_using_url
+licenses:
+- MIT
+metadata: {}
+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.5.1
+signing_key: 
+specification_version: 4
+summary: Elasticsearch Using Url output plugin for Embulk
+test_files: []