From 53cfae1db400ec66e433516b73d49f2b69b595cd Mon Sep 17 00:00:00 2001
From: pedrocarlo <pedro.muniz.carlo@gmail.com>
Date: Fri, 22 Aug 2025 11:11:26 -0300
Subject: [PATCH] return Error from step if IO failed

---
 core/io/mod.rs   | 4 ++++
 core/types.rs    | 9 ++++++++-
 core/vdbe/mod.rs | 5 +++++
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/core/io/mod.rs b/core/io/mod.rs
index 992eabac0..341d1fe68 100644
--- a/core/io/mod.rs
+++ b/core/io/mod.rs
@@ -214,6 +214,10 @@ impl Completion {
         self.inner.result.get().is_some_and(|val| val.is_some())
     }
 
+    pub fn get_error(&self) -> Option<CompletionError> {
+        self.inner.result.get().and_then(|res| *res)
+    }
+
     /// Checks if the Completion completed or errored
     pub fn finished(&self) -> bool {
         self.inner.result.get().is_some()
diff --git a/core/types.rs b/core/types.rs
index 67fdbfdb8..cc9c78021 100644
--- a/core/types.rs
+++ b/core/types.rs
@@ -14,7 +14,7 @@ use crate::translate::plan::IterationDirection;
 use crate::vdbe::sorter::Sorter;
 use crate::vdbe::Register;
 use crate::vtab::VirtualTableCursor;
-use crate::{turso_assert, Completion, Result, IO};
+use crate::{turso_assert, Completion, CompletionError, Result, IO};
 use std::fmt::{Debug, Display};
 
 const MAX_REAL_SIZE: u8 = 15;
@@ -2502,6 +2502,13 @@ impl IOCompletions {
             IOCompletions::Many(completions) => completions.iter().for_each(|c| c.abort()),
         }
     }
+
+    pub fn get_error(&self) -> Option<CompletionError> {
+        match self {
+            IOCompletions::Single(c) => c.get_error(),
+            IOCompletions::Many(completions) => completions.iter().find_map(|c| c.get_error()),
+        }
+    }
 }
 
 #[derive(Debug)]
diff --git a/core/vdbe/mod.rs b/core/vdbe/mod.rs
index 141158d3b..f3a9425a8 100644
--- a/core/vdbe/mod.rs
+++ b/core/vdbe/mod.rs
@@ -460,6 +460,11 @@ impl Program {
                 if !io.finished() {
                     return Ok(StepResult::IO);
                 }
+                if let Some(err) = io.get_error() {
+                    let err = err.into();
+                    handle_program_error(&pager, &self.connection, &err)?;
+                    return Err(err);
+                }
                 state.io_completions = None;
             }
             // invalidate row