embulk-output-rediskeys 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6c149c7821c9ddb31512c2f1d70b6ee210f41f59
+  data.tar.gz: 16154b58ad4f0926d5ea54a6cf3335bda2c25d7e
+SHA512:
+  metadata.gz: 32f24d5b23ac98f6c3d767370fbc13f0323f13cc2f4cf4b53ce6b982e8e993a6c3fd164bbf7c0511b68f5aedd286018f791dc1f0f6b923926226d5bcf863ee95
+  data.tar.gz: 483793a9bc6f926dbe6f1a487acd3863434fd5e2631e1436064debe1107e006d8d26a1789c1ea3207860b3404f640a4dd7f8c26a7acaf0a669b6d6eaa3f4ef32

data/.gitignore

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

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.0.4.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 @@
+# Rediskeys output plugin for Embulk
+
+
+## Overview
+
+* **Plugin type**: output
+* **Load all or nothing**: no
+* **Resume supported**: no
+* **Cleanup supported**: yes
+
+## Configuration
+
+- host: Redis hostname (string, default:localhost).
+- port: Redis port number (int, default: 6379).
+- db: Number of Redis DB to dump columns (int, default: 0).
+- key_prefix: Prefix of column name to set. (string, required).
+- encode: Type of eoncoding of data to set (string, default:json).
+
+## Example
+
+```yaml
+out:
+  type: rediskeys
+  host: localhost
+  port: 6379
+  db: 4
+  key_prefix: test_key
+  encode: json
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-output-rediskeys.gemspec

@@ -0,0 +1,21 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-output-rediskeys"
+  spec.version       = "0.1.0"
+  spec.authors       = ["dokuma"]
+  spec.summary       = "Redis output plugin for Embulk"
+  spec.description   = "Dumps records to Redis."
+  spec.email         = ["dokuma.h@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/dokuma/embulk-output-rediskeys"
+
+  spec.files         = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+  spec.has_rdoc      = false
+
+  spec.add_dependency 'redis', ['>= 3.0.5']
+  spec.add_development_dependency 'embulk', ['>= 0.8.9']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/output/rediskeys.rb

@@ -0,0 +1,107 @@
+module Embulk
+  module Output
+
+    class Redis < OutputPlugin
+      require 'redis'
+      require 'json'
+      require 'set'
+
+      Plugin.register_output("rediskeys", self)
+
+      def self.transaction(config, schema, count, &control)
+        # configuration code:
+        task = {
+          'host' => config.param('host', :string, :default => 'localhost'),
+          'port' => config.param('port', :integer, :default => 6379),
+          'db' => config.param('db', :integer, :default => 0),
+          'key_prefix' => config.param('key_prefix', :string, :default => ''),
+          'encode' => config.param('encode', :string, :default => 'json')
+        }
+
+        # resumable output:
+        # resume(task, schema, count, &control)
+
+        # non-resumable output:
+        task_reports = yield(task)
+        puts "Redis output finished. Commit reports = #{task_reports.to_json}"
+
+        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
+        # initialization code:
+        puts "Redis output thread #{index}..."
+        super
+        @rows = 0
+        @processed_keys = [].to_set
+        @unique_keys = [].to_set
+        @redis = ::Redis.new(:host => task['host'], :port => task['port'], :db => task['db'])
+      end
+
+      def close
+      end
+
+      def add(page)
+        # output code:
+        page.each do |records|
+          puts "Schema: #{schema.names}"
+          # puts "Record: #{records}"
+          records.each do |record|
+            hash = JSON.parse(record)
+
+            k = nil
+            v = hash.select{|key,v|
+              k = key
+              key.match(/^#{task['key_prefix']}/)
+            }
+
+            puts "KEY: #{k}"
+
+            @processed_keys << k
+            unless @unique_keys.include? k
+              case task['encode']
+              when 'json'
+                v = v[k].to_json
+                @redis.set(k, v)
+              when 'hash'
+                v = v[k]
+                puts "VALUE: #{v}"
+                puts "FLATTEN: #{v.to_a.flatten}"
+                @redis.hmset(k, v.to_a.flatten)
+              end
+              @unique_keys << k
+            else
+              puts "Warning: #{k} is already exists"
+            end
+            @rows += 1  # inrement anyway
+          end
+        end
+      end
+
+      def finish
+      end
+
+      def abort
+      end
+
+      def commit
+        task_report = {
+          "rows" => @rows,
+          "processed_keys" => @processed_keys.inspect,
+          "unique_keys" => @unique_keys.inspect
+        }
+        return task_report
+      end
+    end
+
+  end
+end
+

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification
+name: embulk-output-rediskeys
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- dokuma
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-06-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.5
+- !ruby/object:Gem::Dependency
+  name: embulk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.9
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.9
+- !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 Redis.
+email:
+- dokuma.h@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-output-rediskeys.gemspec
+- lib/embulk/output/rediskeys.rb
+homepage: https://github.com/dokuma/embulk-output-rediskeys
+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: Redis output plugin for Embulk
+test_files: []