fix: glaredb table drop

aljazerzen · Sep 6, 2024 · 41b02fd · 41b02fd
1 parent 8007dfd
commit 41b02fd
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/connector_arrow/src/postgres/schema.rs b/connector_arrow/src/postgres/schema.rs
@@ -102,6 +102,16 @@ impl SchemaEdit for super::PostgresConnection {
 
         match res {
             Ok(_) => Ok(()),
+            Err(e)
+                if matches!(e.code(), Some(&SqlState::UNDEFINED_TABLE)) ||
+                    // GlareDB will return such errors
+                    e.as_db_error().map_or(false, |e| {
+                        e.message().starts_with("Error during planning: Table ")
+                            && e.message().ends_with(" does not exist")
+                    }) =>
+            {
+                Err(TableDropError::TableNonexistent)
+            }
             Err(e) if matches!(e.code(), Some(&SqlState::UNDEFINED_TABLE)) => {
                 Err(TableDropError::TableNonexistent)
             }