x

Author: forsaken628

src/query/service/src/pipelines/processors/transforms/hash_join/hash_join_build_state.rs

@@ -744,6 +744,7 @@ impl HashJoinBuildState {
                     "Aborted query, because the hash table is uninitialized.",
                 ));
             }
+            HashJoinHashTable::NestedLoop(_) => unreachable!(),
             HashJoinHashTable::SkipDuplicatesSerializer(table) => insert_binary_key! {
               &mut table.hash_table, &table.hash_method, chunk, build_keys, valids, chunk_index as u32, entry_size, &mut local_raw_entry_spaces,
             },

src/query/service/src/pipelines/processors/transforms/hash_join/hash_join_probe_state.rs

@@ -165,6 +165,7 @@ impl HashJoinProbeState {
                 // Continue to probe hash table and process data blocks.
                 self.result_blocks(probe_state, keys, &table.hash_table)
             }
+            HashJoinHashTable::NestedLoop(_) => unreachable!(),
             HashJoinHashTable::Null => Err(ErrorCode::AbortedQuery(
                 "Aborted query, because the hash table is uninitialized.",
             )),
@@ -376,6 +377,7 @@ impl HashJoinProbeState {
                 // Continue to probe hash table and process data blocks.
                 self.result_blocks(probe_state, keys, &table.hash_table)
             }
+            HashJoinHashTable::NestedLoop(_) => unreachable!(),
             HashJoinHashTable::Null => Err(ErrorCode::AbortedQuery(
                 "Aborted query, because the hash table is uninitialized.",
             )),

src/query/service/src/pipelines/processors/transforms/hash_join/hash_join_state.rs

@@ -78,6 +78,7 @@ pub struct FixedKeyHashJoinHashTable<
 
 pub enum HashJoinHashTable {
     Null,
+    NestedLoop(Vec<DataBlock>),
     Serializer(SerializerHashJoinHashTable),
     SkipDuplicatesSerializer(SkipDuplicatesSerializerHashJoinHashTable),
     SingleBinary(SingleBinaryHashJoinHashTable),
@@ -348,9 +349,9 @@ impl HashJoinState {
 }
 
 impl HashJoinHashTable {
-    pub fn len(&self) -> usize {
-        match self {
-            HashJoinHashTable::Null => 0,
+    pub fn size(&self) -> Option<usize> {
+        let n = match self {
+            HashJoinHashTable::Null | HashJoinHashTable::NestedLoop(_) => return None,
             HashJoinHashTable::Serializer(table) => table.hash_table.len(),
             HashJoinHashTable::SingleBinary(table) => table.hash_table.len(),
             HashJoinHashTable::KeysU8(table) => table.hash_table.len(),
@@ -367,11 +368,7 @@ impl HashJoinHashTable {
             HashJoinHashTable::SkipDuplicatesKeysU64(table) => table.hash_table.len(),
             HashJoinHashTable::SkipDuplicatesKeysU128(table) => table.hash_table.len(),
             HashJoinHashTable::SkipDuplicatesKeysU256(table) => table.hash_table.len(),
-        }
-    }
-
-    #[allow(dead_code)]
-    pub fn is_empty(&self) -> bool {
-        self.len() == 0
+        };
+        Some(n)
     }
 }

src/query/service/src/pipelines/processors/transforms/hash_join/result_blocks.rs

@@ -54,7 +54,9 @@ impl HashJoinProbeState {
             JoinType::InnerAny | JoinType::RightAny
         ) {
             let hash_table = unsafe { &*self.hash_join_state.hash_table.get() };
-            probe_state.used_once = Some(MutableBitmap::from_len_zeroed(hash_table.len()))
+            probe_state.used_once = Some(MutableBitmap::from_len_zeroed(
+                hash_table.size().unwrap_or_default(),
+            ))
         }
         let no_other_predicate = self
             .hash_join_state