Back VortexSession with ArcSwap for mutable but less errorprone session

Signed-off-by: Robert Kruszewski <github@robertk.io>

Author: claude

Cargo.lock

@@ -133,9 +133,10 @@ pub trait ExecuteParentKernel<V: VTable> {
 }
 
 pub fn initialize(session: &VortexSession) {
-    session
-        .kernels()
-        .register_execute_parent_kernel(parent_id, Child, Kernel);
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
+    kernels.register_execute_parent_kernel(parent_id, Child, Kernel);
 }
 ```
 

docs/developer-guide/internals/execution.md

@@ -95,14 +95,13 @@ all built-in components and encodings:
 let session = VortexSession::default();
 ```
 
-For tests or specialized use-cases, sessions can be assembled from individual components using
-the `.with::<T>()` builder:
+For tests or specialized use-cases, sessions can be assembled from individual components by
+starting from an empty session and chaining the `.with::<T>()` helpers:
 
 ```rust
-let session = VortexSession::builder()
+let session = VortexSession::empty()
     .with::<ArraySession>()
     .with::<LayoutSession>()
     .with::<ScalarFnSession>()
-    .with::<RuntimeSession>()
-    .build();
+    .with::<RuntimeSession>();
 ```

docs/developer-guide/internals/session.md

@@ -23,7 +23,9 @@ use vortex_session::VortexSession;
 use crate::ALP;
 
 pub(super) fn initialize(session: &VortexSession) {
-    let kernels = session.kernels();
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
     kernels.register_execute_parent_kernel(Binary.id(), ALP, CompareExecuteAdaptor(ALP));
     kernels.register_execute_parent_kernel(Filter.id(), ALP, FilterExecuteAdaptor(ALP));
     kernels.register_execute_parent_kernel(Mask.id(), ALP, MaskExecuteAdaptor(ALP));

encodings/alp/src/alp/rules.rs

@@ -14,7 +14,9 @@ use vortex_session::VortexSession;
 use crate::alp_rd::ALPRD;
 
 pub(crate) fn initialize(session: &VortexSession) {
-    let kernels = session.kernels();
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
     kernels.register_execute_parent_kernel(Slice.id(), ALPRD, SliceExecuteAdaptor(ALPRD));
     kernels.register_execute_parent_kernel(Filter.id(), ALPRD, FilterExecuteAdaptor(ALPRD));
     kernels.register_execute_parent_kernel(Dict.id(), ALPRD, TakeExecuteAdaptor(ALPRD));

encodings/alp/src/alp_rd/kernel.rs

@@ -15,7 +15,9 @@ use vortex_session::VortexSession;
 use crate::ByteBool;
 
 pub(crate) fn initialize(session: &VortexSession) {
-    let kernels = session.kernels();
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
     kernels.register_execute_parent_kernel(Binary.id(), ByteBool, BooleanExecuteAdaptor(ByteBool));
     kernels.register_execute_parent_kernel(Cast.id(), ByteBool, CastExecuteAdaptor(ByteBool));
     kernels.register_execute_parent_kernel(Dict.id(), ByteBool, TakeExecuteAdaptor(ByteBool));

encodings/bytebool/src/kernel.rs

@@ -13,7 +13,9 @@ use vortex_session::VortexSession;
 use crate::DateTimeParts;
 
 pub(crate) fn initialize(session: &VortexSession) {
-    let kernels = session.kernels();
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
     kernels.register_execute_parent_kernel(
         Binary.id(),
         DateTimeParts,

encodings/datetime-parts/src/compute/kernel.rs

@@ -13,7 +13,9 @@ use vortex_session::VortexSession;
 use crate::DecimalByteParts;
 
 pub(crate) fn initialize(session: &VortexSession) {
-    let kernels = session.kernels();
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
     kernels.register_execute_parent_kernel(
         Binary.id(),
         DecimalByteParts,

encodings/decimal-byte-parts/src/decimal_byte_parts/compute/kernel.rs

@@ -16,7 +16,9 @@ use crate::OnPair;
 
 // TODO: implement ListExecute & TakeExecute for OnPair
 pub(super) fn initialize(session: &VortexSession) {
-    let kernels = session.kernels();
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
     kernels.register_execute_parent_kernel(Filter.id(), OnPair, FilterExecuteAdaptor(OnPair));
     kernels.register_execute_parent_kernel(Binary.id(), OnPair, CompareExecuteAdaptor(OnPair));
     kernels.register_execute_parent_kernel(

encodings/experimental/onpair/src/kernel.rs

@@ -21,7 +21,9 @@ use vortex_session::VortexSession;
 use crate::BitPacked;
 
 pub(crate) fn initialize(session: &VortexSession) {
-    let kernels = session.kernels();
+    let Some(kernels) = session.kernels() else {
+        return;
+    };
     kernels.register_execute_parent_kernel(
         Between.id(),
         BitPacked,