one default array registry (#2701)

Author: onursatici

bench-vortex/src/clickbench.rs

@@ -20,8 +20,7 @@ use vortex::TryIntoArray;
 use vortex::dtype::DType;
 use vortex::dtype::arrow::FromArrowType;
 use vortex::error::{VortexError, vortex_err};
-use vortex::file::{DEFAULT_REGISTRY, VORTEX_FILE_EXTENSION, VortexWriteOptions};
-use vortex::layout::{LayoutRegistry, LayoutRegistryExt};
+use vortex::file::{VORTEX_FILE_EXTENSION, VortexWriteOptions};
 use vortex::stream::ArrayStreamAdapter;
 use vortex_datafusion::persistent::VortexFormat;
 
@@ -222,10 +221,7 @@ pub async fn register_vortex_files(
         .try_collect::<Vec<_>>()
         .await?;
 
-    let format = Arc::new(VortexFormat::new(
-        DEFAULT_REGISTRY.clone(),
-        Arc::new(LayoutRegistry::default()),
-    ));
+    let format = Arc::new(VortexFormat::default());
     let table_path = vortex_dir
         .to_str()
         .ok_or_else(|| vortex_err!("Path is not valid UTF-8"))?;

pyvortex/src/registry.rs

@@ -45,7 +45,7 @@ pub(crate) struct PyRegistry {
 impl PyRegistry {
     #[new]
     fn new() -> Self {
-        let mut array = ArrayRegistry::default();
+        let mut array = ArrayRegistry::canonical_only();
         array.register_many(DEFAULT_REGISTRY.vtables().cloned());
         let layout = LayoutRegistry::default();
         Self {

vortex-array/src/context.rs

@@ -17,8 +17,8 @@ use crate::vtable::VTableRef;
 pub type ArrayContext = VTableContext<VTableRef>;
 pub type ArrayRegistry = VTableRegistry<VTableRef>;
 
-impl Default for ArrayRegistry {
-    fn default() -> Self {
+impl ArrayRegistry {
+    pub fn canonical_only() -> Self {
         let mut this = Self::empty();
 
         // Register the canonical encodings

vortex-datafusion/src/persistent/format.rs

@@ -118,7 +118,7 @@ impl FileFormatFactory for VortexFormatFactory {
 impl Default for VortexFormat {
     fn default() -> Self {
         Self::new(
-            Arc::new(ArrayRegistry::default()),
+            DEFAULT_REGISTRY.clone(),
             Arc::new(LayoutRegistry::default()),
         )
     }

vortex-file/src/lib.rs

@@ -159,7 +159,7 @@ mod forever_constant {
 /// A default registry containing the built-in Vortex encodings and layouts.
 pub static DEFAULT_REGISTRY: LazyLock<Arc<ArrayRegistry>> = LazyLock::new(|| {
     // Register the compressed encodings that Vortex ships with.
-    let mut registry = ArrayRegistry::default();
+    let mut registry = ArrayRegistry::canonical_only();
     registry.register_many([
         ALPEncoding.vtable(),
         ALPRDEncoding.vtable(),

vortex-ipc/src/iterator.rs

@@ -166,7 +166,9 @@ mod test {
             .collect_to_buffer()
             .unwrap();
 
-        let reader = SyncIPCReader::try_new(Cursor::new(ipc_buffer), Default::default()).unwrap();
+        let reader =
+            SyncIPCReader::try_new(Cursor::new(ipc_buffer), ArrayRegistry::canonical_only())
+                .unwrap();
 
         assert_eq!(reader.dtype(), array.dtype());
         let result = reader.read_all().unwrap().to_primitive().unwrap();

vortex-ipc/src/messages/decoder.rs

@@ -168,7 +168,7 @@ mod test {
             ipc_bytes.extend_from_slice(buf.as_ref());
         }
 
-        let mut decoder = MessageDecoder::new(ArrayRegistry::default());
+        let mut decoder = MessageDecoder::new(ArrayRegistry::canonical_only());
 
         // Since we provide all bytes up-front, we should never hit a NeedMore.
         let mut buffer = BytesMut::from(ipc_bytes.as_ref());

vortex-ipc/src/stream.rs

@@ -201,9 +201,10 @@ mod test {
             .await
             .unwrap();
 
-        let reader = AsyncIPCReader::try_new(Cursor::new(ipc_buffer), Default::default())
-            .await
-            .unwrap();
+        let reader =
+            AsyncIPCReader::try_new(Cursor::new(ipc_buffer), ArrayRegistry::canonical_only())
+                .await
+                .unwrap();
 
         assert_eq!(reader.dtype(), array.dtype());
         let result = reader.read_all().await.unwrap().to_primitive().unwrap();