feat: adding utility to transform protobuf into model object

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

@@ -114,6 +114,20 @@ public MutateRowsRequest toProto(RequestContext requestContext) {
         .build();
   }
 
+  /**
+   * Wraps the protobuf {@link MutateRowsRequest}.
+   *
+   * <p>WARNING: Please note that the project id & instance id in the table name will be overwritten
+   * by the configuration in the BigtableDataClient.
+   */
+  public static BulkMutation fromProtobuf(@Nonnull MutateRowsRequest request) {
+    BulkMutation bulkMutation =
+        BulkMutation.create(NameUtil.extractTableIdFromTableName(request.getTableName()));
+    bulkMutation.builder = request.toBuilder();
+
+    return bulkMutation;
+  }
+
   /** Creates a copy of {@link BulkMutation}. */
   @Override
   public BulkMutation clone() {

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

@@ -136,4 +136,19 @@ public CheckAndMutateRowRequest toProto(RequestContext requestContext) {
         .setAppProfileId(requestContext.getAppProfileId())
         .build();
   }
+
+  /**
+   * Wraps the protobuf {@link CheckAndMutateRowRequest}.
+   *
+   * <p>WARNING: Please note that the project id & instance id in the table name will be overwritten
+   * by the configuration in the BigtableDataClient.
+   */
+  public static ConditionalRowMutation fromProtobuf(@Nonnull CheckAndMutateRowRequest request) {
+    String tableId = NameUtil.extractTableIdFromTableName(request.getTableName());
+    ConditionalRowMutation rowMutation =
+        ConditionalRowMutation.create(tableId, request.getRowKey());
+    rowMutation.builder = request.toBuilder();
+
+    return rowMutation;
+  }
 }

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

@@ -137,4 +137,19 @@ public ReadModifyWriteRowRequest toProto(RequestContext requestContext) {
         .setAppProfileId(requestContext.getAppProfileId())
         .build();
   }
+
+  /**
+   * Wraps the protobuf {@link ReadModifyWriteRowRequest}.
+   *
+   * <p>WARNING: Please note that the project id & instance id in the table name will be overwritten
+   * by the configuration in the BigtableDataClient.
+   */
+  public static ReadModifyWriteRow fromProtobuf(ReadModifyWriteRowRequest request) {
+    String tableId = NameUtil.extractTableIdFromTableName(request.getTableName());
+
+    ReadModifyWriteRow row = ReadModifyWriteRow.create(tableId, request.getRowKey());
+    row.builder = request.toBuilder();
+
+    return row;
+  }
 }

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

@@ -214,4 +214,18 @@ public MutateRowsRequest toBulkProto(RequestContext requestContext) {
             Entry.newBuilder().setRowKey(key).addAllMutations(mutation.getMutations()).build())
         .build();
   }
+
+  /**
+   * Wraps the protobuf {@link MutateRowRequest}.
+   *
+   * <p>WARNING: Please note that the project id & instance id in the table name will be overwritten
+   * by the configuration in the BigtableDataClient.
+   */
+  public static RowMutation fromProtobuf(@Nonnull MutateRowRequest request) {
+    String tableId = NameUtil.extractTableIdFromTableName(request.getTableName());
+
+    // TODO: Should we give an option for Mutation.fromProto(List<com.google.bigtable.v2.Mutation>)?
+    return RowMutation.create(
+        tableId, request.getRowKey(), Mutation.fromProtoUnsafe(request.getMutationsList()));
+  }
 }

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

@@ -16,6 +16,7 @@
 package com.google.cloud.bigtable.data.v2.models;
 
 import com.google.api.core.InternalApi;
+import com.google.bigtable.v2.MutateRowRequest;
 import com.google.bigtable.v2.MutateRowsRequest;
 import com.google.common.base.Preconditions;
 import com.google.protobuf.ByteString;
@@ -177,4 +178,11 @@ public MutateRowsRequest.Entry toProto() {
         .addAllMutations(mutation.getMutations())
         .build();
   }
+
+  /** Wraps the protobuf {@link MutateRowRequest}. */
+  public static RowMutationEntry fromProtobuf(MutateRowsRequest.Entry request) {
+    // TODO: Should we have an option for Mutation.fromProto(List<com.google.bigtable.v2.Mutation>)?
+    return new RowMutationEntry(
+        request.getRowKey(), Mutation.fromProtoUnsafe(request.getMutationsList()));
+  }
 }

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/models/BulkMutationTest.java

@@ -147,4 +147,18 @@ public void addRowMutationEntry() {
     bulkMutation.add(entry);
     assertThat(bulkMutation.toProto(REQUEST_CONTEXT).getEntriesList()).contains(entry.toProto());
   }
+
+  @Test
+  public void fromProtobufTest() {
+    BulkMutation expected =
+        BulkMutation.create(TABLE_ID)
+            .add(
+                "key",
+                Mutation.create().setCell("fake-family", "fake-qualifier", 10_000L, "fake-value"));
+
+    MutateRowsRequest protoRequest = expected.toProto(REQUEST_CONTEXT);
+    BulkMutation actualBulkMutation = BulkMutation.fromProtobuf(protoRequest);
+
+    assertThat(actualBulkMutation.toProto(REQUEST_CONTEXT)).isEqualTo(protoRequest);
+  }
 }

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/models/ConditionalRowMutationTest.java

@@ -162,4 +162,18 @@ public void serializationTest() throws IOException, ClassNotFoundException {
     ConditionalRowMutation actual = (ConditionalRowMutation) ois.readObject();
     assertThat(actual.toProto(REQUEST_CONTEXT)).isEqualTo(expected.toProto(REQUEST_CONTEXT));
   }
+
+  @Test
+  public void fromProtobufTest() {
+    ConditionalRowMutation mutation =
+        ConditionalRowMutation.create(TABLE_ID, TEST_KEY)
+            .condition(Filters.FILTERS.key().regex("test"))
+            .then(Mutation.create().setCell("family1", "qualifier1", 10_000L, "value"))
+            .otherwise(Mutation.create().deleteFamily("family"));
+
+    CheckAndMutateRowRequest protoRequest = mutation.toProto(REQUEST_CONTEXT);
+    ConditionalRowMutation actualRequest = ConditionalRowMutation.fromProtobuf(protoRequest);
+
+    assertThat(actualRequest.toProto(REQUEST_CONTEXT)).isEqualTo(protoRequest);
+  }
 }

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/models/ReadModifyWriteRowTest.java

@@ -113,4 +113,17 @@ public void serializationTest() throws IOException, ClassNotFoundException {
     ReadModifyWriteRow actual = (ReadModifyWriteRow) ois.readObject();
     assertThat(actual.toProto(REQUEST_CONTEXT)).isEqualTo(expected.toProto(REQUEST_CONTEXT));
   }
+
+  @Test
+  public void fromProtobufTest() {
+    ReadModifyWriteRow expected =
+        ReadModifyWriteRow.create(TABLE_ID, "row-key")
+            .increment("fake-family", ByteString.copyFromUtf8("fake-qualifier"), 1)
+            .append("fake-family", "fake-qualifier", "fake-value");
+
+    ReadModifyWriteRowRequest protoRequest = expected.toProto(REQUEST_CONTEXT);
+    ReadModifyWriteRow actualRequest = ReadModifyWriteRow.fromProtobuf(protoRequest);
+
+    assertThat(actualRequest.toProto(REQUEST_CONTEXT)).isEqualTo(protoRequest);
+  }
 }

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/models/RowMutationEntryTest.java

@@ -165,4 +165,17 @@ public void testWithLongValue() {
                 .setValue(ByteString.copyFrom(Longs.toByteArray(100_000L)))
                 .build());
   }
+
+  @Test
+  public void fromProtobufTest() {
+    RowMutationEntry expected =
+        RowMutationEntry.create("row-key")
+            .setCell("fake-family", "fake-qualifier", 10_000L, "fake-value")
+            .deleteFamily("fake-family");
+
+    MutateRowsRequest.Entry protoRequest = expected.toProto();
+    RowMutationEntry actualRequest = RowMutationEntry.fromProtobuf(protoRequest);
+
+    assertThat(actualRequest.toProto()).isEqualTo(protoRequest);
+  }
 }

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/models/RowMutationTest.java

@@ -139,4 +139,17 @@ public void testWithLongValue() {
                 .setValue(ByteString.copyFrom(Longs.toByteArray(100_000L)))
                 .build());
   }
+
+  @Test
+  public void fromProtobufTest() {
+    RowMutation rowMutation =
+        RowMutation.create("fake-table", "fake-key")
+            .setCell("fake-family", "fake-qualifier-1", "fake-value")
+            .setCell("fake-family", "fake-qualifier-2", 30_000L, "fake-value-2");
+
+    MutateRowRequest protoRequest = rowMutation.toProto(REQUEST_CONTEXT);
+    RowMutation actualRequest = RowMutation.fromProtobuf(protoRequest);
+
+    assertThat(actualRequest.toProto(REQUEST_CONTEXT)).isEqualTo(protoRequest);
+  }
 }