embulk-input-redash 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d1d42839e0f5cf37c46e861248891cb07c0e8570
+  data.tar.gz: 79aa8f785f59ebcb2da89399e4187e95b7f54565
+SHA512:
+  metadata.gz: a65dd5913911bc6641cf0347be0181db291726dec4d3ca2db2845bda35889d1eab056f0a9e98e5155e9e0f07673c97fe991e4b54692644648837793462d04f75
+  data.tar.gz: 8344f3104c2ed9535369fbc20f226dbaa03e3f761c405e55e3fd91c26d21d17e5dc403485672c72d57a7b6dc29ed6b944360c91b20e35b40eb233b4e00bce473

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,30 @@
+# Redash input plugin for Embulk
+
+embulk-input-redash is the Embulk input plugin for [Redash](https://redash.io).
+
+## Overview
+
+* **Plugin type**: input
+* **Resume supported**: yes
+* **Cleanup supported**: yes
+* **Guess supported**: yes
+
+## Configuration
+
+- **url**: description (string, required)
+- **api_key**: description (string, required)
+
+## Example
+
+```yaml
+in:
+  type: redash
+  url: https://localhost:5000/api/queries/1234/results.json
+  api_key: YOUR_REDASH_API_KEY
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-input-redash.gemspec

@@ -0,0 +1,20 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-input-redash"
+  spec.version       = "0.1.0"
+  spec.authors       = ["ariarijp"]
+  spec.summary       = "Redash input plugin for Embulk"
+  spec.description   = "Loads records from Redash."
+  spec.email         = ["takuya.arita@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/ariarijp/embulk-input-redash"
+
+  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 'rest-client'
+  spec.add_development_dependency 'embulk', ['>= 0.8.17']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/input/redash.rb

@@ -0,0 +1,8 @@
+require 'embulk/input/redash/plugin'
+
+module Embulk
+  module Input
+    module Redash
+    end
+  end
+end

data/lib/embulk/input/redash/plugin.rb

@@ -0,0 +1,72 @@
+require 'rest-client'
+
+module Embulk
+  module Input
+    module Redash
+      class Plugin < InputPlugin
+        ::Embulk::Plugin.register_input('redash', self)
+
+        def self.transaction(config, &control)
+          task = {
+            'url' => config.param('url', :string),
+            'api_key' => config.param('api_key', :string)
+          }
+
+          columns = embulk_columns(config)
+
+          resume(task, columns, 1, &control)
+        end
+
+        def self.resume(task, columns, count)
+          task_reports = yield(task, columns, count)
+
+          next_config_diff = {}
+          next_config_diff
+        end
+
+        def self.guess(_config)
+          sample_records = get_rows(_config['url'], _config['api_key']).first(10)
+          columns = Guess::SchemaGuess.from_hash_records(sample_records)
+          { 'columns' => columns }
+        end
+
+        def init
+          @url = task['url']
+          @api_key = task['api_key']
+        end
+
+        def run
+          column_names = @schema.map(&:name)
+
+          Plugin.get_rows(@url, @api_key).each do |row|
+            page_builder.add(row.values_at(*column_names))
+          end
+
+          page_builder.finish
+
+          task_report = {}
+
+          task_report
+        end
+
+        def self.embulk_columns(config)
+          config.param(:columns, :array).map do |column|
+            name = column['name']
+            type = column['type'].to_sym
+
+            Column.new(nil, name, type, column['format'])
+          end
+        end
+
+        private
+
+        def self.get_rows(url, api_key)
+          res = RestClient.get(url, params: { 'api_key' => api_key })
+          data = JSON.parse(res.body)
+
+          data['query_result']['data']['rows']
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: embulk-input-redash
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- ariarijp
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-03-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: embulk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.17
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.17
+- !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 Redash.
+email:
+- takuya.arita@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-input-redash.gemspec
+- lib/embulk/input/redash.rb
+- lib/embulk/input/redash/plugin.rb
+homepage: https://github.com/ariarijp/embulk-input-redash
+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.8
+signing_key: 
+specification_version: 4
+summary: Redash input plugin for Embulk
+test_files: []