embulk-parser-mahout 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 2d007c978b923dcd51d399d4fd3af4f838656e56
+  data.tar.gz: cf4c38a4b7e77c77d5df8eee97209dfd7ea14fa2
+SHA512:
+  metadata.gz: 175df85396025576190e56c1830e301723618a27bf9bd8b50e6d26f337a25f59023bc752a9d46bc9b570651b6e57c589dc34ab0eaad74d8ccb098464c8a939dc
+  data.tar.gz: 2ffc1386778f9e76802b49341b2487a67cdcc9d4ee770116f1d70549504f1e5cc16d94a202c9d9fba4b080b0630c808907875ec3fb3633c15e755dfd77311f65

data/.gitignore

@@ -0,0 +1,8 @@
+*~
+/pkg/
+/tmp/
+/.bundle/
+/Gemfile.lock
+/.idea
+/*.iml
+

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,38 @@
+# Mahout parser plugin for Embulk
+
+TODO: Write short description here and embulk-parser-mahout.gemspec file.
+
+## Overview
+
+* **Plugin type**: parser
+* **Guess supported**: no
+
+## Configuration
+
+- **option1**: description (integer, required)
+- **option2**: description (string, default: `"myvalue"`)
+- **option3**: description (string, default: `null`)
+
+## Example
+
+```yaml
+in:
+  type: any file input plugin type
+  parser:
+    type: mahout
+    option1: example1
+    option2: example2
+```
+
+(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-mahout
+$ embulk guess -g mahout config.yml -o guessed.yml
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-parser-mahout.gemspec

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

data/lib/embulk/guess/mahout.rb

@@ -0,0 +1,63 @@
+module Embulk
+  module Guess
+
+    # TODO implement guess plugin to make this command work:
+    #      $ embulk guess -g "mahout" 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/mahout.rb"
+
+    # class Mahout < GuessPlugin
+    #   Plugin.register_guess("mahout", self)
+    #
+    #   def guess(config, sample_buffer)
+    #     if sample_buffer[0,2] == GZIP_HEADER
+    #       guessed = {}
+    #       guessed["type"] = "mahout"
+    #       guessed["property1"] = "guessed-value"
+    #       return {"parser" => guessed}
+    #     else
+    #       return {}
+    #     end
+    #   end
+    # end
+
+    # class Mahout < TextGuessPlugin
+    #   Plugin.register_guess("mahout", self)
+    #
+    #   def guess_text(config, sample_text)
+    #     js = JSON.parse(sample_text) rescue nil
+    #     if js && js["mykeyword"] == "keyword"
+    #       guessed = {}
+    #       guessed["type"] = "mahout"
+    #       guessed["property1"] = "guessed-value"
+    #       return {"parser" => guessed}
+    #     else
+    #       return {}
+    #     end
+    #   end
+    # end
+
+    # class Mahout < LineGuessPlugin
+    #   Plugin.register_guess("mahout", 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"] = "mahout"
+    #       guessed["property1"] = "guessed-value"
+    #       return {"parser" => guessed}
+    #     else
+    #       return {}
+    #     end
+    #   end
+    # end
+
+  end
+end

data/lib/embulk/parser/mahout.rb

@@ -0,0 +1,50 @@
+module Embulk
+  module Parser
+
+    class Mahout < ParserPlugin
+      Plugin.register_parser("mahout", self)
+
+      def self.transaction(config, &control)
+        # configuration code:
+        task = {
+          "command" => config.param("command", :string, default: "recommenditembased"),                     # integer, required
+          "schema" => config.param("schema", :array)
+        }
+
+        columns = task["schema"].each_with_index.map do |col, index|
+          Column.new(index, col["name"], col["type"].to_sym)
+        end
+
+        yield(task, columns)
+      end
+
+      def init
+        @command = task["command"]
+        @col = task["columns"]
+      end
+
+      def run(file_input)
+        while file = file_input.next_file
+
+          text = file.read
+          text.each_line do |row|
+
+            record = []
+
+            user_id = row.match(/^\d*/)[0]
+            record.push(user_id)
+            recommend_items = row.scan(/(\d*):(\d*\.\d*)/)
+            recommend_items.each do |item|
+              record += [item[0], item[1]]
+            end
+            page_builder.add(record)
+
+          end
+
+        end
+        page_builder.finish
+      end
+    end
+
+  end
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: embulk-parser-mahout
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- kihengk
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2017-09-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.31
+  name: embulk
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.31
+- !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: Parses Mahout files read by other file input plugins.
+email:
+- kihengk@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-parser-mahout.gemspec
+- lib/embulk/guess/mahout.rb
+- lib/embulk/parser/mahout.rb
+homepage: https://github.com/kihengk/embulk-parser-mahout
+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.6.6
+signing_key:
+specification_version: 4
+summary: Mahout parser plugin for Embulk
+test_files: []