google-cloud-spanner 1.16.1 → 2.3.0

data/lib/google/cloud/spanner/commit.rb

@@ -96,8 +96,8 @@ module Google
           rows.delete_if(&:nil?)
           rows.delete_if(&:empty?)
           @mutations += rows.map do |row|
-            Google::Spanner::V1::Mutation.new(
-              insert_or_update: Google::Spanner::V1::Mutation::Write.new(
+            V1::Mutation.new(
+              insert_or_update: V1::Mutation::Write.new(
                 table: table, columns: row.keys.map(&:to_s),
                 values: [Convert.object_to_grpc_value(row.values).list_value]
               )
@@ -154,8 +154,8 @@ module Google
           rows.delete_if(&:nil?)
           rows.delete_if(&:empty?)
           @mutations += rows.map do |row|
-            Google::Spanner::V1::Mutation.new(
-              insert: Google::Spanner::V1::Mutation::Write.new(
+            V1::Mutation.new(
+              insert: V1::Mutation::Write.new(
                 table: table, columns: row.keys.map(&:to_s),
                 values: [Convert.object_to_grpc_value(row.values).list_value]
               )
@@ -211,8 +211,8 @@ module Google
           rows.delete_if(&:nil?)
           rows.delete_if(&:empty?)
           @mutations += rows.map do |row|
-            Google::Spanner::V1::Mutation.new(
-              update: Google::Spanner::V1::Mutation::Write.new(
+            V1::Mutation.new(
+              update: V1::Mutation::Write.new(
                 table: table, columns: row.keys.map(&:to_s),
                 values: [Convert.object_to_grpc_value(row.values).list_value]
               )
@@ -270,8 +270,8 @@ module Google
           rows.delete_if(&:nil?)
           rows.delete_if(&:empty?)
           @mutations += rows.map do |row|
-            Google::Spanner::V1::Mutation.new(
-              replace: Google::Spanner::V1::Mutation::Write.new(
+            V1::Mutation.new(
+              replace: V1::Mutation::Write.new(
                 table: table, columns: row.keys.map(&:to_s),
                 values: [Convert.object_to_grpc_value(row.values).list_value]
               )
@@ -306,8 +306,8 @@ module Google
         #
         def delete table, keys = []
           @mutations += [
-            Google::Spanner::V1::Mutation.new(
-              delete: Google::Spanner::V1::Mutation::Delete.new(
+            V1::Mutation.new(
+              delete: V1::Mutation::Delete.new(
                 table: table, key_set: key_set(keys)
               )
             )
@@ -323,20 +323,20 @@ module Google
         protected
 
         def key_set keys
-          return Google::Spanner::V1::KeySet.new all: true if keys.nil?
+          return V1::KeySet.new all: true if keys.nil?
           keys = [keys] unless keys.is_a? Array
-          return Google::Spanner::V1::KeySet.new all: true if keys.empty?
+          return V1::KeySet.new all: true if keys.empty?
           if keys_are_ranges? keys
             key_ranges = keys.map do |r|
               Convert.to_key_range r
             end
-            return Google::Spanner::V1::KeySet.new ranges: key_ranges
+            return V1::KeySet.new ranges: key_ranges
           end
           key_list = keys.map do |key|
             key = [key] unless key.is_a? Array
             Convert.object_to_grpc_value(key).list_value
           end
-          Google::Spanner::V1::KeySet.new keys: key_list
+          V1::KeySet.new keys: key_list
         end
 
         def keys_are_ranges? keys

data/lib/google/cloud/spanner/commit_response.rb

@@ -0,0 +1,87 @@
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+require "google/cloud/spanner/commit_response/commit_stats"
+
+module Google
+  module Cloud
+    module Spanner
+      ##
+      # CommitResponse is a timestamp at which the transaction committed
+      # with additional attributes of commit stats.
+      #
+      # @example
+      #   require "google/cloud/spanner"
+      #
+      #   spanner = Google::Cloud::Spanner.new
+      #
+      #   db = spanner.client "my-instance", "my-database"
+      #
+      #   timestamp = db.commit do |c|
+      #     c.update "users", [{ id: 1, name: "Charlie", active: false }]
+      #     c.insert "users", [{ id: 2, name: "Harvey",  active: true }]
+      #   end
+      #
+      #   puts timestamp
+      #
+      # @example With commit stats.
+      #   require "google/cloud/spanner"
+      #
+      #   spanner = Google::Cloud::Spanner.new
+      #
+      #   db = spanner.client "my-instance", "my-database"
+      #
+      #   commit_options = { return_commit_stats: true }
+      #   commit_resp = db.commit commit_options: commit_options do |c|
+      #     c.update "users", [{ id: 1, name: "Charlie", active: false }]
+      #     c.insert "users", [{ id: 2, name: "Harvey",  active: true }]
+      #   end
+      #
+      #   puts commit_resp.timestamp
+      #   puts commit_resp.stats.mutation_count
+      #
+      class CommitResponse
+        ##
+        # @private Creates a new CommitResponse instance.
+        def initialize grpc
+          @grpc = grpc
+        end
+
+        ##
+        # The timestamp at which the transaction committed.
+        # @return [Time]
+        def timestamp
+          Convert.timestamp_to_time @grpc.commit_timestamp
+        end
+
+        ##
+        # Additional statistics about a commit.
+        # @return [CommitStats, nil] Commit stats or nil if not stats not
+        #   present.
+        def stats
+          CommitStats.from_grpc @grpc.commit_stats if @grpc.commit_stats
+        end
+
+        ##
+        # @private
+        # Creates a new Commit responsee instance from a
+        # `Google::Cloud::Spanner::V1::CommitResponse`.
+        def self.from_grpc grpc
+          new grpc
+        end
+      end
+    end
+  end
+end

data/lib/google/cloud/spanner/commit_response/commit_stats.rb

@@ -0,0 +1,51 @@
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+require "google/cloud/spanner/convert"
+
+module Google
+  module Cloud
+    module Spanner
+      class CommitResponse
+        ##
+        # # CommitStats
+        #
+        # Statistical information of a transaction commit.
+        #
+        class CommitStats
+          ##
+          # @private Creates a new CommitStats instance.
+          def initialize grpc
+            @grpc = grpc
+          end
+
+          # The total number of the mutations for the transaction.
+          # @return [Integer]
+          def mutation_count
+            @grpc.mutation_count
+          end
+
+          ##
+          # @private
+          # Creates a new CommitStats instance from a
+          # `Google::Cloud::Spanner::V1::CommitResponse::CommitStats`.
+          def self.from_grpc grpc
+            new grpc
+          end
+        end
+      end
+    end
+  end
+end

data/lib/google/cloud/spanner/convert.rb

@@ -17,6 +17,7 @@ require "time"
 require "date"
 require "stringio"
 require "base64"
+require "bigdecimal"
 require "google/cloud/spanner/data"
 
 module Google
@@ -67,6 +68,9 @@ module Google
               Google::Protobuf::Value.new bool_value: false
             when Integer
               Google::Protobuf::Value.new string_value: obj.to_s
+            # BigDecimal must be put before Numeric.
+            when BigDecimal
+              Google::Protobuf::Value.new string_value: obj.to_s("F")
             when Numeric
               if obj == Float::INFINITY
                 Google::Protobuf::Value.new string_value: "Infinity"
@@ -125,6 +129,9 @@ module Google
               :BOOL
             when Integer
               :INT64
+            # BigDecimal must be put before Numeric.
+            when BigDecimal
+              :NUMERIC
             when Numeric
               :FLOAT64
             when Time
@@ -169,12 +176,12 @@ module Google
             return field.to_grpc_type if field.respond_to? :to_grpc_type
 
             if Array === field
-              Google::Spanner::V1::Type.new(
+              V1::Type.new(
                 code: :ARRAY,
                 array_element_type: grpc_type_for_field(field.first)
               )
             else
-              Google::Spanner::V1::Type.new(code: field)
+              V1::Type.new(code: field)
             end
           end
 
@@ -214,6 +221,8 @@ module Google
               end
             when :STRUCT
               Data.from_grpc value.list_value.values, type.struct_type.fields
+            when :NUMERIC
+              BigDecimal value.string_value
             end
           end
 
@@ -274,24 +283,24 @@ module Google
               range_opts.delete :end_closed
             end
 
-            Google::Spanner::V1::KeyRange.new range_opts
+            V1::KeyRange.new range_opts
           end
 
           def to_key_set keys
-            return Google::Spanner::V1::KeySet.new(all: true) if keys.nil?
+            return V1::KeySet.new(all: true) if keys.nil?
             keys = [keys] unless keys.is_a? Array
-            return Google::Spanner::V1::KeySet.new(all: true) if keys.empty?
+            return V1::KeySet.new(all: true) if keys.empty?
 
             if keys_are_ranges? keys
               key_ranges = keys.map { |r| to_key_range(r) }
-              return Google::Spanner::V1::KeySet.new(ranges: key_ranges)
+              return V1::KeySet.new(ranges: key_ranges)
             end
 
             key_list = keys.map do |key|
               key = [key] unless key.is_a? Array
               object_to_grpc_value(key).list_value
             end
-            Google::Spanner::V1::KeySet.new keys: key_list
+            V1::KeySet.new keys: key_list
           end
 
           def keys_are_ranges? keys

data/lib/google/cloud/spanner/credentials.rb

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 
-require "google/cloud/spanner/v1/credentials"
+require "google/cloud/spanner/v1/spanner/credentials"
 
 module Google
   module Cloud
@@ -37,7 +37,7 @@ module Google
       #
       #   spanner.project_id #=> "my-project"
       #
-      class Credentials < Google::Cloud::Spanner::V1::Credentials
+      class Credentials < Google::Cloud::Spanner::V1::Spanner::Credentials
       end
     end
   end

data/lib/google/cloud/spanner/data.rb

@@ -216,9 +216,9 @@ module Google
         ##
         # @private
         def to_grpc_type
-          Google::Spanner::V1::Type.new(
+          V1::Type.new(
             code: :STRUCT,
-            struct_type: Google::Spanner::V1::StructType.new(
+            struct_type: V1::StructType.new(
               fields: @grpc_fields
             )
           )

data/lib/google/cloud/spanner/database.rb

@@ -229,7 +229,6 @@ module Google
           true
         end
 
-
         # @private
         DATBASE_OPERATION_METADAT_FILTER_TEMPLATE = [
           "(metadata.@type:CreateDatabaseMetadata AND " \
@@ -248,10 +247,10 @@ module Google
         #   response.
         #
         #   The response returns a list of
-        #   {Google::Longrunning::Operation long-running operations} whose names
+        #   `Google::Longrunning::Operation` long-running operations whose names
         #   are prefixed by a database name within the specified instance.
         #   The long-running operation
-        #   {Google::Longrunning::Operation#metadata metadata} field type
+        #   `Google::Longrunning::Operation#metadata` metadata field type
         #   `metadata.type_url` describes the type of the metadata.
         #
         #   The filter expression must specify the field name,
@@ -384,10 +383,9 @@ module Google
             )
           end
 
-          grpc = service.list_database_operations \
-            instance_id,
-            filter: database_filter,
-            page_size: page_size
+          grpc = service.list_database_operations instance_id,
+                                                  filter: database_filter,
+                                                  page_size: page_size
           Database::Job::List.from_grpc grpc, service
         end
 
@@ -573,20 +571,22 @@ module Google
         #
         #   The permissions that can be checked on a database are:
         #
-        #   * pubsub.databases.create
-        #   * pubsub.databases.list
-        #   * pubsub.databases.update
-        #   * pubsub.databases.updateDdl
-        #   * pubsub.databases.get
-        #   * pubsub.databases.getDdl
-        #   * pubsub.databases.getIamPolicy
-        #   * pubsub.databases.setIamPolicy
-        #   * pubsub.databases.beginReadOnlyTransaction
-        #   * pubsub.databases.beginOrRollbackReadWriteTransaction
-        #   * pubsub.databases.read
-        #   * pubsub.databases.select
-        #   * pubsub.databases.write
-        #   * pubsub.databases.drop
+        #  * spanner.databases.beginPartitionedDmlTransaction
+        #  * spanner.databases.create
+        #  * spanner.databases.createBackup
+        #  * spanner.databases.list
+        #  * spanner.databases.update
+        #  * spanner.databases.updateDdl
+        #  * spanner.databases.get
+        #  * spanner.databases.getDdl
+        #  * spanner.databases.getIamPolicy
+        #  * spanner.databases.setIamPolicy
+        #  * spanner.databases.beginReadOnlyTransaction
+        #  * spanner.databases.beginOrRollbackReadWriteTransaction
+        #  * spanner.databases.read
+        #  * spanner.databases.select
+        #  * spanner.databases.write
+        #  * spanner.databases.drop
         #
         # @return [Array<Strings>] The permissions that have access.
         #
@@ -611,7 +611,7 @@ module Google
 
         ##
         # @private Creates a new Database instance from a
-        # Google::Spanner::Admin::Database::V1::Database.
+        # `Google::Cloud::Spanner::Admin::Database::V1::Database`.
         def self.from_grpc grpc, service
           new grpc, service
         end

data/lib/google/cloud/spanner/database/backup_info.rb

@@ -94,7 +94,7 @@ module Google
 
           ##
           # @private Creates a new Database::BackupInfo instance from a
-          # Google::Spanner::Admin::Database::V1::BackupInfo.
+          # `Google::Cloud::Spanner::Admin::Database::V1::BackupInfo`.
           def self.from_grpc grpc
             new grpc
           end

data/lib/google/cloud/spanner/database/job.rb

@@ -52,7 +52,7 @@ module Google
         #
         class Job
           ##
-          # @private The Google::Gax::Operation gRPC object.
+          # @private The `Gapic::Operation` gRPC object.
           attr_accessor :grpc
 
           ##
@@ -89,7 +89,7 @@ module Google
             return nil unless done?
             return nil unless @grpc.grpc_op.result == :response
             return nil unless @grpc.results.instance_of? \
-              Google::Spanner::Admin::Database::V1::Database
+              Admin::Database::V1::Database
             Database.from_grpc @grpc.results, service
           end
 
@@ -201,7 +201,7 @@ module Google
           end
 
           ##
-          # @private New Database::Job from a Google::Gax::Operation object.
+          # @private New Database::Job from a `Gapic::Operation` object.
           def self.from_grpc grpc, service
             new.tap do |job|
               job.instance_variable_set :@grpc, grpc

data/lib/google/cloud/spanner/database/job/list.rb

@@ -146,7 +146,7 @@ module Google
             # @private
             #
             # New Database::Job::List from a
-            # Google::Gax::PagedEnumerable<Google::Longrunning::Operation>
+            # `Gapic::PagedEnumerable<Google::Longrunning::Operation>`
             # object. Operation object is a database operation.
             #
             def self.from_grpc grpc, service
@@ -154,7 +154,7 @@ module Google
                 service.databases.instance_variable_get "@operations_client"
               jobs = new(Array(grpc.response.operations).map do |job_grpc|
                 Job.from_grpc \
-                  Google::Gax::Operation.new(job_grpc, operations_client),
+                  Gapic::Operation.new(job_grpc, operations_client),
                   service
               end)
               jobs.grpc = grpc