embulk-input-redis_custom_restock_noti 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: bee0463925a41ae1eb89d517a41eee3075f6a1b7ac169affef945743f41bf30e
+  data.tar.gz: d3b47ece081746d69226e5eef927516637a112a49d47407c5302e680becd87e3
+SHA512:
+  metadata.gz: 93b242aafe6689f021f2d19ecd11538fe8dfb28ccc7dda93e597d473f4853270edcd9192bad2dfb30b7b7a1e283614e08bf4e7e8f3f903d20152f3f76e61e97f
+  data.tar.gz: cb588865edaece99a651dc2369e22983729c83588ab975c07cae80fbb4cef5457687f935a4716649b915f7dbac5083ca47adef962bb9cf0735a606aee3c20dca

data/.gitignore

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

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.1.15.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,32 @@
+# Redis Custom Restock Noti input plugin for Embulk
+
+TODO: Write short description here and embulk-input-redis_custom_restock_noti.gemspec file.
+
+## Overview
+
+* **Plugin type**: input
+* **Resume supported**: yes
+* **Cleanup supported**: yes
+* **Guess supported**: no
+
+## Configuration
+
+- **option1**: description (integer, required)
+- **option2**: description (string, default: `"myvalue"`)
+- **option3**: description (string, default: `null`)
+
+## Example
+
+```yaml
+in:
+  type: redis_custom_restock_noti
+  option1: example1
+  option2: example2
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-input-redis_custom_restock_noti.gemspec

@@ -0,0 +1,21 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-input-redis_custom_restock_noti"
+  spec.version       = "0.1.0"
+  spec.authors       = ["iktae_lee"]
+  spec.summary       = "Redis Custom Restock Noti input plugin for Embulk"
+  spec.description   = "Loads records from Redis Custom Restock Noti."
+  spec.email         = ["iktae_lee@kolon.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://gitlab.kolonfnc.com/iktae_lee/embulk-input-redis_custom_restock_noti.git"
+
+  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_dependency 'redis', ['>= 3.0.5']
+  spec.add_development_dependency 'embulk', ['>= 0.9.23']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/input/redis_custom_restock_noti.rb

@@ -0,0 +1,88 @@
+module Embulk
+  module Input
+
+    class RedisCustomRestockNoti < InputPlugin
+      require 'redis'
+      
+      Plugin.register_input("redis_custom_restock_noti", self)
+
+      def self.transaction(config, &control)
+        task = {
+          'host' => config.param('host', :string, :default => 'localhost'),
+          'port' => config.param('port', :integer, :default => 6379),
+          'db' => config.param('db', :integer, :default => 0),
+          'keys_pattern' => config.param('keys_pattern', :string, :default => ''),
+          'columns' =>
+            config.param('columns', :array, :default => []).inject({}){|a, col|
+              a[col['name']] = col['type'].to_sym
+              a
+            },
+          'rows' => 0
+        }
+
+        columns = task['columns'].map.with_index{|(name, type), i|
+          Column.new(i, name, type)
+        }
+
+        puts "Redis input started."
+        commit_reports = yield(task, columns, 1)
+        puts "Redis input finished. Commit reports = #{commit_reports.to_json}"
+
+        return {}
+      end
+
+      # def self.resume(task, columns, count, &control)
+      #   task_reports = yield(task, columns, count)
+
+      #   next_config_diff = {}
+      #   return next_config_diff
+      # end
+
+      # TODO
+      # def self.guess(config)
+      #   sample_records = [
+      #     {"example"=>"a", "column"=>1, "value"=>0.1},
+      #     {"example"=>"a", "column"=>2, "value"=>0.2},
+      #   ]
+      #   columns = Guess::SchemaGuess.from_hash_records(sample_records)
+      #   return {"columns" => columns}
+      # end
+
+      # def init
+      #   # initialization code:
+      #   @option1 = task["option1"]
+      #   @option2 = task["option2"]
+      #   @option3 = task["option3"]
+      # end
+
+      def run
+        puts "Redis input thread #{@index}..."
+        r = Redis.new(:host => @task['host'], :port => @task['port'], :db => @task['db'])
+        r.keys("#{@task['keys_pattern']}").each do |key|
+
+          k = key.split(':')
+          v = r.zrange(key, 0, -1, :with_scores => true)
+        
+          v.each do |e|
+            temp = Array.new
+            temp.push(k[1]) # siteUid
+            temp.push(k[3].split('-')[0]) # productCode
+            temp.push(k[3].split('-')[1]) # size
+            temp.push(e[0]) # customerUid
+            temp.push(Time.at(e[1] / 1000)) # score(timestamp)
+            @page_builder.add(temp)
+          end
+
+          @task['rows'] += 1
+        end
+        @page_builder.finish  # don't forget to call finish :-)
+
+        commit_report = {
+          "rows" => @task['rows']
+        }
+        return commit_report
+      end
+    end
+
+  end
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification
+name: embulk-input-redis_custom_restock_noti
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- iktae_lee
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2021-07-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.5
+- !ruby/object:Gem::Dependency
+  name: embulk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.9.23
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.9.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: Loads records from Redis Custom Restock Noti.
+email:
+- iktae_lee@kolon.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-input-redis_custom_restock_noti.gemspec
+- lib/embulk/input/redis_custom_restock_noti.rb
+homepage: https://gitlab.kolonfnc.com/iktae_lee/embulk-input-redis_custom_restock_noti.git
+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: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Redis Custom Restock Noti input plugin for Embulk
+test_files: []