fix: fix name on marco

tonbo-io · Jul 26, 2024 · 6be50ce · 6be50ce
1 parent 9e5ae4b
commit 6be50ce
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/morseldb_marco/src/lib.rs b/src/morseldb_marco/src/lib.rs
@@ -455,7 +455,7 @@ pub fn morsel_record(_args: TokenStream, input: TokenStream) -> TokenStream {
 
                 #(#from_record_batch_fields)*
 
-                let record = TestRef {
+                let record = #struct_ref_name {
                     #(#field_names)*
                 };
                 ::morseldb::record::internal::InternalRecordRef::new(ts, record, null)
@@ -495,7 +495,7 @@ pub fn morsel_record(_args: TokenStream, input: TokenStream) -> TokenStream {
             type Builder = #struct_builder_name;
 
             fn builder(capacity: usize) -> Self::Builder {
-                TestBuilder {
+                #struct_builder_name {
                     #(#builder_init_fields)*
 
                     _null: ::arrow::array::BooleanBufferBuilder::new(capacity),
@@ -536,8 +536,8 @@ pub fn morsel_record(_args: TokenStream, input: TokenStream) -> TokenStream {
             _ts: ::arrow::array::UInt32Builder,
         }
 
-        impl ::morseldb::inmem::immutable::Builder<TestImmutableArrays> for #struct_builder_name {
-            fn push(&mut self, key: Timestamped<&str>, row: Option<TestRef>) {
+        impl ::morseldb::inmem::immutable::Builder<#struct_arrays_name> for #struct_builder_name {
+            fn push(&mut self, key: Timestamped<&str>, row: Option<#struct_ref_name>) {
                 #builder_append_primary_key
                 match row {
                     Some(row) => {

diff --git a/src/record/str.rs b/src/record/str.rs
@@ -1,4 +1,4 @@
-use std::{mem::size_of, sync::Arc};
+use std::{mem, sync::Arc};
 
 use arrow::{
     array::{
@@ -153,7 +153,7 @@ impl Builder<StringColumns> for StringColumnsBuilder {
 
     fn written_size(&self) -> usize {
         self._null.as_slice().len()
-            + self._ts.values_slice().len() * size_of::<u32>()
+            + mem::size_of_val(self._ts.values_slice())
             + self.string.values_slice().len()
     }