rbbigquery 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d3c6dccb9ac37ba7d802bb76407111aaa8950abe
+  data.tar.gz: 82bde743df55391a5fd3f73ec9ddab3ae5cf00ad
+SHA512:
+  metadata.gz: 96674a56caf964edc94456d00668683e42b594738ef0a6773a9d8712d36e4b7273333adbe7fac1601e2757f62d7a67703a6e04056ee8fd4320cf1423d4e97fea
+  data.tar.gz: 64dfd128bc5a4824f823abad8293a550347f59f371e3aef170b2a500801ffe0537f91fbc8b55d0dea5350d685e5cc172a34a645124198d91b842daf6ae4e6685

data/lib/rbbigquery/client.rb

@@ -0,0 +1,35 @@
+module RbBigQuery
+  class Client
+    attr_accessor :client, :project_id, :bq
+
+    # @params opts [Hash] {:application_name, :application_version, :key_path, :service_email, :project_id}
+    def initialize(opts = {})
+      @client = Google::APIClient.new(
+          application_name: opts[:application_name],
+          application_version: opts[:application_version]
+      )
+
+      key = Google::APIClient::PKCS12.load_key(File.open(
+                                                   opts[:key_path], mode: 'rb'),
+                                               "notasecret"
+      )
+
+      @asserter = Google::APIClient::JWTAsserter.new(
+          opts[:service_email],
+          "https://www.googleapis.com/auth/bigquery",
+          key
+      )
+
+      @client.authorization = @asserter.authorize
+
+      @bq = @client.discovered_api("bigquery", "v2")
+
+      @project_id = opts[:project_id]
+    end
+
+    # @return [RbBigQuery::Table]
+    def find_or_create_table(dataset, table_id, schema)
+      RbBigQuery::Table.new(self, dataset, table_id, schema)
+    end
+  end
+end

data/lib/rbbigquery/schema.rb

@@ -0,0 +1,32 @@
+module RbBigQuery
+  # Schema Builder
+  class Schema
+    attr_accessor :schema
+
+    class << self
+      # Builds schema for BigQuery
+      # @param &blk [Proc] RbBigQuery schema DSL
+      # @return [Array<Hash>]
+      def build(&blk)
+        instance = new
+        instance.schema = []
+        instance.instance_eval &blk
+        instance.schema
+      end
+    end
+
+    def string(name)
+      self.schema.push({
+        type: 'STRING',
+        name: name
+      })
+    end
+
+    def integer(name)
+      self.schema.push({
+        type: 'INTEGER',
+        name: name
+      })
+    end
+  end
+end

data/lib/rbbigquery/table.rb

@@ -0,0 +1,60 @@
+module RbBigQuery
+  class Table
+    attr_accessor :schema
+
+    def initialize(client, dataset, table_id, schema)
+      @client = client
+      @dataset = dataset
+      @table_id = table_id
+      @schema = schema
+
+      create
+    end
+
+    # @return row response json
+    def create
+      response = @client.client.execute({
+         :api_method => @client.bq.tables.insert,
+         :parameters => {
+             'projectId' => @client.project_id,
+             'datasetId' => @dataset
+         },
+         :body_object => {
+             'tableReference' => {
+                 'projectId' => @client.project_id,
+                 'datasetId' => @dataset,
+                 'tableId'   => @table_id
+             },
+             'schema' => {
+                 'fields' => @schema
+             }
+         }
+     })
+
+      JSON.parse(response.body)
+    end
+
+    # insert rows
+    # @param rows [Array<Hash>] [{#{column_name}=>value}]
+    # @return row response json
+    def insert(rows)
+      rows = rows.map do |row|
+        {'json' => row}
+      end
+
+      response = @client.client.execute({
+         :api_method => @client.bq.tabledata.insert_all,
+         :parameters => {
+             'projectId' => @client.project_id,
+             'datasetId' => @dataset,
+             'tableId' => @table_id,
+         },
+         :body_object => {
+             "rows" => rows
+         }
+      })
+
+      JSON.parse(response.body)
+    end
+  end
+end

data/lib/rbbigquery.rb

@@ -0,0 +1,6 @@
+require 'google/api_client'
+require 'json'
+
+require_relative 'rbbigquery/client'
+require_relative 'rbbigquery/table'
+require_relative 'rbbigquery/schema'

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: rbbigquery
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Kai Inui
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-07-03 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'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: "[WIP ]Provides a easier way to handle BigQuery with Ruby."
+email: me@kaiinui.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/rbbigquery.rb
+- lib/rbbigquery/client.rb
+- lib/rbbigquery/schema.rb
+- lib/rbbigquery/table.rb
+homepage: https://github.com/kaiinui/rbbigquery
+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.3.0
+signing_key: 
+specification_version: 4
+summary: "[WIP] Ruby BigQuery client."
+test_files: []
+has_rdoc: