embulk-parser-ltsv 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 33a6060131190c76380a815b6b1a89406158eed4
+  data.tar.gz: 85f2463f7fb53ba2e395393083ddcf52f198f935
+SHA512:
+  metadata.gz: cf27bffe35212b122a46e4d93526fbe8a1ed359c53fc9c8aac6ecfb53d53c7f7458e2512dbfd1d692515a8ed61864e98a2f90c0c48d8ad9c749211828cd1b423
+  data.tar.gz: f2a8d8e9c77aa26093f0b631f145b51b28c911dc7d627b3aeea13f082883b12fd3ad2063342fb734f0a538e456e3e1d06b7efcc474610a41b506c47423221d83

data/.gitignore

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

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,60 @@
+# Ltsv parser plugin for Embulk
+
+
+## Overview
+
+* **Plugin type**: parser
+* **Guess supported**: yes
+
+## Configuration
+
+- **option1**: description (integer, required)
+- **option2**: description (string, default: `"myvalue"`)
+- **null_value_pattern**: null value pattern. (string, default: `null`)
+
+## Example
+
+```yaml
+
+in:
+  type: file
+  path_prefix: /Users/toyama-h/access_log-20150616.ltsv.gz
+  decoders:
+  - {type: gzip}
+  parser:
+    type: ltsv
+    charset: UTF-8
+    newline: CRLF
+    null_value_pattern: ^(-|null|NULL)$
+    schema:
+      - {name: host, type: string}
+      - {name: ip_address, type: string}
+      - {name: server, type: string}
+      - {name: remote_user, type: string}
+      - {name: log_time, type: timestamp, time_format: '%d/%b/%Y:%H:%M:%S %z'}
+      - {name: method, type: string}
+      - {name: path, type: string}
+      - {name: protocol, type: string}
+      - {name: status, type: long}
+      - {name: size, type: string}
+      - {name: referer, type: string}
+      - {name: user_agent, type: string}
+      - {name: response_time, type: long}
+exec: {}
+out: {type: stdout}
+
+
+```
+
+(If guess supported) you don't have to write `parser:` section in the configuration file. After writing `in:` section, you can let embulk guess `parser:` section using this command:
+
+```
+$ embulk gem install embulk-parser-ltsv
+$ embulk guess -g ltsv config.yml -o guessed.yml
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-parser-ltsv.gemspec

@@ -0,0 +1,19 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-parser-ltsv"
+  spec.version       = "0.1.0"
+  spec.authors       = ["toyama0919"]
+  spec.summary       = "Ltsv parser plugin for Embulk"
+  spec.description   = "Parses Ltsv files read by other file input plugins."
+  spec.email         = ["toyama0919@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/toyama0919/embulk-parser-ltsv"
+
+  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 'bundler', ['~> 1.0']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/guess/ltsv.rb

@@ -0,0 +1,53 @@
+module Embulk
+  module Guess
+    class LtsvParserGuessPlugin < LineGuessPlugin
+      Plugin.register_guess("ltsv", self)
+
+      def guess_lines(config, sample_lines)
+        columns = {}
+        sample_lines.each do |line|
+          hash = {}
+          array = line.split("\t").each { |pair|
+            key, value = pair.split(":", 2)
+            hash[key] = value
+          }
+
+          hash.each do |k, v|
+            columns[k] = get_embulk_type(v)
+          end
+        end
+        schema = []
+        columns.each do |k,v|
+          schema << {'name' => k, 'type' => v}
+        end
+        guessed = {}
+        guessed["type"] = "ltsv"
+        guessed["schema"] = schema
+        return {"parser" => guessed}
+      end
+
+      private
+
+      def get_embulk_type(val)
+        if val =~ /^\d+\.\d+$/
+          return "double"
+        end
+
+        if val =~ /^\d+$/
+          return "long"
+        end
+
+        begin
+          Time.parse(val)
+          return "timestamp"
+        rescue => e
+        end
+
+        if val =~ /^(true|false)$/i
+          return "boolean"
+        end
+        return "string"
+      end
+    end
+  end
+end

data/lib/embulk/parser/ltsv.rb

@@ -0,0 +1,103 @@
+module Embulk
+  module Parser
+
+    class Ltsv < ParserPlugin
+      Plugin.register_parser("ltsv", self)
+
+      def self.transaction(config, &control)
+        parser_task = config.load_config(Java::LineDecoder::DecoderTask)
+        task = {
+          "decoder_task" => DataSource.from_java(parser_task.dump),
+          "schema" => config.param("schema", :array),
+          "null_value_pattern" => config.param("null_value_pattern", :string, default: nil),
+          "null_empty_string" => config.param("null_empty_string", :bool, default: false),
+          "delimiter" => config.param("delimiter", :string, default: "\t"),
+          "label_delimiter" => config.param("label_delimiter", :string, default: ":")
+        }
+        columns = task["schema"].each_with_index.map do |c, i|
+          Column.new(i, c["name"], c["type"].to_sym)
+        end
+        yield(task, columns)
+      end
+
+      def init
+        @delimiter = task["delimiter"]
+        @label_delimiter = task["label_delimiter"]
+        @null_value_pattern = task["null_value_pattern"] ? Regexp.new(task["null_value_pattern"]) : nil
+        @null_empty_string = task["null_empty_string"]
+        @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
+          while line = decoder.poll
+            begin
+              array = line.split(@delimiter).map { |pair|
+                pair.split(@label_delimiter, 2)
+              }
+              @page_builder.add(make_record(Hash[*array.flatten]))
+            rescue => e
+              puts "\n#{e.message}\n#{e.backtrace.join("\n")}"
+            end
+          end
+        end
+        page_builder.finish
+      end
+
+      private
+
+      def make_record(e)
+        @task["schema"].map do |c|
+          convert_value(e, c)
+        end
+      end
+
+      def convert_value(e, c)
+        v = convert_value_to_nil(e[c["name"]])
+        return nil if v.nil?
+        case c["type"]
+        when "string"
+          v
+        when "long"
+          v.to_i
+        when "double"
+          v.to_f
+        when "boolean"
+          ["yes", "true", "1"].include?(v.downcase)
+        when "timestamp"
+          if v.empty?
+            nil
+          else
+            c["time_format"] ? Time.strptime(v, c["time_format"]) : Time.parse(v)
+          end
+        else
+          raise "Unsupported type #{c['type']}"
+        end
+      end
+
+      def convert_value_to_nil(value)
+        if value and @null_empty_string
+          value = (value == '') ? nil : value
+        end
+        if value and @null_value_pattern
+          value = match_regexp(@null_value_pattern, value) ? nil : value
+        end
+        value
+      end
+
+      def match_regexp(regexp, string)    
+        begin    
+          return regexp.match(string)    
+        rescue ArgumentError => e    
+          raise e unless e.message.index("invalid byte sequence in".freeze).zero?    
+          log.info "invalid byte sequence is replaced in `#{string}`"    
+          string = string.scrub('?')   
+          retry    
+        end    
+        return true    
+      end    
+    end
+  end
+end

metadata

@@ -0,0 +1,81 @@
+--- !ruby/object:Gem::Specification
+name: embulk-parser-ltsv
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- toyama0919
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-08-21 00:00:00.000000000 Z
+dependencies:
+- !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: Parses Ltsv files read by other file input plugins.
+email:
+- toyama0919@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-parser-ltsv.gemspec
+- lib/embulk/guess/ltsv.rb
+- lib/embulk/parser/ltsv.rb
+homepage: https://github.com/toyama0919/embulk-parser-ltsv
+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.2
+signing_key: 
+specification_version: 4
+summary: Ltsv parser plugin for Embulk
+test_files: []
+has_rdoc: