embulk-input-big-query-async 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: e1e61ab475f7f94c2420064083bb8faf71e03ca29c31342cd4f288fd0af323f6
+  data.tar.gz: a88983894e55334d9a09a45c239a493ed362ab71d0ecf6d46c6df5887b915eef
+SHA512:
+  metadata.gz: 7fd62497ffc4bbbd314368abe870a62ea3cf2746133361d572704ef906050f30907380436f640d7941ac71a4c15af41ebc40d94adfdd2140fcdd0cb2bc9aaa58
+  data.tar.gz: 0e2ff2c50ceba30a3c0c352898452257a6fb3b2a11d3df8695fad6a5703adbd5fab04979ef703df0cd1206200ef13ab3d0bd5f79ee36ebb268d2ebb3549ffcce

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in embulk-input-big-query-async.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2017 TODO: Write your name
+
+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,83 @@
+# Embulk::Input::Bigquery
+
+This is Embulk input plugin from Bigquery.
+
+## Installation
+
+install it yourself as:
+
+    $ embulk gem install embulk-input-big-query-async
+
+## Usage
+
+```
+in:
+  type: bigquery-async
+  project: 'project-name'
+  keyfile: '/home/hogehoge/bigquery-keyfile.json'
+  sql: 'SELECT price,category_id FROM [ecsite.products] GROUP BY category_id'
+  columns:
+    - {name: price, type: long}
+    - {name: category_id, type: string}
+  max: 2000
+  synchronous_method: true
+out:
+  type: stdout
+```
+
+If, table name is changeable, then
+
+```
+in:
+  type: bibigquery-asyncquery
+  project: 'project-name'
+  keyfile: '/home/hogehoge/bigquery-keyfile.json'
+  sql_erb: 'SELECT price,category_id FROM [ecsite.products_<%= params["date"].strftime("%Y%m")  %>] GROUP BY category_id'
+  erb_params:
+    date: "require 'date'; (Date.today - 1)"
+  columns:
+    - {name: price, type: long}
+    - {name: category_id, type: long}
+    - {name: month, type: timestamp, format: '%Y-%m', eval: 'require "time"; Time.parse(params["date"]).to_i'}
+```
+
+## Optional Configuration
+This plugin uses the gem [`google-cloud(Google Cloud Client Library for Ruby)`](https://github.com/GoogleCloudPlatform/google-cloud-ruby) and queries data using the synchronous method or the asynchronous  method.
+Therefore some optional configuration items comply with the Google Cloud Client Library.
+
+### optional bigquery parameter 
+
+The detail of follows params is [here](https://github.com/GoogleCloudPlatform/google-cloud-ruby/blob/master/google-cloud-bigquery/lib/google/cloud/bigquery/project.rb).
+
+- max :
+  - default value : **null** and null value is interpreted as no maximum row count in the Google Cloud Client Library. This param is supported only synchronous method.
+- cache :
+  - default value : **null** and null value is interpreted as true in the Google Cloud Client Library. 
+- timeout :
+  - default value : **null** and null value is interpreted as 10000 milliseconds in the Google Cloud Client Library. This param is supported only synchronous method.
+- dryrun :
+  - default value : **null** and null value is interpreted as false in the Google Cloud Client Library. This param is supported only synchronous method.
+- standard_sql :
+  - default value : **null** and null value is interpreted as true in the Google Cloud Client Library.
+- legacy_sql :
+  - default value : **null** and null value is interpreted as false in the Google Cloud Client Library.
+- large_results :
+  - default value : **null** and null value is interpreted as false in the Google Cloud Client Library. This param is supported only asynchronous method.
+- write : 
+  - default value : **null** and null value is interpreted as empty in the Google Cloud Client Library. This param is supported only asynchronous method.
+
+### the bigquery method
+Big query library in Google Cloud Client Library has [two methods](https://github.com/GoogleCloudPlatform/google-cloud-ruby/blob/master/google-cloud-bigquery/lib/google/cloud/bigquery/project.rb) for query.
+
+The default method in this plugin is synchronous_method.
+The logic which how select query method is [here](https://github.com/ykoyano/embulk-input-bigquery/blob/master/lib/embulk/input/bigquery.rb#L41).
+ 
+- synchronous_method:
+   - type : boolean
+   - default value : **null**
+   - This method uses `query` method in the Google Cloud Client Library.
+   - It should be noted that the number of records for `query` method is **limited**. Therefore, if you get many records, you should use `query_job` method with asynchronous_method option.
+- asynchronous_method:
+   - type : boolean
+   - default value : **null**
+   - This method uses `query_job` method in the Google Cloud Client Library.

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/embulk-input-big-query-async.gemspec

@@ -0,0 +1,24 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'embulk/input/big-query-async/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-input-big-query-async"
+  spec.version       = Embulk::Input::Bigqueryasync::VERSION
+  spec.authors       = ["Angelos Alexopoulos"]
+  spec.email         = ["alexopoulos7@gmail.com"]
+  spec.description   = %q{embulk input plugin from bigquery.}
+  spec.summary       = %q{Embulk input plugin from bigquery.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_dependency "google-cloud-bigquery", '~> 0.26.0'
+end

data/lib/embulk/input/big-query-async.rb

@@ -0,0 +1,146 @@
+require "embulk/input/bigquery/version"
+require "google/cloud/bigquery"
+require 'erb'
+
+module Embulk
+  module Input
+    class InputBigquery < InputPlugin
+			Plugin.register_input('big-query-async', self)
+
+			def self.transaction(config, &control)
+				sql = config[:sql]
+				params = {}
+				unless sql
+					sql_erb = config[:sql_erb]
+					erb = ERB.new(sql_erb)
+					erb_params = config[:erb_params]
+					erb_params.each do |k, v|
+						params[k] = eval(v)
+					end
+
+					sql = erb.result(binding)
+				end
+
+				task = {
+					project: config[:project],
+					keyfile: config[:keyfile],
+					sql: sql,
+					columns: config[:columns],
+					params: params,
+					synchronous_method: config[:synchronous_method],
+					asynchronous_method: config[:asynchronous_method],
+					dataset: config[:dataset],
+					table: config[:table],
+					option: {
+						cache: config[:cache],
+						standard_sql: config[:standard_sql],
+						legacy_sql: config[:legacy_sql],
+					}
+				}
+
+				if task[:synchronous_method] || !task[:asynchronous_method]
+					task[:option].merge!(
+						{
+							max: config[:max],
+							timeout: config[:timeout],
+							dryrun:  config[:dryrun],
+						}
+					)
+				else
+					task[:option].merge!(
+						{
+							large_results: config[:legacy_sql],
+							write: config[:write],
+						}
+					)
+				end
+
+				columns = []
+				config[:columns].each_with_index do |c, i|
+					columns << Column.new(i, c['name'], c['type'].to_sym)
+				end
+
+				yield(task, columns, 1)
+
+				return {}
+			end
+
+			def run
+				bq = Google::Cloud::Bigquery.new(project: @task[:project], keyfile: @task[:keyfile])
+				params = @task[:params]
+				@task[:columns] = values_to_sym(@task[:columns], 'name')
+				option = keys_to_sym(@task[:option])
+				if @task[:synchronous_method] || @task[:asynchronous_method].nil?
+					run_synchronous_query(bq, option)
+				else
+					if @task[:dataset]
+						dataset = bq.dataset(@task[:dataset])
+						option[:table] = dataset.table(@task[:table])
+						if option[:table].nil?
+							option[:table] = dataset.create_table(@task[:table])
+						end
+					end
+					run_asynchronous_query(bq, option)
+				end
+				@page_builder.finish
+				return {}
+			end
+
+			def run_synchronous_query(bq, option)
+				rows = bq.query(@task[:sql], **option)
+				rows.each do |row|
+					record = extract_record(row)
+					@page_builder.add(record)
+				end
+			end
+
+			def run_asynchronous_query(bq, option)
+				job = bq.query_job(@task[:sql], **option)
+				job.wait_until_done!
+				return {} if job.failed?
+				results = job.query_results
+				while results
+					results.each do |row|
+						record = extract_record(row)
+						@page_builder.add(record)
+					end
+					results = results.next
+				end
+			end
+
+			def extract_record(row)
+				columns = []
+				@task[:columns].each do |c|
+					val = row[c['name']]
+					if c['eval']
+						val = eval(c['eval'], binding)
+					end
+					columns << val
+				end
+				return columns
+			end
+
+			def values_to_sym(hashs, key)
+				hashs.map do |h|
+					h[key] = h[key].to_sym
+					h
+				end
+			end
+
+			def keys_to_sym(hash)
+				ret = {}
+				hash.each do |key, value|
+					ret[key.to_sym] = value
+				end
+				ret
+			end
+
+                        def values_to_sym(hashs, key)
+				hashs.map do |h|
+					h[key] = h[key].to_sym
+					h
+				end
+			end
+    end
+  end
+end

data/lib/embulk/input/big-query-async/version.rb

@@ -0,0 +1,7 @@
+module Embulk
+  module Input
+    module Bigqueryasync
+      VERSION = "0.0.2"
+    end
+  end
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification
+name: embulk-input-big-query-async
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Angelos Alexopoulos
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-04-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: google-cloud-bigquery
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.26.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.26.0
+description: embulk input plugin from bigquery.
+email:
+- alexopoulos7@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-input-big-query-async.gemspec
+- lib/embulk/input/big-query-async.rb
+- lib/embulk/input/big-query-async/version.rb
+- pkg/embulk-input-big-query-async-0.0.1.gem
+homepage: ''
+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: Embulk input plugin from bigquery.
+test_files: []