feat: Support Idempotency in MutateRow API

Author: shantstepanian

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/models/RowMutation.java

@@ -17,6 +17,7 @@
 
 import com.google.api.core.BetaApi;
 import com.google.api.core.InternalApi;
+import com.google.bigtable.v2.Idempotency;
 import com.google.bigtable.v2.MutateRowRequest;
 import com.google.bigtable.v2.MutateRowsRequest;
 import com.google.bigtable.v2.MutateRowsRequest.Entry;
@@ -38,6 +39,7 @@ public final class RowMutation implements MutationApi<RowMutation>, Serializable
   private final TargetId targetId;
   private final ByteString key;
   private final Mutation mutation;
+  private ByteString idempotency_token;
 
   private RowMutation(TargetId targetId, ByteString key, Mutation mutation) {
     Preconditions.checkNotNull(targetId, "target id can't be null.");
@@ -255,6 +257,11 @@ public RowMutation mergeToCell(
     return this;
   }
 
+  public RowMutation setIdempotency(@Nonnull ByteString token) {
+    this.idempotency_token = token;
+    return this;
+  }
+
   @InternalApi
   public MutateRowRequest toProto(RequestContext requestContext) {
     MutateRowRequest.Builder builder = MutateRowRequest.newBuilder();
@@ -266,6 +273,12 @@ public MutateRowRequest toProto(RequestContext requestContext) {
       builder.setTableName(resourceName);
     }
 
+    if (idempotency_token != null) {
+      Idempotency.Builder idempotencyBuilder = Idempotency.newBuilder();
+      idempotencyBuilder.setToken(idempotency_token);
+      builder.setIdempotency(idempotencyBuilder);
+    }
+
     return builder
         .setAppProfileId(requestContext.getAppProfileId())
         .setRowKey(key)

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/it/MutateRowIT.java

@@ -152,4 +152,44 @@ public void testOnAuthorizedView() throws Exception {
         .getTableAdminClient()
         .deleteAuthorizedView(testEnvRule.env().getTableId(), testAuthorizedView.getId());
   }
+
+  @Test
+  public void testIdempotentAggregates() throws Exception {
+    String rowKey = UUID.randomUUID().toString();
+    String familyId = testEnvRule.env().getFamilyId();
+
+    // First mutation w/ idempotency token.
+    testEnvRule
+        .env()
+        .getDataClient()
+        .mutateRowAsync(
+            RowMutation.create(testEnvRule.env().getTableId(), rowKey)
+            .addToCell(familyId, "cq", 0, 10)
+            .setIdempotency(ByteString.copyFromUtf8("abcdefgh"))
+                )
+        .get(1, TimeUnit.MINUTES);
+
+    // Now replay the mutation.
+    testEnvRule
+        .env()
+        .getDataClient()
+        .mutateRowAsync(
+            RowMutation.create(testEnvRule.env().getTableId(), rowKey)
+            .addToCell(familyId, "cq", 0, 10)
+            .setIdempotency(ByteString.copyFromUtf8("abcdefgh"))
+                )
+        .get(1, TimeUnit.MINUTES);
+
+    Row row =
+        testEnvRule
+            .env()
+            .getDataClient()
+            .readRowsCallable()
+            .first()
+            .call(Query.create(testEnvRule.env().getTableId()).rowKey(rowKey));
+
+    assertThat(row.getCells()).hasSize(1);
+    // Ensure that the increment is only applied once.
+    assertThat(row.getCells().get(0).getValue()).isEqualTo(10);Add commentMore actions
+  }
 }