samples(pubsub): pubsub_commit_avro_schema sample

cy-yun · cy-yun · commit 7ad684745561 · 2025-08-05T14:14:35.000-07:00
diff --git a/google-cloud-pubsub/samples/acceptance/schemas_test.rb b/google-cloud-pubsub/samples/acceptance/schemas_test.rb
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 require_relative "helper"
+require_relative "../pubsub_commit_avro_schema"
 require_relative "../pubsub_commit_proto_schema"
 require_relative "../pubsub_create_avro_schema"
 require_relative "../pubsub_create_topic_with_schema"
@@ -183,6 +184,32 @@
         end
       end
     end
+
+    it "supports pubsub_commit_avro_schema & pubsub_commit_list_schema_revisions" do
+      schema = Google::Cloud::PubSub::V1::Schema.new name: schema_id, 
+                                                     type: :AVRO,
+                                                     definition: avsc_definition
+      @schema = schemas.create_schema parent: pubsub.project_path,
+                                      schema: schema,
+                                      schema_id: schema_id
+
+      rev_id = @schema.revision_id
+
+      schema1 = nil
+      # pubsub_commit_avro_schema
+      out, _err = capture_io do
+        schema1 = commit_avro_schema schema_id: schema_id, avsc_file: avsc_file
+      end
+      refute_equal out, "Schema commited with revision #{rev_id}."
+      assert_includes out, "Schema commited with revision"
+
+      # pubsub_list_schema_revisions
+      out, _err = capture_io do
+        list_schema_revisions schema_id: schema_id
+      end
+
+      assert_includes out, schema1.revision_id
+    end
   end
 
   describe "PROTOCOL_BUFFER" do
diff --git a/google-cloud-pubsub/samples/pubsub_commit_avro_schema.rb b/google-cloud-pubsub/samples/pubsub_commit_avro_schema.rb
@@ -0,0 +1,38 @@
+# Copyright 2025 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
+#
+#     http://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/pubsub"
+
+def commit_avro_schema schema_id:, avsc_file:
+  # [START pubsub_commit_avro_schema]
+  # schema_id = "your-schema-id"
+  # avsc_file = "path/to/a/avsc_file.avsc"
+
+  pubsub = Google::Cloud::Pubsub.new
+
+  schemas = pubsub.schemas
+
+  schema = schemas.get_schema name: pubsub.schema_path(schema_id)
+
+  definition = File.read avsc_file
+
+  schema.definition = definition
+
+  result = schemas.commit_schema name: schema.name,
+                                 schema: schema
+
+  puts "Schema commited with revision #{result.revision_id}."
+  result
+  # [END pubsub_commit_avro_schema]
+end
