embulk-filter-ruby_proc 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 911204f43abd2acd2d9ac90317e474994fbdf221
+  data.tar.gz: 4521e06844679d721b5e9a21e7d01a368b600278
+SHA512:
+  metadata.gz: 8da351fbf8fe0f39f9e29289bd0365bce6ecc03af0f833c2088fda6fdc9a7b0789781f99a70b44e452adda9824804e192e8256e1290e5f61154c5d497084f3d9
+  data.tar.gz: 2f1ad9b592fb2210ada815239e5bed14f1f4265c2fc43ee715102542e87f97dacef2998031ad4b44b693ecf0fc3aef1b6afdd0b3ea2ce4e3e9c6da32d2fbd600

data/.gitignore

@@ -0,0 +1,6 @@
+*~
+/pkg/
+/tmp/
+/.bundle/
+/Gemfile.lock
+/example/out*

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,80 @@
+# Ruby Proc filter plugin for Embulk
+
+This plugin is inspired by [mgi166/embulk-filter-eval: Eval ruby code on filtering](https://github.com/mgi166/embulk-filter-eval "mgi166/embulk-filter-eval: Eval ruby code on filtering")
+
+This plugin apply ruby proc to each record.
+
+## Overview
+
+* **Plugin type**: filter
+
+## Configuration
+
+- **columns**: filter definition (hash, required)
+- **requires**: pre required libraries (array, default: `[]`)
+
+## Example
+
+### input
+```csv
+id,account,time,purchase,comment,data
+1,32864,2015-01-27 19:23:49,20150127,embulk,"{\"foo\": \"bar\", \"events\": [{\"id\": 1, \"name\": \"Name1\"}, {\"id\": 2, \"name\": \"Name2\"}]}"
+2,14824,2015-01-27 19:01:23,20150127,embulk jruby,NULL
+3,27559,2015-01-28 02:20:02,20150128,"Embulk ""csv"" parser plugin",NULL
+4,11270,2015-01-29 11:54:36,20150129,NULL,NULL
+```
+
+### config
+```yaml
+# ...
+
+filters:
+  - type: ruby_proc
+    requires:
+      - cgi
+    columns:
+      - name: data
+        proc: |
+          ->(data) do
+            data["events"] = data["events"].map.with_index do |e, idx|
+              e.tap { |e_| e_["idx"] = idx }
+            end
+            data.to_json
+          end
+      - name: id
+        proc: |
+          ->(id) do
+            id * 2
+          end
+        type: string
+      - name: comment
+        proc: |
+          ->(comment, record) do
+            return [record["account"].to_s].to_json unless comment
+            comment.upcase.split(" ").map { |s| CGI.escape(s) }.to_json
+          end
+        skip_nil: false
+        type: json
+    target: events
+
+# ...
+
+```
+
+### preview
+```
++-----------+--------------+-------------------------+-------------------------+------------------------------------------+------------------------------------------------------------------------------------------+
+| id:string | account:long |          time:timestamp |      purchase:timestamp |                             comment:json |                                                                                data:json |
++-----------+--------------+-------------------------+-------------------------+------------------------------------------+------------------------------------------------------------------------------------------+
+|         2 |       32,864 | 2015-01-27 19:23:49 UTC | 2015-01-27 00:00:00 UTC |                               ["EMBULK"] | {"events":[{"id":1,"name":"Name1","idx":0},{"id":2,"name":"Name2","idx":1}],"foo":"bar"} |
+|         4 |       14,824 | 2015-01-27 19:01:23 UTC | 2015-01-27 00:00:00 UTC |                       ["EMBULK","JRUBY"] |                                                                                          |
+|         6 |       27,559 | 2015-01-28 02:20:02 UTC | 2015-01-28 00:00:00 UTC | ["EMBULK","%22CSV%22","PARSER","PLUGIN"] |                                                                                          |
+|         8 |       11,270 | 2015-01-29 11:54:36 UTC | 2015-01-29 00:00:00 UTC |                                ["11270"] |                                                                                          |
++-----------+--------------+-------------------------+-------------------------+------------------------------------------+------------------------------------------------------------------------------------------+
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-filter-ruby_proc.gemspec

@@ -0,0 +1,19 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-filter-ruby_proc"
+  spec.version       = "0.1.0"
+  spec.authors       = ["joker1007"]
+  spec.summary       = "Ruby Proc filter plugin for Embulk"
+  spec.description   = "Filter each record by ruby proc"
+  spec.email         = ["kakyoin.hierophant@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/kakyoin.hierophant/embulk-filter-ruby_proc"
+
+  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_development_dependency 'embulk', ['>= 0.8.1']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/example/config.yml

@@ -0,0 +1,67 @@
+in:
+  type: file
+  path_prefix: ./sample_
+  parser:
+    charset: UTF-8
+    newline: CRLF
+    type: csv
+    delimiter: ','
+    quote: '"'
+    escape: '\'
+    null_string: 'NULL'
+    trim_if_not_quoted: false
+    skip_header_lines: 1
+    allow_extra_columns: false
+    allow_optional_columns: false
+    columns:
+    - {name: id, type: long}
+    - {name: account, type: long}
+    - {name: time, type: timestamp, format: '%Y-%m-%d %H:%M:%S'}
+    - {name: purchase, type: timestamp, format: '%Y%m%d'}
+    - {name: comment, type: string}
+    - {name: data, type: json}
+
+filters:
+  - type: ruby_proc
+    requires:
+      - cgi
+    columns:
+      - name: data
+        proc: |
+          ->(data) do
+            data["events"] = data["events"].map.with_index do |e, idx|
+              e.tap { |e_| e_["idx"] = idx }
+            end
+            data.to_json
+          end
+      - name: id
+        proc: |
+          ->(id) do
+            id * 2
+          end
+        type: string
+      - name: comment
+        proc: |
+          ->(comment, record) do
+            return [record["account"].to_s].to_json unless comment
+            comment.upcase.split(" ").map { |s| CGI.escape(s) }.to_json
+          end
+        skip_nil: false
+        type: json
+    target: events
+
+out:
+  type: file
+  path_prefix: ./out_
+  file_ext: tsv
+  formatter:
+    type: csv
+    delimiter: "\t"
+    newline: CRLF
+    newline_in_field: LF
+    charset: UTF-8
+    quote_policy: MINIMAL
+    quote: '"'
+    escape: "\\"
+    null_string: 'NULL'
+    default_timezone: 'UTC'

data/example/sample_01.csv

@@ -0,0 +1,5 @@
+id,account,time,purchase,comment,data
+1,32864,2015-01-27 19:23:49,20150127,embulk,"{\"foo\": \"bar\", \"events\": [{\"id\": 1, \"name\": \"Name1\"}, {\"id\": 2, \"name\": \"Name2\"}]}"
+2,14824,2015-01-27 19:01:23,20150127,embulk jruby,NULL
+3,27559,2015-01-28 02:20:02,20150128,"Embulk ""csv"" parser plugin",NULL
+4,11270,2015-01-29 11:54:36,20150129,NULL,NULL

data/lib/embulk/filter/ruby_proc.rb

@@ -0,0 +1,70 @@
+module Embulk
+  module Filter
+
+    class RubyProc < FilterPlugin
+      Plugin.register_filter("ruby_proc", self)
+
+      def self.transaction(config, in_schema, &control)
+        task = {
+          "columns" => config.param("columns", :array),
+          "requires" => config.param("requires", :array, default: []),
+        }
+
+        out_columns = in_schema.map do |col|
+          target = task["columns"].find { |filter_col| filter_col["name"] == col.name }
+          if target
+            type = target["type"] ? target["type"].to_sym : col.type
+            Embulk::Column.new(index: col.index, name: col.name, type: type || col.type, format: target["format"] || col.format)
+          else
+            col
+          end
+        end
+
+        yield(task, out_columns)
+      end
+
+      def init
+        task["requires"].each do |lib|
+          require lib
+        end
+        @procs = Hash[task["columns"].map {|col|
+          [col["name"], eval(col["proc"])]
+        }]
+        @skip_nils = Hash[task["columns"].map {|col|
+          [col["name"], col["skip_nil"].nil? ? true : !!col["skip_nil"]]
+        }]
+      end
+
+      def close
+      end
+
+      def add(page)
+        page.each do |record|
+          record_hash = hashrize(record)
+          @procs.each do |col, pr|
+            next unless record_hash.has_key?(col)
+            next if record_hash[col].nil? && @skip_nils[col]
+
+            if pr.arity == 1
+              record_hash[col] = pr.call(record_hash[col])
+            else
+              record_hash[col] = pr.call(record_hash[col], record_hash)
+            end
+          end
+          page_builder.add(record_hash.values)
+        end
+      end
+
+      def finish
+        page_builder.finish
+      end
+
+      private
+
+      def hashrize(record)
+        Hash[in_schema.names.zip(record)]
+      end
+    end
+
+  end
+end

metadata

@@ -0,0 +1,96 @@
+--- !ruby/object:Gem::Specification
+name: embulk-filter-ruby_proc
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- joker1007
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2016-02-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: embulk
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.1
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.1
+  prerelease: false
+  type: :development
+- !ruby/object:Gem::Dependency
+  name: bundler
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  prerelease: false
+  type: :development
+- !ruby/object:Gem::Dependency
+  name: rake
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  prerelease: false
+  type: :development
+description: Filter each record by ruby proc
+email:
+- kakyoin.hierophant@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-filter-ruby_proc.gemspec
+- example/config.yml
+- example/sample_01.csv
+- lib/embulk/filter/ruby_proc.rb
+homepage: https://github.com/kakyoin.hierophant/embulk-filter-ruby_proc
+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.8
+signing_key:
+specification_version: 4
+summary: Ruby Proc filter plugin for Embulk
+test_files: []