update snapshot producer

apache · Mar 26, 2024 · d100dbc · d100dbc
1 parent b7c6e9b
commit d100dbc
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/core/src/main/java/org/apache/iceberg/SnapshotProducer.java b/core/src/main/java/org/apache/iceberg/SnapshotProducer.java
@@ -237,10 +237,31 @@ public Snapshot apply() {
 
     OutputFile manifestList = manifestListPath();
 
+    EncryptionManager encryptionManager = ops.encryption();
+    EncryptedOutputFile encryptedManifestList = encryptionManager.encrypt(manifestList);
+    ByteBuffer manifestListEncryptionKey = encryptedManifestList.keyMetadata().encryptionKey();
+    String manifestListKeyMetadata = null;
+    if (manifestListEncryptionKey != null) {
+      Preconditions.checkArgument(
+          encryptionManager instanceof StandardEncryptionManager,
+          "Encryption manager for encrypted manifest list files can currently only be an instance of "
+              + StandardEncryptionManager.class);
+      ByteBuffer wrappedEncryptionKey =
+          ((StandardEncryptionManager) encryptionManager).wrapKey(manifestListEncryptionKey);
+
+      ByteBuffer manifestListAADPrefix = encryptedManifestList.keyMetadata().aadPrefix();
+      manifestListKeyMetadata =
+          Base64.getEncoder()
+              .encodeToString(
+                  EncryptionUtil.createKeyMetadata(wrappedEncryptionKey, manifestListAADPrefix)
+                      .buffer()
+                      .array());
+    }
+
     try (ManifestListWriter writer =
         ManifestLists.write(
             ops.current().formatVersion(),
-            manifestList,
+            encryptedManifestList.encryptingOutputFile(),
             snapshotId(),
             parentSnapshotId,
             sequenceNumber)) {
@@ -269,7 +290,8 @@ public Snapshot apply() {
         operation(),
         summary(base),
         base.currentSchemaId(),
-        manifestList.location());
+        manifestList.location(),
+        manifestListKeyMetadata);
   }
 
   protected abstract Map<String, String> summary();