fix: fix BidiBlobWriteSessionConfigs to respect preconditions (#2481)

BenWhitehead · web-flow · commit 955d78af0fec · 2024-04-01T14:54:28.000-04:00
diff --git a/google-cloud-storage/src/main/java/com/google/cloud/storage/UnifiedOpts.java b/google-cloud-storage/src/main/java/com/google/cloud/storage/UnifiedOpts.java
@@ -598,6 +598,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getObjectChecksumsBuilder().setCrc32C(val);
+        return b;
+      };
+    }
+
     @Override
     public int hashCode() {
       return Objects.hash(val);
@@ -753,6 +761,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        customerSuppliedKey(b.getCommonObjectRequestParamsBuilder(), val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<UpdateObjectRequest.Builder> updateObject() {
       return b -> {
@@ -1045,6 +1061,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getWriteObjectSpecBuilder().setIfGenerationMatch(val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<ReadObjectRequest.Builder> readObject() {
       return b -> b.setIfGenerationMatch(val);
@@ -1106,6 +1130,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getWriteObjectSpecBuilder().setIfGenerationNotMatch(val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<ReadObjectRequest.Builder> readObject() {
       return b -> b.setIfGenerationNotMatch(val);
@@ -1157,6 +1189,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getWriteObjectSpecBuilder().getResourceBuilder().setKmsKey(val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<ComposeObjectRequest.Builder> composeObject() {
       return b -> b.setKmsKey(val);
@@ -1217,6 +1257,15 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getObjectChecksumsBuilder()
+            .setMd5Hash(ByteString.copyFrom(BaseEncoding.base64().decode(val)));
+        return b;
+      };
+    }
+
     @Override
     public int hashCode() {
       return Objects.hash(val);
@@ -1252,6 +1301,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getWriteObjectSpecBuilder().setIfMetagenerationMatch(val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<ReadObjectRequest.Builder> readObject() {
       return b -> b.setIfMetagenerationMatch(val);
@@ -1337,6 +1394,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getWriteObjectSpecBuilder().setIfMetagenerationNotMatch(val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<ReadObjectRequest.Builder> readObject() {
       return b -> b.setIfMetagenerationNotMatch(val);
@@ -1452,6 +1517,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getWriteObjectSpecBuilder().setPredefinedAcl(val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<UpdateObjectRequest.Builder> updateObject() {
       return b -> b.setPredefinedAcl(val);
@@ -1704,6 +1777,14 @@ public Mapper<WriteObjectRequest.Builder> writeObject() {
       };
     }
 
+    @Override
+    public Mapper<BidiWriteObjectRequest.Builder> bidiWriteObject() {
+      return b -> {
+        b.getWriteObjectSpecBuilder().getResourceBuilder().setContentType(val);
+        return b;
+      };
+    }
+
     @Override
     public Mapper<UpdateObjectRequest.Builder> updateObject() {
       return b -> {
