embulk-output-amazon_cloudsearch 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 8179993373e5f421335547da4e2e9a6c97ece4ae
+  data.tar.gz: d3c25cd409761f73a31704813fffc526a1ca6ef9
+SHA512:
+  metadata.gz: 4b09e9c367106ecc2f3dd918ebe163df6d92381eb268368532426f71261735d140d8ce72ab6e9a3ec7a26a439b31d783d81dce810794cd05b5c18888b69bb569
+  data.tar.gz: 7319f3688377320336278a126369f9b616ea41222bfaad1217d4727021c9f82e15c67fee1146d7cc1ad0d2c1bfe9d9e7762f061ffe2da369cf978d42c4550b99

data/.gitignore

@@ -0,0 +1,5 @@
+*~
+/pkg/
+/tmp/
+/.bundle/
+/Gemfile.lock

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.1.15.0

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,36 @@
+# Amazon Cloudsearch output plugin for Embulk
+
+Embulk output plugin to insert data into Amazon CloudSearch
+
+## Overview
+
+* **Plugin type**: output
+* **Load all or nothing**: no
+* **Resume supported**: no
+* **Cleanup supported**: no
+
+## Configuration
+
+- **endpoint**: Amazon CloudSearch Document Endpoint URL (string, required)
+- **id_column**: document id column (string, required)
+- **upload_columns**: index columns (string, required)
+- **batch_size**: number of records in one bulk request (int, default: 1000)
+- **stub_response**: CloudSearch API Client stubbing (boolean, default: `false`)
+
+## Example
+
+```yaml
+out:
+  type: amazon_cloudsearch
+  endpoint: https://cloudsearch.example.com/endpoint
+  id_column: id
+  upload_columns:
+    - title
+    - timestamp
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-output-amazon_cloudsearch.gemspec

@@ -0,0 +1,19 @@
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-output-amazon_cloudsearch"
+  spec.version       = "0.1.0"
+  spec.authors       = ["Kenichi Takahashi"]
+  spec.summary       = "Amazon Cloudsearch output plugin for Embulk"
+  spec.description   = "Dumps records to Amazon Cloudsearch."
+  spec.email         = ["kenichi.taka@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/kenchan/embulk-output-amazon_cloudsearch"
+
+  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 'aws-sdk-cloudsearchdomain', ['~> 1.9']
+  spec.add_development_dependency 'embulk', ['>= 0.8.39']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/example/example.csv

@@ -0,0 +1,3 @@
+id,title,timestamp
+1,Hello embulk,2019-05-01 00:00:00.0
+2,Hello CloudSearch,2019-05-31 23:59:59.0

data/example/example.yml

@@ -0,0 +1,22 @@
+in:
+  type: file
+  path_prefix: example/example.csv
+  parser:
+    type: csv
+    charset: UTF-8
+    newline: CRLF
+    null_string: 'NULL'
+    skip_header_lines: 1
+    comment_line_marker: '#'
+    columns:
+      - {name: id, type: string}
+      - {name: title, type: string}
+      - {name: timestamp, type: timestamp, format: "%Y-%m-%d %H:%M:%S.%N", timezone: "+09:00"}
+out:
+  type: amazon_cloudsearch
+  stub_response: true
+  endpoint: https://example.com
+  id_column: id
+  upload_columns:
+    - title
+    - timestamp

data/lib/embulk/output/amazon_cloudsearch.rb

@@ -0,0 +1,102 @@
+require 'json'
+require 'aws-sdk-cloudsearchdomain'
+
+module Embulk
+  module Output
+
+    class AmazonCloudsearch < OutputPlugin
+      Plugin.register_output("amazon_cloudsearch", self)
+
+
+      def self.transaction(config, schema, count, &control)
+        # configuration code:
+        task = {
+          'endpoint' => config.param('endpoint', :string),
+          'id_column' => config.param('id_column', :string),
+          'upload_columns' => config.param('upload_columns', :array),
+          'stub_response' => config.param('stub_response', :bool, default: false),
+          'batch_size' => config.param('batch_size', :integer, default: 1000)
+        }
+
+        # resumable output:
+        # resume(task, schema, count, &control)
+
+        # non-resumable output:
+        task_reports = yield(task)
+        next_config_diff = {}
+        return next_config_diff
+      end
+
+      # def self.resume(task, schema, count, &control)
+      #   task_reports = yield(task)
+      #
+      #   next_config_diff = {}
+      #   return next_config_diff
+      # end
+
+      def init
+        @endpoint = task['endpoint']
+        @id_column = task['id_column']
+        @upload_columns = task['upload_columns']
+        @stub_response = task['stub_response']
+        @batch_size = task['batch_size']
+      end
+
+      def close
+      end
+
+      def add(page)
+        # output code:
+        page.each_slice(@batch_size) do |records|
+          documents = records.map do |record|
+            hash = Hash[schema.names.zip(record)]
+
+            {
+              type: "add",
+              id: hash[@id_column].to_s,
+              fields: @upload_columns.inject({}) {|acc, c|
+                acc[c] = hash[c] if hash[c]
+                acc
+              }
+            }
+          end
+
+          begin
+            res = client.upload_documents(
+              documents: documents.to_json,
+              content_type: 'application/json'
+            )
+          rescue => e
+            Embulk.logger.error { "embulk-output-amazon_cloudsearch: #{e}" }
+            Embulk.logger.error { "embulk-output-amazon_cloudsearch: id #{documents.first[:id]}-#{documents.last[:id]}, response #{res}" }
+          else
+            Embulk.logger.debug { "embulk-output-amazon_cloudsearch: response #{res}" }
+
+            unless res.status == 'success'
+              Embulk.logger.error { "embulk-output-amazon_cloudsearch: id #{documents.first[:id]}-#{documents.last[:id]}, response #{res}" }
+            end
+            if res.warnings
+              Embulk.logger.warn { "embulk-output-amazon_cloudsearch: id #{documents.first[:id]}-#{documents.last[:id]}, response #{res}" }
+            end
+          end
+        end
+      end
+
+      def finish
+      end
+
+      def abort
+      end
+
+      def commit
+        task_report = {}
+        return task_report
+      end
+
+      private
+      def client
+        @_client ||= c = Aws::CloudSearchDomain::Client.new(endpoint: @endpoint, stub_responses: @stub_response)
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,110 @@
+--- !ruby/object:Gem::Specification
+name: embulk-output-amazon_cloudsearch
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Kenichi Takahashi
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2019-06-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  name: aws-sdk-cloudsearchdomain
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.39
+  name: embulk
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.39
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  name: bundler
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  name: rake
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Dumps records to Amazon Cloudsearch.
+email:
+- kenichi.taka@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-output-amazon_cloudsearch.gemspec
+- example/example.csv
+- example/example.yml
+- lib/embulk/output/amazon_cloudsearch.rb
+homepage: https://github.com/kenchan/embulk-output-amazon_cloudsearch
+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.6.14
+signing_key:
+specification_version: 4
+summary: Amazon Cloudsearch output plugin for Embulk
+test_files: []