From 37a7ec74777ea1ac516465b883bac5156541cce9 Mon Sep 17 00:00:00 2001
From: PThorpe92 <preston@turso.tech>
Date: Mon, 25 Aug 2025 09:50:57 -0400
Subject: [PATCH] Update append_frames_vectored to use new encryption_ctx and
 apply review

---
 core/storage/pager.rs | 7 ++++++-
 core/storage/wal.rs   | 6 +++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/core/storage/pager.rs b/core/storage/pager.rs
index 6290129d6..1149acf61 100644
--- a/core/storage/pager.rs
+++ b/core/storage/pager.rs
@@ -1291,7 +1291,12 @@ impl Pager {
         if !pages.is_empty() {
             let c = wal
                 .borrow_mut()
-                .append_frames_vectored(pages, page_sz, commit_frame)?;
+                .append_frames_vectored(pages, page_sz, commit_frame)
+                .inspect_err(|_| {
+                    for c in completions.iter() {
+                        c.abort();
+                    }
+                })?;
             completions.push(c);
         }
         Ok(completions)
diff --git a/core/storage/wal.rs b/core/storage/wal.rs
index 8d28827cd..15fd04afe 100644
--- a/core/storage/wal.rs
+++ b/core/storage/wal.rs
@@ -1442,9 +1442,9 @@ impl Wal for WalFile {
             let plain = page.get_contents().as_ptr();
 
             let data_to_write: std::borrow::Cow<[u8]> = {
-                let key = self.encryption_key.borrow();
-                if let Some(k) = key.as_ref() {
-                    Cow::Owned(encrypt_page(plain, page_id as usize, k)?)
+                let ectx = self.encryption_ctx.borrow();
+                if let Some(ctx) = ectx.as_ref() {
+                    Cow::Owned(ctx.encrypt_page(plain, page_id as usize)?)
                 } else {
                     Cow::Borrowed(plain)
                 }