feat: Add updateSchedule method to ScheduleService

PiperOrigin-RevId: 532625344

google/cloud/aiplatform/v1beta1/schedule_service.proto

@@ -24,6 +24,7 @@ import "google/cloud/aiplatform/v1beta1/operation.proto";
 import "google/cloud/aiplatform/v1beta1/schedule.proto";
 import "google/longrunning/operations.proto";
 import "google/protobuf/empty.proto";
+import "google/protobuf/field_mask.proto";
 
 option csharp_namespace = "Google.Cloud.AIPlatform.V1Beta1";
 option go_package = "cloud.google.com/go/aiplatform/apiv1beta1/aiplatformpb;aiplatformpb";
@@ -106,6 +107,21 @@ service ScheduleService {
     option (google.api.method_signature) = "name";
     option (google.api.method_signature) = "name,catch_up";
   }
+
+  // Updates an active or paused Schedule.
+  //
+  // When the Schedule is updated, new runs will be scheduled starting from the
+  // updated next execution time after the update time based on the
+  // time_specification in the updated Schedule. All unstarted runs before the
+  // update time will be skipped while already created runs will NOT be paused
+  // or canceled.
+  rpc UpdateSchedule(UpdateScheduleRequest) returns (Schedule) {
+    option (google.api.http) = {
+      patch: "/v1beta1/{schedule.name=projects/*/locations/*/schedules/*}"
+      body: "schedule"
+    };
+    option (google.api.method_signature) = "schedule,update_mask";
+  }
 }
 
 // Request message for
@@ -274,3 +290,18 @@ message ResumeScheduleRequest {
   // field. Default to false.
   bool catch_up = 2 [(google.api.field_behavior) = OPTIONAL];
 }
+
+// Request message for
+// [ScheduleService.UpdateSchedule][google.cloud.aiplatform.v1beta1.ScheduleService.UpdateSchedule].
+message UpdateScheduleRequest {
+  // Required. The Schedule which replaces the resource on the server.
+  // The following restrictions will be applied:
+  //   * The scheduled request type cannot be changed.
+  //   * The output_only fields will be ignored if specified.
+  Schedule schedule = 1 [(google.api.field_behavior) = REQUIRED];
+
+  // Required. The update mask applies to the resource. See
+  // [google.protobuf.FieldMask][google.protobuf.FieldMask].
+  google.protobuf.FieldMask update_mask = 2
+      [(google.api.field_behavior) = REQUIRED];
+}