kura 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: df031ec7381a9dfbb95f0f57a937358131add7cb
+  data.tar.gz: 873e797a516da04f5c3f7672c6ce3a02336c809a
+SHA512:
+  metadata.gz: b8e971c7582ee24f13c98b31f8129439b38831074966217e6174ea94e1a932cf1346a222b30f2f9e72a27a91f9c059c173eddbce3dceddc6428afbae609e1c61
+  data.tar.gz: fdcbabf8bb94ba1016236fdce57cdfda61fadab796e345665e37ed3cb13e87ea08f2f1854d2ed29009b95077a7b4fde8e1e65845e11a0e360d24e6f8d99a402a

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/vendor/bundle/
+/test/data/service_account.json

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 2.2.2
+before_install: gem install bundler -v 1.10.5

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in kura.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2015 Chikanaga Tomoyuki
+
+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,39 @@
+# Kura
+
+Kura is an interface to the [BigQuery API v2](https://cloud.google.com/bigquery/docs/reference/v2/).
+This is a wrapper of google-api-client.gem.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'kura'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install kura
+
+## Usage
+
+```
+client = Kura.client(project_id, email, private_key)
+client.load("dataset", "table", "gs://mybucket/data.csv", wait: 120)
+client.query("dataset", "result", "SELECT * FROM [dataset.table];", wait: 120)
+client.extract("dataset", "result", "gs://mybucket/extracted.csv", wait: 120)
+```
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/nagachika/kura.
+
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+

data/Rakefile

@@ -0,0 +1,10 @@
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.libs << "lib"
+  t.test_files = FileList['test/**/*_test.rb']
+end
+
+task :default => :test

data/bin/console

@@ -0,0 +1,13 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "kura"
+
+service_account_json = File.expand_path("../../test/data/service_account.json", __FILE__)
+if File.readable?(service_account_json)
+  obj = JSON.parse(File.binread(service_account_json))
+  @client = Kura.client(obj["project_id"], obj["client_email"], obj["private_key"])
+end
+
+require "pry"
+Pry.start(@client)

data/kura.gemspec

@@ -0,0 +1,33 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'kura/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "kura"
+  spec.version       = Kura::VERSION
+  spec.authors       = ["Chikanaga Tomoyuki"]
+  spec.email         = ["nagachika@ruby-lang.org"]
+
+  spec.summary       = %q{Interface to BigQuery API v2.}
+  spec.description   = %q{Kura is an interfece to BigQUery API v2. It is a wrapper of google-api-client.}
+  spec.homepage      = "https://github.com/nagachika/kura/"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+  spec.has_rdoc      = false
+
+  spec.required_ruby_version = '>= 2.1'
+
+  spec.add_runtime_dependency "google-api-client", "~> 0.8.5"
+
+  spec.add_development_dependency "bundler", "~> 1.10"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "test-unit"
+  spec.add_development_dependency "simplecov"
+  spec.add_development_dependency "test-unit-power_assert"
+  spec.add_development_dependency "pry"
+end

data/lib/kura.rb

@@ -0,0 +1,41 @@
+# coding: utf-8
+
+require "kura/version"
+require "kura/client"
+
+module Kura
+  class ApiError < RuntimeError
+    def initialize(reason, message)
+      @reason = reason
+      super(message)
+    end
+    attr_reader :reason
+  end
+
+  class TimeoutError < RuntimeError
+  end
+
+  def self.get_private_key(private_key)
+    private_key_file = nil
+    private_key_content = nil
+    if private_key.respond_to?(:to_path)
+      private_key_file = private_key.to_path
+    elsif private_key.is_a?(String) and File.readable?(private_key)
+      private_key_file = private_key
+    end
+    if private_key_file
+      private_key_content = File.binread(private_key_file)
+    else
+      private_key_content = private_key
+    end
+    if private_key_content
+      private_key = OpenSSL::PKey.read(private_key_content)
+    end
+    private_key
+  end
+
+  def self.client(project_id, email_address, private_key)
+    private_key = get_private_key(private_key)
+    self::Client.new(project_id, email_address, private_key)
+  end
+end

data/lib/kura/client.rb

@@ -0,0 +1,219 @@
+# coding: utf-8
+
+require "google/api_client"
+require "kura/version"
+
+module Kura
+  class Client
+    def initialize(project_id, email_address, private_key)
+      @project_id = project_id
+      @scope = "https://www.googleapis.com/auth/bigquery"
+      @email_address = email_address
+      @private_key = private_key
+      auth = Signet::OAuth2::Client.new(
+        token_credential_uri: "https://accounts.google.com/o/oauth2/token",
+        audience: "https://accounts.google.com/o/oauth2/token",
+        scope: @scope,
+        issuer: @email_address,
+        signing_key: @private_key)
+      @api = Google::APIClient.new(application_name: "Kura", application_version: Kura::VERSION, authorization: auth)
+      @api.authorization.fetch_access_token!
+      @bigquery_api = @api.discovered_api("bigquery", "v2")
+    end
+
+    def datasets(project_id: @project_id, all: false, limit: 1000)
+      r = @api.execute(api_method: @bigquery_api.datasets.list, parameters: { projectId: project_id, all: all, maxResult: limit })
+      unless r.success?
+        raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+      end
+      r.data.datasets
+    end
+
+    def dataset(dataset_id, project_id: @project_id)
+      r = @api.execute(api_method: @bigquery_api.datasets.get, parameters: { projectId: project_id, datasetId: dataset_id })
+      unless r.success?
+        if r.data.error["code"] == 404
+          return nil
+        else
+          raise Kura::ApiError.new(r.data.error["reason"], r.data.error["message"])
+        end
+      end
+      r.data
+    end
+
+    def insert_dataset(dataset_id)
+      r = @api.execute(api_method: @bigquery_api.datasets.insert, parameters: { projectId: @project_id }, body_object: { datasetReference: { datasetId: dataset_id } })
+      unless r.success?
+        raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+      end
+      r.data
+    end
+
+    def delete_dataset(dataset_id, delete_contents: false)
+      r = @api.execute(api_method: @bigquery_api.datasets.delete, parameters: { projectId: @project_id, datasetId: dataset_id, deleteContents: delete_contents })
+      unless r.success?
+        raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+      end
+      r.data
+    end
+
+    def patch_dataset(dataset_id, project_id: @project_id, access: nil, description: nil, default_table_expiration_ms: nil, friendly_name: nil )
+      body = {}
+      body["access"] = access if access
+      body["defaultTableExpirationMs"] = default_table_expiration_ms if default_table_expiration_ms
+      body["description"] = description if description
+      body["friendlyName"] = friendly_name if friendly_name
+      r = @api.execute(api_method: @bigquery_api.datasets.patch, parameters: { projectId: project_id, datasetId: dataset_id }, body_object: body)
+      unless r.success?
+        raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+      end
+      r.data
+    end
+
+    def table(dataset_id, table_id, project_id: @project_id)
+      params = { projectId: project_id, datasetId: dataset_id, tableId: table_id }
+      r = @api.execute(api_method: @bigquery_api.tables.get, parameters: params)
+      unless r.success?
+        if r.data["error"]["code"] == 404
+          return nil
+        else
+          raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+        end
+      end
+      r.data
+    end
+
+    def delete_table(dataset_id, table_id)
+      params = { projectId: @project_id, datasetId: dataset_id, tableId: table_id }
+      r = @api.execute(api_method: @bigquery_api.tables.delete, parameters: params)
+      unless r.success?
+        if r.data["error"]["code"] == 404
+          return nil
+        else
+          raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+        end
+      end
+      r.data
+    end
+
+    def mode_to_write_disposition(mode)
+      unless %i{ append truncate empty }.include?(mode)
+        raise "mode option should be one of :append, :truncate, :empty but #{mode}"
+      end
+      "WRITE_#{mode.to_s.upcase}"
+    end
+    private :mode_to_write_disposition
+
+    def insert_job(configuration, wait: nil)
+      params = { projectId: @project_id }
+      body = { configuration: configuration }
+      r = @api.execute(api_method: @bigquery_api.jobs.insert, parameters: params, body_object: body)
+      unless r.success?
+        raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+      end
+      if wait
+        wait_job(r.data.jobReference.jobId, wait)
+      else
+        r.data.jobReference.jobId
+      end
+    end
+
+    def query(dataset_id, table_id, sql, mode: :truncate, allow_large_result: true, wait: nil)
+      write_disposition = mode_to_write_disposition(mode)
+      configuration = {
+        query: {
+          query: sql,
+          destinationTable: { projectId: @project_id, datasetId: dataset_id, tableId: table_id },
+          writeDisposition: write_disposition,
+          allowLargeResults: allow_large_result,
+        }
+      }
+      insert_job(configuration, wait: wait)
+    end
+
+    def load(dataset_id, table_id, source_uris, schema, delimiter: ",", mode: :append, wait: nil)
+      write_disposition = mode_to_write_disposition(mode)
+      source_uris = [source_uris] if source_uris.is_a?(String)
+      configuration = {
+        load: {
+          sourceUris: source_uris,
+          destinationTable: {
+            projectId: @project_id,
+            datasetId: dataset_id,
+            tableId: table_id,
+          },
+          fieldDelimiter: delimiter,
+          writeDisposition: write_disposition,
+          schema: { fields: schema },
+        }
+      }
+      insert_job(configuration, wait: wait)
+    end
+
+    def extract(dataset_id, table_id, dest_uris, wait: nil)
+      dest_uris = [ dest_uris ] if dest_uris.is_a?(String)
+      configuration = {
+        extract: {
+          sourceTable: {
+            projectId: @project_id,
+            datasetId: dataset_id,
+            tableId: table_id,
+          },
+          destinationUris: dest_uris,
+        }
+      }
+      insert_job(configuration, wait: wait)
+    end
+
+    def copy(src_dataset_id, src_table_id, dest_dataset_id, dest_table_id, mode: :truncate, wait: nil)
+      write_disposition = mode_to_write_disposition(mode)
+      configuration = {
+        copy: {
+          destinationTable: {
+            projectId: @project_id,
+            datasetId: dest_dataset_id,
+            tableId: dest_table_id,
+          },
+          sourceTable: {
+            projectId: @project_id,
+            datasetId: src_dataset_id,
+            tableId: src_table_id,
+          },
+          writeDisposition: write_disposition,
+        }
+      }
+      insert_job(configuration, wait: wait)
+    end
+
+    def job(job_id)
+      params = { projectId: @project_id, jobId: job_id }
+      r = @api.execute(api_method: @bigquery_api.jobs.get, parameters: params)
+      unless r.success?
+        raise Kura::ApiError.new(r.data["error"]["reason"], r.data["error"]["message"])
+      end
+      r.data
+    end
+
+    def job_finished?(job_id)
+      r = job(job_id)
+      if r.status.state == "DONE"
+        if r.status["errorResult"]
+          raise Kura::ApiError.new(r.status.errorResult.reason, r.status.errorResult.message)
+        end
+        return true
+      end
+      return false
+    end
+
+    def wait_job(job_id, timeout=60*10)
+      expire = Time.now + timeout
+      while expire > Time.now
+        if job_finished?(job_id)
+          return true
+        end
+        sleep 1
+      end
+      raise Kura::TimeoutError, "wait job timeout"
+    end
+  end
+end

data/lib/kura/version.rb

@@ -0,0 +1,3 @@
+module Kura
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,153 @@
+--- !ruby/object:Gem::Specification
+name: kura
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Chikanaga Tomoyuki
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-07-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: google-api-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.5
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !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'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  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: simplecov
+  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: test-unit-power_assert
+  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: pry
+  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'
+description: Kura is an interfece to BigQUery API v2. It is a wrapper of google-api-client.
+email:
+- nagachika@ruby-lang.org
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- kura.gemspec
+- lib/kura.rb
+- lib/kura/client.rb
+- lib/kura/version.rb
+homepage: https://github.com/nagachika/kura/
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.1'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Interface to BigQuery API v2.
+test_files: []