wip

Author: shubhangi-google

google-cloud-storage-control-v2/lib/google/cloud/storage/control/v2/storage_control/operations.rb

@@ -283,6 +283,7 @@ def get_operation request, options = nil
 
                 # Customize the options with defaults
                 metadata = @config.rpcs.get_operation.metadata.to_h
+                binding.pry
 
                 # Set x-goog-api-client, x-goog-user-project and x-goog-api-version headers
                 metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
@@ -304,7 +305,6 @@ def get_operation request, options = nil
                                        retry_policy: @config.rpcs.get_operation.retry_policy
 
                 options.apply_defaults timeout:      @config.timeout,
-                                       metadata:     @config.metadata,
                                        retry_policy: @config.retry_policy
 
                 @operations_stub.call_rpc :get_operation, request, options: options do |response, operation|

google-cloud-storage-control/samples/acceptance/helper.rb

@@ -26,9 +26,11 @@ def random_folder_name prefix: "ruby-storage-control-folder-samples-test-"
   t = Time.now.utc.iso8601.gsub ":", "-"
   "#{prefix}-#{t}-#{SecureRandom.hex 4}".downcase
 end
+def storage_client
+  @storage_client ||= Google::Cloud::Storage.new
+end
 
 def create_bucket_helper bucket_name, uniform_bucket_level_access: nil, hierarchical_namespace: nil
-  storage_client = Google::Cloud::Storage.new
   retry_resource_exhaustion do
     storage_client.create_bucket bucket_name do |b|
       b.uniform_bucket_level_access = uniform_bucket_level_access
@@ -38,7 +40,6 @@ def create_bucket_helper bucket_name, uniform_bucket_level_access: nil, hierarch
 end
 
 def delete_bucket_helper bucket_name
-  storage_client = Google::Cloud::Storage.new
   retry_resource_exhaustion do
     bucket = storage_client.bucket bucket_name
     return unless bucket

google-cloud-storage-control/samples/acceptance/storage_control_anywhere_cache_test.rb

@@ -21,6 +21,7 @@
 require_relative "../storage_control_pause_anywhere_cache"
 require_relative "../storage_control_resume_anywhere_cache"
 require_relative "../storage_control_disable_anywhere_cache"
+require 'pry'
 
 describe "Storage Control Anywhere Cache" do
   let(:bucket_name) { random_bucket_name }
@@ -29,11 +30,11 @@
   let(:anywhere_cache_name) { "projects/_/buckets/#{bucket_name}/anywhereCaches/#{zone}" }
 
   before :all do
-    @bucket = create_bucket_helper bucket_name
+    bucket = create_bucket_helper bucket_name
   end
 
-  after do
-    delete_bucket_helper bucket_name until count_anywhere_caches(bucket_name).zero?
+  after :all do
+    delete_bucket_helper bucket_name #{}until count_anywhere_caches(bucket_name).zero?
   end
 
   it "handles Anywhere cache lifecycle in sequence" do
@@ -74,14 +75,14 @@
   end
 end
 
-def count_anywhere_caches bucket_name
-  sleep 900
-  storage_control_client = Google::Cloud::Storage::Control.storage_control
-  # Set project to "_" to signify global bucket
-  parent = "projects/_/buckets/#{bucket_name}"
-  request = Google::Cloud::Storage::Control::V2::ListAnywhereCachesRequest.new(
-    parent: parent
-  )
-  result = storage_control_client.list_anywhere_caches request
-  result.response.anywhere_caches.count
-end
+# def count_anywhere_caches bucket_name
+#   sleep 900
+#   storage_control_client = Google::Cloud::Storage::Control.storage_control
+#   # Set project to "_" to signify global bucket
+#   parent = "projects/_/buckets/#{bucket_name}"
+#   request = Google::Cloud::Storage::Control::V2::ListAnywhereCachesRequest.new(
+#     parent: parent
+#   )
+#   result = storage_control_client.list_anywhere_caches request
+#   result.response.anywhere_caches.count
+# end

google-cloud-storage-control/samples/storage_control_create_anywhere_cache.rb

@@ -47,16 +47,19 @@ def create_anywhere_cache bucket_name:, zone:
         name: name
       )
       result = storage_control_client.get_anywhere_cache get_request
-      until result.state == "running"
-        sleep 1800 # Wait for 1/2 hour before checking again
+      min_delay = 900 # 15 minutes
+      max_delay = 1800 # 30 minutes
+      while result.state != "running"
+        puts "Cache not running yet, current state is #{result.state}. Retrying in #{min_delay} seconds."
+        sleep min_delay
+        min_delay = [min_delay * 2, max_delay].min # Exponential backoff with a max delay
         result = storage_control_client.get_anywhere_cache get_request
-        puts "AnywhereCache status check retried"
       end
       puts "AnywhereCache created - #{result.name}"
     else
       puts "AnywhereCache create operation failed"
     end
-  rescue StandardError => e
+  rescue Google::Cloud::Error=> e
     puts "Error creating AnywhereCache: #{e.message}"
   end
 end