try try

Signed-off-by: Bugen Zhao <[email protected]>
risingwavelabs · Sep 3, 2024 · 8a144d7 · 8a144d7
1 parent a7d946f
commit 8a144d7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/common/src/hash/table_distribution.rs b/src/common/src/hash/table_distribution.rs
@@ -75,7 +75,7 @@ impl TableDistribution {
     ) -> Self {
         let compute_vnode = if let Some(vnode_col_idx_in_pk) = vnode_col_idx_in_pk {
             ComputeVnode::VnodeColumnIndex {
-                vnodes: vnodes.expect("vnodes must be `Some` as vnode column is set"),
+                vnodes: vnodes.unwrap_or_else(|| Bitmap::singleton().into()),
                 vnode_col_idx_in_pk,
             }
         } else if !dist_key_in_pk_indices.is_empty() {
@@ -139,7 +139,7 @@ impl TableDistribution {
         match &self.compute_vnode {
             ComputeVnode::DistKeyIndices { vnodes, .. } => vnodes,
             ComputeVnode::VnodeColumnIndex { vnodes, .. } => vnodes,
-            ComputeVnode::Singleton => &*SINGLETON_VNODES,
+            ComputeVnode::Singleton => &SINGLETON_VNODES,
         }
     }