feat: Add support for snapshot clone BigQuery

Author: hemanshv

apis/Google.Cloud.BigQuery.V2/Google.Cloud.BigQuery.V2.IntegrationTests/SnapshotCloneTest.cs

@@ -0,0 +1,81 @@
+// Copyright 2023 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.
+
+using Google.Apis.Bigquery.v2.Data;
+using Xunit;
+
+namespace Google.Cloud.BigQuery.V2.IntegrationTests;
+
+[Collection(nameof(BigQueryFixture))]
+public class SnapshotCloneTest
+{
+    private readonly BigQueryFixture _fixture;
+    public SnapshotCloneTest(BigQueryFixture fixture)
+    {
+        _fixture = fixture;
+    }
+
+    private TableReference GetTableReference(string datasetId, string tableId) =>
+        new TableReference { ProjectId = _fixture.ProjectId, DatasetId = datasetId, TableId = tableId };
+
+    [Fact]
+    public void CreateTableSnapshot()
+    {
+        var client = BigQueryClient.Create(_fixture.ProjectId);
+        var datasetId = _fixture.DatasetId;
+        var tableId = _fixture.PeopleTableId;
+
+        var sourceTableReference = GetTableReference(datasetId, tableId);
+        var destinationTableReference = GetTableReference(datasetId, tableId + "SnapshotTable");
+        var jobOptions = new CreateCopyJobOptions
+        {
+            OperationType = CopyOperationType.Snapshot
+        };
+        var job = client.CreateCopyJob(sourceTableReference, destinationTableReference, jobOptions).PollUntilCompleted();
+        Assert.Null(job.Status.ErrorResult);
+
+        var destTable = client.GetTable(_fixture.DatasetId, destinationTableReference.TableId);
+        Assert.NotNull(destTable.Resource.SnapshotDefinition);
+
+        var baseTableReference = destTable.Resource.SnapshotDefinition.BaseTableReference;
+        Assert.Equal(sourceTableReference.TableId, baseTableReference.TableId);
+        Assert.Equal(sourceTableReference.ProjectId, baseTableReference.ProjectId);
+        Assert.Equal(sourceTableReference.DatasetId, baseTableReference.DatasetId);
+    }
+
+    [Fact]
+    public void CreateTableClone()
+    {
+        var client = BigQueryClient.Create(_fixture.ProjectId);
+        var datasetId = _fixture.DatasetId;
+        var tableId = _fixture.PeopleTableId;
+
+        var sourceTableReference = GetTableReference(datasetId, tableId);
+        var destinationTableReference = GetTableReference(datasetId, tableId + "CloneTable");
+        var jobOptions = new CreateCopyJobOptions
+        {
+            OperationType = CopyOperationType.Clone
+        };
+        var job = client.CreateCopyJob(sourceTableReference, destinationTableReference, jobOptions).PollUntilCompleted();
+        Assert.Null(job.Status.ErrorResult);
+
+        var destTable = client.GetTable(_fixture.DatasetId, destinationTableReference.TableId);
+        Assert.NotNull(destTable.Resource.CloneDefinition);
+
+        var baseTableReference = destTable.Resource.CloneDefinition.BaseTableReference;
+        Assert.Equal(sourceTableReference.TableId, baseTableReference.TableId);
+        Assert.Equal(sourceTableReference.ProjectId, baseTableReference.ProjectId);
+        Assert.Equal(sourceTableReference.DatasetId, baseTableReference.DatasetId);
+    }
+}

apis/Google.Cloud.BigQuery.V2/Google.Cloud.BigQuery.V2.Tests/CreateCopyJobOptionsTest.cs

@@ -1,4 +1,4 @@
-// Copyright 2017 Google Inc. All Rights Reserved.
+// Copyright 2017 Google Inc. All Rights Reserved.
 // 
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -27,11 +27,13 @@ public void ModifyRequest()
                 CreateDisposition = CreateDisposition.CreateIfNeeded,
                 WriteDisposition = WriteDisposition.WriteIfEmpty,
                 DestinationEncryptionConfiguration = new EncryptionConfiguration { KmsKeyName = "projects/1/locations/us/keyRings/1/cryptoKeys/1" },
+                OperationType = CopyOperationType.Clone
             };
             JobConfigurationTableCopy request = new JobConfigurationTableCopy();
             options.ModifyRequest(request);
             Assert.Equal("CREATE_IF_NEEDED", request.CreateDisposition);
             Assert.Equal("WRITE_EMPTY", request.WriteDisposition);
+            Assert.Equal("CLONE", request.OperationType);
             Assert.Equal("projects/1/locations/us/keyRings/1/cryptoKeys/1", request.DestinationEncryptionConfiguration.KmsKeyName);
         }        
     }

apis/Google.Cloud.BigQuery.V2/Google.Cloud.BigQuery.V2/CopyOperationType.cs

@@ -0,0 +1,53 @@
+// Copyright 2023 Google Inc. All Rights Reserved.
+// 
+// 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.
+
+namespace Google.Cloud.BigQuery.V2
+{
+    /// <summary>
+    /// Indicates different operation types supported in table copy job.
+    /// </summary>
+    public enum CopyOperationType
+    {
+        /// <summary>
+        /// Operation not specified.
+        /// </summary>
+        [ApiValue("OPERATION_TYPE_UNSPECIFIED")]
+        Unspecified,
+
+        /// <summary>
+        /// The source and destination table have the same table type.
+        /// </summary>
+        [ApiValue("COPY")]
+        Copy,
+
+        /// <summary>
+        /// The source table type is TABLE and the destination table type is SNAPSHOT.
+        /// </summary>
+        [ApiValue("SNAPSHOT")]
+        Snapshot,
+
+        /// <summary>
+        /// The source table type is SNAPSHOT and the destination table type is TABLE.
+        /// </summary>
+        [ApiValue("RESTORE")]
+        Restore,
+
+        /// <summary>
+        /// The source and destination table have the same table type, 
+        /// but we charge only for the delta bytes.
+        /// </summary>
+        [ApiValue("CLONE")]
+        Clone,
+    }
+}

apis/Google.Cloud.BigQuery.V2/Google.Cloud.BigQuery.V2/CreateCopyJobOptions.cs

@@ -1,4 +1,4 @@
-// Copyright 2017 Google Inc. All Rights Reserved.
+// Copyright 2017 Google Inc. All Rights Reserved.
 // 
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -38,6 +38,11 @@ public sealed class CreateCopyJobOptions : JobCreationOptions
         /// </summary>
         public EncryptionConfiguration DestinationEncryptionConfiguration { get; set; }
 
+        /// <summary>
+        /// Specifies the operation type for the job, if any.
+        /// </summary>
+        public CopyOperationType? OperationType { get; set; }
+
         internal void ModifyRequest(JobConfigurationTableCopy copy)
         {
             if (CreateDisposition != null)
@@ -52,6 +57,10 @@ internal void ModifyRequest(JobConfigurationTableCopy copy)
             {
                 copy.DestinationEncryptionConfiguration = DestinationEncryptionConfiguration;
             }
+            if (OperationType != null)
+            {
+                copy.OperationType = EnumMap.ToApiValue(OperationType.Value);
+            }
         }
     }
 }