embulk-output-groonga 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3a982df5a40609112ffe3d24a191986a450ba722
+  data.tar.gz: ed668bb7ed6768fe11e322d9eb6385139fcdef65
+SHA512:
+  metadata.gz: 3b32c1340d87ff3bd5988e192887b812e7280e2c5c8b8fc390a1d6df1113c9cbda13ee562394f6754bdf2c7a4771a6bab9cf3678f54745ee796e1edd36215946
+  data.tar.gz: 97c5ea885c1986ca37a75fc4fbaa47957519d24d332f5dbc25a52ab155f5d0d50a005297de225deb019ae3a3d39fd43d309d6d16ebec6499c0a41c402076cfc5

data/.gitignore

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

data/CHANGES.md

@@ -0,0 +1,4 @@
+0.1.1 (2015-08-10)
+------------------
+* First release
+

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,54 @@
+# Groonga output plugin for Embulk
+
+Embulk output plugin to load insert data into groonga (full text search engine)
+
+## Overview
+
+* **Plugin type**: output
+* **Load all or nothing**: no
+* **Resume supported**: no
+* **Cleanup supported**: yes
+
+## Configuration
+
+- **table**: output table name (string, required)
+- **key_column**: this column convert column name into _key (string, required)
+- **host**: groonga server name (string, rerquired)
+- **port**: groonga port (integer, default: 10041)
+- **protocol**: connection protocol gqtp or http, (string, default: http )
+
+
+## Installation
+
+* install embulk command isself.
+* install embulk-output-groonga.
+
+### Embulk install
+
+install [embulk](https://github.com/embulk/embulk#quick-start)
+For libyajl2 installation, *Embulk 0.6.10 required*.
+
+### install gem from github master.
+
+install embulk-output-groonga
+
+```
+git clone https://github.com/hiroyuki-sato/embulk-output-groonga.git
+cd embulk-output-groonga
+rake build
+embulk gem install pkg/embulk-output-groonga-0.1.0.gem
+```
+
+## Example
+
+Installation step are the following.
+
+```yaml
+out:
+  type: groonga
+  table: Site
+  key_column: key
+  host: localhost
+  port: 10041
+  protocol: http
+```

data/Rakefile

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

data/embulk-output-groonga.gemspec

@@ -0,0 +1,20 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-output-groonga"
+  spec.version       = "0.1.1"
+  spec.authors       = ["Hiroyuki Sato"]
+  spec.summary       = "Groonga output plugin for Embulk"
+  spec.description   = "Dumps records to Groonga."
+  spec.email         = ["hiroysato@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/hiroyuki-sato/embulk-output-groonga"
+
+  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 'groonga-command-parser', ['>= 0.1.4']
+  spec.add_dependency 'groonga-client', ['>= 0.1.8']
+  spec.add_development_dependency 'bundler', ['~> 1.0']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/output/groonga.rb

@@ -0,0 +1,120 @@
+require 'groonga/client'
+
+module Embulk
+  module Output
+
+    FLUSH_SIZE = 1_000
+    class GroongaOutputPlugin < OutputPlugin
+      Plugin.register_output("groonga", self)
+
+      def self.transaction(config, schema, count, &control)
+        # configuration code:
+        task = {
+          "host" => config.param("host", :string),
+          "port" => config.param("port", :integer, default: 10041),
+          "protocol" => config.param("protocol", :string, default: 'http'),
+          "key_column" => config.param("key_column",:string),
+          "table" => config.param("table",:string),
+#          "create_table" => config.param("create_table",:string)
+        }
+        prot = task['protocol']
+        raise RuntimeError,"Unknown protocol #{prot}. supported protocol: gqtp, http" unless %w[gqtp http].include?(prot)
+
+        # resumable output:
+        # resume(task, schema, count, &control)
+
+        # non-resumable output:
+        commit_reports = yield(task)
+        next_config_diff = {}
+        return next_config_diff
+      end
+
+      #def self.resume(task, schema, count, &control)
+      #  commit_reports = yield(task)
+      #
+      #  next_config_diff = {}
+      #  return next_config_diff
+      #end
+
+      def init
+        # initialization code:
+        host = task["host"]
+        port = task["port"]
+        protocol = task["protocol"].to_sym
+        @client = Groonga::Client.open({:host => host,
+                                        :port => port,
+                                        :protocol => protocol})
+        @key_column = task["key_column"]
+        @out_table = task["table"]
+
+#        create_table
+      end
+
+      def close
+        @client.close
+      end
+
+      def add(page)
+        # output code:
+        records = []
+        idx = 0
+        page.each_with_index do |record,idx|
+          hash = Hash[schema.names.zip(record)]
+          v = hash.delete(@key_column)
+          hash['_key'] = v
+          records << hash
+          if( idx > 0 && idx % FLUSH_SIZE == 0 )
+            ret = @client.load({:table => @out_table,
+                                :values => records })
+             Embulk.logger.info "groonga inserted #{ret.body} / #{records.size}"
+             if( ret.body != records.size )
+               raise RuntimeError,"inserted value does not match"
+             end
+             records.clear
+          end
+        end
+        if( records.size > 0 )
+          ret = @client.load({:table => @out_table,
+                              :values => records })
+           Embulk.logger.info "groonga inserted #{ret.body} / #{records.size}"
+           if( ret.body != records.size )
+             raise RuntimeError,"inserted value does not match"
+           end
+        end
+      end
+
+      def finish
+      end
+
+      def abort
+      end
+
+      def commit
+        commit_report = {}
+        return commit_report
+      end
+      private
+      def table_names
+        # TODO Error check
+        @client.table_list.map{ |x| x['name'] }
+      end
+
+      def table_exist?(name)
+        # TODO Error check
+        table_names.include?(name)
+      end
+
+      def create_table
+        return if table_exist?(@out_table)
+        create_table = @task['create_table']
+
+        # TODO Error check
+
+        @client.execute(create_table)
+
+      end
+
+    end
+
+  end
+end

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: embulk-output-groonga
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Hiroyuki Sato
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-12-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: groonga-command-parser
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.4
+- !ruby/object:Gem::Dependency
+  name: groonga-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.8
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.1.8
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !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 Groonga.
+email:
+- hiroysato@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- CHANGES.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-output-groonga.gemspec
+- lib/embulk/output/groonga.rb
+homepage: https://github.com/hiroyuki-sato/embulk-output-groonga
+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.4.5.1
+signing_key: 
+specification_version: 4
+summary: Groonga output plugin for Embulk
+test_files: []
+has_rdoc: