embulk-filter-mysql 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e111c45ae7b1c0778579d8763d25578646d48b94
+  data.tar.gz: 3adb179342813e12ddbe8c5c390c4ac518feac87
+SHA512:
+  metadata.gz: 247a0ae786842b1be33978bcc1ff321e6085f110594fda24b25f84fe6473c57e481fb728d993a4346656aa297775897c3c2a268e15c7ab269b22263a4f76039f
+  data.tar.gz: ee2156901e57868adcebe358198f34314a03d51a8a8d83c5992f3de4f1c6a1d60520f8c8c61138ccb133edb1afd8dfd6fe99f42d007cffe7310272002f552b4a

data/.gitignore

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

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,107 @@
+# Mysql filter plugin for Embulk
+
+Mysql filter plugin for Embulk. Execute prepared statements query.
+
+## Overview
+
+* **Plugin type**: filter
+
+## Configuration
+
+- **host**: host(string, default: 'localhost')
+- **user**: user(string, required)
+- **password**: password(string, required)
+- **database**: database(string, required)
+- **port**: port(integer, default: 3306)
+- **query**: query(string, required)
+- **params**: params(array, required)
+- **keep_input**: keep_input(bool, default: true)
+
+## Example
+
+```yaml
+in:
+  type: s3
+  bucket: machine-learning-production
+  path_prefix: customer-approval/batch-prediction/result/
+  endpoint: s3.amazonaws.com
+  auth_method: {{ env.EMBULK_S3_AUTH_METHOD }}
+  decoders:
+  - {type: gzip}
+  parser:
+    type: csv
+    delimiter: ","
+    skip_header_lines: 1
+    allow_extra_columns: true
+    allow_optional_columns: true
+    columns:
+      - {name: user_id, type: long}
+      - {name: trueLabel, type: string}
+      - {name: bestAnswer, type: string}
+      - {name: score, type: double}
+filters:
+  - type: mysql
+    host: {{ env.DATABASE_HOST | default: "localhost" }}
+    user: {{ env.APPLICATION_USERNAME }}
+    password: {{ env.APPLICATION_DATABASE_PASS }}
+    database: {{ env.APPLICATION_DATABASE }}
+    keep_input: true
+    query: |
+      select 
+        id,
+        last_name,
+        first_name,
+        company_name
+      from
+        user
+      where id = ?
+    params:
+      - id
+out:
+  type: stdout
+```
+
+#### input CSV
+```
+user_id,trueLabel,bestAnswer,score
+1,0,0,1.5
+2,0,0,1.5
+3,0,0,1.5
+```
+
+#### Running query
+```
+  select 
+    id,
+    last_name,
+    first_name,
+    company_name
+  from
+    user
+  where id = 1;
+
+  select 
+    id,
+    last_name,
+    first_name,
+    company_name
+  from
+    user
+  where id = 2;
+
+  select 
+    id,
+    last_name,
+    first_name,
+    company_name
+  from
+    user
+  where id = 3;
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-filter-mysql.gemspec

@@ -0,0 +1,20 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-filter-mysql"
+  spec.version       = "0.1.0"
+  spec.authors       = ["toyama0919"]
+  spec.summary       = "Mysql filter plugin for Embulk. Execute prepared statements query."
+  spec.description   = "Mysql"
+  spec.email         = ["toyama0919@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/toyama0919/embulk-filter-mysql"
+
+  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 'ffi-mysql'
+  spec.add_development_dependency 'embulk', ['>= 0.8.3']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/filter/mysql.rb

@@ -0,0 +1,132 @@
+# coding: UTF-8
+require 'ffi-mysql'
+ 
+module Embulk
+  module Filter
+    class Mysql < FilterPlugin
+      Plugin.register_filter("mysql", self)
+
+      def self.transaction(config, in_schema, &control)
+        task = {
+          "host" => config.param("host", :string, default: 'localhost'),
+          "user" => config.param("user", :string),
+          "password" => config.param("password", :string),
+          "database" => config.param("database", :string),
+          "port" => config.param("port", :integer, default: 3306),
+          "query" => config.param("query", :string),
+          "params" => config.param("params", :array),
+          "keep_input" => config.param("keep_input", :bool, default: false)
+        }
+        connection = ::Mysql.real_connect(task['host'], task['user'], task['password'], task['database'], task['port'])
+        statement = connection.prepare(task['query'])
+        columns = []
+        columns = columns + in_schema if task['keep_input']
+
+        statement.result_metadata.fetch_fields.each do |field|
+          columns << Column.new(nil, field.name, get_type(field.type))
+        end
+
+        yield(task, columns)
+      end
+
+      # ::Mysql::Field::TYPE_DECIMAL     = 0
+      # ::Mysql::Field::TYPE_TINY        = 1
+      # ::Mysql::Field::TYPE_SHORT       = 2
+      # ::Mysql::Field::TYPE_LONG        = 3
+      # ::Mysql::Field::TYPE_FLOAT       = 4
+      # ::Mysql::Field::TYPE_DOUBLE      = 5
+      # ::Mysql::Field::TYPE_NULL        = 6
+      # ::Mysql::Field::TYPE_TIMESTAMP   = 7
+      # ::Mysql::Field::TYPE_LONGLONG    = 8
+      # ::Mysql::Field::TYPE_INT24       = 9
+      # ::Mysql::Field::TYPE_DATE        = 10
+      # ::Mysql::Field::TYPE_TIME        = 11
+      # ::Mysql::Field::TYPE_DATETIME    = 12
+      # ::Mysql::Field::TYPE_YEAR        = 13
+      # ::Mysql::Field::TYPE_NEWDATE     = 14
+      # ::Mysql::Field::TYPE_VARCHAR     = 15
+      # ::Mysql::Field::TYPE_BIT         = 16
+      # ::Mysql::Field::TYPE_TIMESTAMP2  = 17
+      # ::Mysql::Field::TYPE_DATETIME2   = 18
+      # ::Mysql::Field::TYPE_TIME2       = 19
+      # ::Mysql::Field::TYPE_JSON        = 245
+      # ::Mysql::Field::TYPE_NEWDECIMAL  = 246
+      # ::Mysql::Field::TYPE_ENUM        = 247
+      # ::Mysql::Field::TYPE_SET         = 248
+      # ::Mysql::Field::TYPE_TINY_BLOB   = 249
+      # ::Mysql::Field::TYPE_MEDIUM_BLOB = 250
+      # ::Mysql::Field::TYPE_LONG_BLOB   = 251
+      # ::Mysql::Field::TYPE_BLOB        = 252
+      # ::Mysql::Field::TYPE_VAR_STRING  = 253
+      # ::Mysql::Field::TYPE_STRING      = 254
+      # ::Mysql::Field::TYPE_GEOMETRY    = 255
+      # ::Mysql::Field::TYPE_CHAR        = TYPE_TINY
+      # ::Mysql::Field::TYPE_INTERVAL    = TYPE_ENUM
+      def self.get_type(type)
+        case type
+        when ::Mysql::Field::TYPE_TINY
+          :boolean
+        when ::Mysql::Field::TYPE_SHORT, ::Mysql::Field::TYPE_LONG
+          :long
+        when ::Mysql::Field::TYPE_DOUBLE, ::Mysql::Field::TYPE_FLOAT
+          :double
+        when ::Mysql::Field::TYPE_DATE, ::Mysql::Field::TYPE_DATETIME, ::Mysql::Field::TYPE_TIMESTAMP
+          :timestamp
+        when ::Mysql::Field::TYPE_BLOB, ::Mysql::Field::TYPE_STRING, ::Mysql::Field::TYPE_VAR_STRING, ::Mysql::Field::TYPE_VARCHAR
+          :string
+        else
+          raise
+        end
+      end
+
+      def init
+        @connection = ::Mysql.real_connect(task['host'], task['user'], task['password'], task['database'], task['port'])
+        @statement = @connection.prepare(task['query'])
+        @params = task['params']
+        @keep_input = task['keep_input']
+      end
+
+      def close
+        Embulk.logger.info "connection closing..."
+        @connection.close
+      end
+
+      def add(page)
+        page.each do |record|
+          hash = Hash[in_schema.names.zip(record)]
+          prepare_params = @params ? @params.map{ |param| hash[param] } : []
+          query_results = @statement.execute(*prepare_params)
+          query_results.each do |values|
+            converted = []
+            converted = record + converted if @keep_input
+            values.each do |value|
+              converted << cast(value)
+            end
+            page_builder.add(converted)
+          end
+        end
+      end
+
+      def finish
+        page_builder.finish
+      end
+
+      def cast(value)
+        if (value.class == String)
+          value.force_encoding('UTF-8')
+        elsif (value.class == ::Mysql::Time)
+          Time.local(
+            value.year,
+            value.month,
+            value.day,
+            value.hour,
+            value.minute,
+            value.second
+          )
+        else
+          value
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification
+name: embulk-filter-mysql
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- toyama0919
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2016-04-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: ffi-mysql
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.3
+  name: embulk
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.3
+- !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: Mysql
+email:
+- toyama0919@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-filter-mysql.gemspec
+- lib/embulk/filter/mysql.rb
+homepage: https://github.com/toyama0919/embulk-filter-mysql
+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: Mysql filter plugin for Embulk. Execute prepared statements query.
+test_files: []