embulk-parser-mysqldump_tab 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 540119b437ab3189dafba2b577bd0ef444c72583
+  data.tar.gz: ac15037cbdaab0a87a50a392837f28b14e742e65
+SHA512:
+  metadata.gz: a5f756f601edcb1bd9f2eec7845cb9775b165c1c65ce99426160542d24f0b19ee604487364bdbe78614c7c6bfb129ba045ce211780697ac7eda13df0ead88653
+  data.tar.gz: 53ea4300b9ed686a312d4ce8a8da9d82a843c14eeaaa59274a669d12c209310fe0399088ec43126c87c264e5ba289f1ef4b78518d02dc3c5a73e578415c497ca

data/.gitignore

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

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.1.5.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,39 @@
+# Mysqldump Tab parser plugin for Embulk
+
+Embulk parser plugin for mysqldump file that dumped with the --tab option
+
+## Overview
+
+* **Plugin type**: parser
+* **Guess supported**: no
+
+## Configuration
+
+
+## Example
+
+```yaml
+in:
+  type: file
+  path_prefix: /path/to/dump/users.txt
+  parser:
+    type: mysqldump_tab
+    columns:
+    - {name: id, type: long}
+    - {name: name, type: string}
+    - {name: email, type: string}
+out:
+  type: stdout
+```
+
+
+```
+$ embulk gem install embulk-parser-mysqldump_tab
+$ embulk guess -g mysqldump_tab config.yml -o guessed.yml
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-parser-mysqldump_tab.gemspec

@@ -0,0 +1,19 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-parser-mysqldump_tab"
+  spec.version       = "0.1.0"
+  spec.authors       = ["inouet"]
+  spec.summary       = "Mysqldump Tab parser plugin for Embulk"
+  spec.description   = "Embulk parser plugin for mysqldump file that dumped with the --tab option."
+  spec.email         = ["inudog@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/inouet/embulk-parser-mysqldump_tab"
+  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 'YOUR_GEM_DEPENDENCY', ['~> YOUR_GEM_DEPENDENCY_VERSION']
+  spec.add_development_dependency 'embulk', ['>= 0.8.23']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/guess/mysqldump_tab.rb

@@ -0,0 +1,63 @@
+module Embulk
+  module Guess
+
+    # TODO implement guess plugin to make this command work:
+    #      $ embulk guess -g "mysqldump_tab" partial-config.yml
+    #
+    #      Depending on the file format the plugin uses, you can use choose
+    #      one of binary guess (GuessPlugin), text guess (TextGuessPlugin),
+    #      or line guess (LineGuessPlugin).
+
+    # require "embulk/parser/mysqldump_tab.rb"
+
+    # class MysqldumpTab < GuessPlugin
+    #   Plugin.register_guess("mysqldump_tab", self)
+    #
+    #   def guess(config, sample_buffer)
+    #     if sample_buffer[0,2] == GZIP_HEADER
+    #       guessed = {}
+    #       guessed["type"] = "mysqldump_tab"
+    #       guessed["property1"] = "guessed-value"
+    #       return {"parser" => guessed}
+    #     else
+    #       return {}
+    #     end
+    #   end
+    # end
+
+    # class MysqldumpTab < TextGuessPlugin
+    #   Plugin.register_guess("mysqldump_tab", self)
+    #
+    #   def guess_text(config, sample_text)
+    #     js = JSON.parse(sample_text) rescue nil
+    #     if js && js["mykeyword"] == "keyword"
+    #       guessed = {}
+    #       guessed["type"] = "mysqldump_tab"
+    #       guessed["property1"] = "guessed-value"
+    #       return {"parser" => guessed}
+    #     else
+    #       return {}
+    #     end
+    #   end
+    # end
+
+    # class MysqldumpTab < LineGuessPlugin
+    #   Plugin.register_guess("mysqldump_tab", self)
+    #
+    #   def guess_lines(config, sample_lines)
+    #     all_line_matched = sample_lines.all? do |line|
+    #       line =~ /mypattern/
+    #     end
+    #     if all_line_matched
+    #       guessed = {}
+    #       guessed["type"] = "mysqldump_tab"
+    #       guessed["property1"] = "guessed-value"
+    #       return {"parser" => guessed}
+    #     else
+    #       return {}
+    #     end
+    #   end
+    # end
+
+  end
+end

data/lib/embulk/parser/mysqldump_tab.rb

@@ -0,0 +1,105 @@
+module Embulk
+  module Parser
+
+    class MysqldumpTab < ParserPlugin
+
+      DUMMY_STRING         = "\v"
+      FIELDS_TERMINATED_BY = "\t"
+      FIELDS_ESCAPED_BY    = '\\'
+      FIELDS_ENCLOSED_BY   = ''
+      LINES_TERMINATED_BY  = "\n"
+
+      Plugin.register_parser("mysqldump_tab", self)
+
+      def self.transaction(config, &control)
+        # configuration code:
+
+        parser_task = config.load_config(Java::LineDecoder::DecoderTask)
+
+        task = {
+          "decoder_task" => DataSource.from_java(parser_task.dump)
+          # "option1" => config.param("option1", :integer),                     # integer, required
+          # "option2" => config.param("option2", :string, default: "myvalue"),  # string, optional
+          # "option3" => config.param("option3", :string, default: nil),        # string, optional
+        }
+
+        # https://github.com/treasure-data/embulk-input-jira/blob/master/lib/embulk/input/jira.rb#L22
+        attributes = {}
+        columns = config.param(:columns, :array).map do |column|
+          name = column["name"]
+          type = column["type"].to_sym
+          attributes[name] = type
+          Column.new(nil, name, type, column["format"])
+        end
+
+        task[:attributes] = attributes
+
+        # parser option
+        # task[:option1] = config['option1']
+        # task[:option1] = config.param(:option1, :integer, default: 5)
+
+        yield(task, columns)
+      end
+
+      def init
+        # initialization code:
+        # @option1 = task["option1"]
+        # @option2 = task["option2"]
+        # @option3 = task["option3"]
+
+        @decoder_task = task.param("decoder_task", :hash).load_task(Java::LineDecoder::DecoderTask)
+      end
+
+      def run(file_input)
+        decoder = Java::LineDecoder.new(file_input.instance_eval { @java_file_input }, @decoder_task)
+
+        while decoder.nextFile
+          buffer = ''
+          while line = decoder.poll
+            buffer = buffer + line
+            if in_column?(line)
+              buffer = buffer.gsub(/#{Regexp.escape(FIELDS_ESCAPED_BY)}/, LINES_TERMINATED_BY)
+              next
+            end
+            cols = parse_line(buffer)
+            page_builder.add(cols)
+            buffer = ''
+          end
+
+          # When output has not ended
+          if buffer.length > 0
+              cols = parse_line(buffer)
+              page_builder.add(cols)
+          end
+        end
+
+        page_builder.finish
+      end
+
+      def parse_line(line)
+        # Escape "escaped TAB" temporarily
+        line = line.gsub(/\\#{FIELDS_TERMINATED_BY}/, DUMMY_STRING)
+
+        # Split with separator (TAB)
+        cols = line.split(FIELDS_TERMINATED_BY)
+        cols.map! { |item| item.gsub(/#{DUMMY_STRING}/, FIELDS_TERMINATED_BY) }
+
+        len = task[:attributes].length
+        cols = adjust_column(cols, len)
+        return cols
+      end
+
+      def in_column?(line)
+        /#{Regexp.escape(FIELDS_ESCAPED_BY)}$/.match(line) ? true : false # escaped new line
+      end
+
+      # Adjust array length
+      def adjust_column(arr, len)
+        arr = arr.slice(0, len) # Truncate if more than len
+        arr.fill(0, len) { |i| arr[i] } # If it is less than len, fill it with nil
+      end
+
+    end
+
+  end
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: embulk-parser-mysqldump_tab
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- inouet
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-10-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: embulk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.23
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.23
+- !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: Embulk parser plugin for mysqldump file that dumped with the --tab option.
+email:
+- inudog@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .ruby-version
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-parser-mysqldump_tab.gemspec
+- lib/embulk/guess/mysqldump_tab.rb
+- lib/embulk/parser/mysqldump_tab.rb
+homepage: https://github.com/inouet/embulk-parser-mysqldump_tab
+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.0.14.1
+signing_key: 
+specification_version: 4
+summary: Mysqldump Tab parser plugin for Embulk
+test_files: []