rbbigquery 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d3c6dccb9ac37ba7d802bb76407111aaa8950abe
-  data.tar.gz: 82bde743df55391a5fd3f73ec9ddab3ae5cf00ad
+  metadata.gz: 0a38bb0cb98e9729c1911cc9691c8caf95770be9
+  data.tar.gz: e6c8d395dce39e7815499f47a8b5aff5b9f11e7d
 SHA512:
-  metadata.gz: 96674a56caf964edc94456d00668683e42b594738ef0a6773a9d8712d36e4b7273333adbe7fac1601e2757f62d7a67703a6e04056ee8fd4320cf1423d4e97fea
-  data.tar.gz: 64dfd128bc5a4824f823abad8293a550347f59f371e3aef170b2a500801ffe0537f91fbc8b55d0dea5350d685e5cc172a34a645124198d91b842daf6ae4e6685
+  metadata.gz: 2ae2996daebbac4462844c0db8f6ba2748c9680d66829c62cc4282383e344b600621f2b1df2941797ebf62d1ebb101587b865f2821c373b49202d8ceceb7cb11
+  data.tar.gz: 4429b951a3792db9fff2f57396ef9cbcefedf4b69900e64e404844b2f27074ecd61488781ea35f2170492d065fe1dfd428826501c7acad7120afdf438f6b8df0

data/lib/rbbigquery/client.rb

@@ -8,28 +8,91 @@ module RbBigQuery
           application_name: opts[:application_name],
           application_version: opts[:application_version]
       )
+      @bq = @client.discovered_api("bigquery", "v2")
+      @project_id = opts[:project_id]
+      authorize(opts[:service_email], opts[:key_path])
+    end
 
-      key = Google::APIClient::PKCS12.load_key(File.open(
-                                                   opts[:key_path], mode: 'rb'),
-                                               "notasecret"
-      )
+    # @return [RbBigQuery::Table]
+    def find_or_create_table(dataset, table_id, schema)
+      RbBigQuery::Table.new(self, dataset, table_id, schema)
+    end
+
+    # Executes provided query.
+    # @param [String] query
+    # @return [String] row response string
+    def query(query)
+      response = @client.execute({
+          :api_method => @bq.jobs.query,
+          :parameters => {
+              'projectId' => @project_id,
+          },
+          :body_object => {
+              'query' => query
+          }
+      })
+
+      build_rows_from_response(response)
+    end
 
-      @asserter = Google::APIClient::JWTAsserter.new(
-          opts[:service_email],
+    private
+
+    def authorize(service_email, key_path)
+      key = Google::APIClient::PKCS12.load_key(File.open(key_path, mode: 'rb'), "notasecret")
+
+      asserter = Google::APIClient::JWTAsserter.new(
+          service_email,
           "https://www.googleapis.com/auth/bigquery",
           key
       )
 
-      @client.authorization = @asserter.authorize
-
-      @bq = @client.discovered_api("bigquery", "v2")
-
-      @project_id = opts[:project_id]
+      @client.authorization = asserter.authorize
     end
 
-    # @return [RbBigQuery::Table]
-    def find_or_create_table(dataset, table_id, schema)
-      RbBigQuery::Table.new(self, dataset, table_id, schema)
+    # Sample response
+    #
+    #{"kind"=>"bigquery#queryResponse",
+    #"schema"=>
+    #    {"fields"=>
+    #         [{"name"=>"screen_name", "type"=>"STRING", "mode"=>"NULLABLE"},
+    #          {"name"=>"text", "type"=>"STRING", "mode"=>"NULLABLE"}]},
+    #    "jobReference"=>
+    #    {"projectId"=>"#{SOME_PROJECTID}", "jobId"=>"#{SOME_JOBID}"},
+    #    "totalRows"=>"15",
+    #    "rows"=>
+    #    [{"f"=>[{"v"=>"huga"}, {"v"=>"text: 5"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 2"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 4"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 3"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 3"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 1"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 1"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 4"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 2"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 5"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 5"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 3"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 1"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 4"}]},
+    #     {"f"=>[{"v"=>"huga"}, {"v"=>"text: 2"}]}],
+    #    "totalBytesProcessed"=>"225",
+    #    "jobComplete"=>true,
+    #    "cacheHit"=>false
+    #}
+    # @return [Array<Hash>]
+    def build_rows_from_response(response)
+      return unless response
+      body = JSON.parse(response.body)
+      schema = body["schema"]["fields"]
+
+      body["rows"].map do |row|
+        row_hash = {}
+        row["f"].each_with_index do |field, index|
+          name = schema[index]["name"]
+          row_hash[name] = field["v"]
+        end
+        row_hash
+      end
     end
   end
 end

data/lib/rbbigquery/table.rb

@@ -11,7 +11,12 @@ module RbBigQuery
       create
     end
 
-    # @return row response json
+    # @return [String] GQL style table name. (dataset.table_id)
+    def sql_name
+      "#{@dataset}.#{@table_id}"
+    end
+
+    # @return [Hash] row response json
     def create
       response = @client.client.execute({
          :api_method => @client.bq.tables.insert,
@@ -36,11 +41,9 @@ module RbBigQuery
 
     # insert rows
     # @param rows [Array<Hash>] [{#{column_name}=>value}]
-    # @return row response json
+    # @return [Hash] row response json
     def insert(rows)
-      rows = rows.map do |row|
-        {'json' => row}
-      end
+      rows = rows.map { |row| {'json' => row} }
 
       response = @client.client.execute({
          :api_method => @client.bq.tabledata.insert_all,

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rbbigquery
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Kai Inui