Add typemod to Column (paradedb#2)

* Add typemod to Column * small change
Weijun-H · Aug 23, 2024 · ee86506 · ee86506
1 parent 317e8ef
commit ee86506
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/supabase-wrappers/src/interface.rs b/supabase-wrappers/src/interface.rs
@@ -302,6 +302,9 @@ pub struct Column {
 
     /// column type OID, can be used to match pg_sys::BuiltinOid
     pub type_oid: Oid,
+
+    /// column type mod
+    pub type_mod: i32,
 }
 
 /// A restiction value used in [`Qual`], either a [`Cell`] or an array of [`Cell`]

diff --git a/supabase-wrappers/src/utils.rs b/supabase-wrappers/src/utils.rs
@@ -246,6 +246,7 @@ pub(super) unsafe fn extract_target_columns(
         let rte = pg_sys::planner_rt_fetch((*var).varno as _, root);
         let attno = (*var).varattno;
         let attname = pg_sys::get_attname((*rte).relid, attno, true);
+        let type_mod = (*var).vartypmod;
         if !attname.is_null() {
             // generated column is not supported
             if pg_sys::get_attgenerated((*rte).relid, attno) > 0 {
@@ -258,6 +259,7 @@ pub(super) unsafe fn extract_target_columns(
                 name: CStr::from_ptr(attname).to_str().unwrap().to_owned(),
                 num: attno as usize,
                 type_oid,
+                type_mod,
             });
         }
     }