fix

Author: zhyass

src/query/expression/src/block.rs

@@ -209,6 +209,11 @@ impl DataBlock {
         &self.columns
     }
 
+    #[inline]
+    pub fn take_columns(self) -> Vec<BlockEntry> {
+        self.columns
+    }
+
     #[inline]
     pub fn columns_mut(&mut self) -> &mut [BlockEntry] {
         &mut self.columns

src/query/expression/src/converts/arrow/to.rs

@@ -227,11 +227,11 @@ impl DataBlock {
         let mut arrays = Vec::with_capacity(self.columns().len());
         for (entry, arrow_field) in self
             .consume_convert_to_full()
-            .columns()
-            .iter()
+            .take_columns()
+            .into_iter()
             .zip(arrow_schema.fields())
         {
-            let column = entry.value.to_owned().into_column().unwrap();
+            let column = entry.value.into_column().unwrap();
             let column = column.maybe_gc();
             let array = column.into_arrow_rs();
 

src/query/storages/fuse/src/io/write/block_writer.rs

@@ -97,9 +97,9 @@ pub fn serialize_block(
 
             let block = block.consume_convert_to_full();
             let batch: Vec<Column> = block
-                .columns()
-                .iter()
-                .map(|x| x.value.as_column().unwrap().clone())
+                .take_columns()
+                .into_iter()
+                .map(|x| x.value.into_column().unwrap())
                 .collect();
 
             writer.start()?;