Statement::column_logical_type

Author: diegoimbert

crates/duckdb/src/column.rs

@@ -2,7 +2,7 @@ use std::str;
 
 use arrow::datatypes::DataType;
 
-use crate::{Error, Result, Statement};
+use crate::{core::LogicalTypeHandle, Error, Result, Statement};
 
 /// Information about a column of a DuckDB query.
 #[derive(Debug)]
@@ -160,6 +160,14 @@ impl Statement<'_> {
     pub fn column_type(&self, idx: usize) -> DataType {
         self.stmt.column_type(idx)
     }
+
+    /// Returns the declared logical data type of the column.
+    ///
+    /// # Caveats
+    /// Panics if the query has not been [`execute`](Statement::execute)d yet.
+    pub fn column_logical_type(&self, idx: usize) -> LogicalTypeHandle {
+        self.stmt.column_logical_type(idx)
+    }
 }
 
 #[cfg(test)]

crates/duckdb/src/raw_statement.rs

@@ -9,7 +9,7 @@ use arrow::{
 use super::{ffi, Result};
 #[cfg(feature = "polars")]
 use crate::arrow2;
-use crate::{error::result_from_duckdb_arrow, Error};
+use crate::{core::LogicalTypeHandle, error::result_from_duckdb_arrow, Error};
 
 /// Private newtype for DuckDB prepared statements that finalize themselves when dropped.
 ///
@@ -215,6 +215,12 @@ impl RawStatement {
         self.schema().field(idx).data_type().to_owned()
     }
 
+    #[inline]
+    pub fn column_logical_type(&self, idx: usize) -> LogicalTypeHandle {
+        let ptr = unsafe { ffi::duckdb_column_logical_type(&mut self.duckdb_result.unwrap() as *mut _, idx as u64) };
+        LogicalTypeHandle { ptr }
+    }
+
     #[inline]
     pub fn schema(&self) -> SchemaRef {
         self.schema.clone().unwrap()