fix(query): fix align state bug for aggregation

Author: sundy-li

src/query/functions-v2/src/aggregates/adaptors/aggregate_null_unary_adaptor.rs

@@ -97,7 +97,7 @@ impl<const NULLABLE_RESULT: bool> AggregateFunction for AggregateNullUnaryAdapto
     #[inline]
     fn state_layout(&self) -> Layout {
         let layout = self.nested.state_layout();
-        let add = usize::from(NULLABLE_RESULT);
+        let add = if NULLABLE_RESULT { layout.align() } else { 0 };
         Layout::from_size_align(layout.size() + add, layout.align()).unwrap()
     }
 

src/query/functions-v2/src/aggregates/adaptors/aggregate_null_variadic_adaptor.rs

@@ -98,7 +98,7 @@ impl<const NULLABLE_RESULT: bool> AggregateFunction
     #[inline]
     fn state_layout(&self) -> Layout {
         let layout = self.nested.state_layout();
-        let add = usize::from(NULLABLE_RESULT);
+        let add = if NULLABLE_RESULT { layout.align() } else { 0 };
         Layout::from_size_align(layout.size() + add, layout.align()).unwrap()
     }
 

src/query/functions-v2/src/aggregates/adaptors/aggregate_ornull_adaptor.rs

@@ -90,7 +90,7 @@ impl AggregateFunction for AggregateFunctionOrNullAdaptor {
     #[inline]
     fn state_layout(&self) -> std::alloc::Layout {
         let layout = self.inner.state_layout();
-        Layout::from_size_align(layout.size() + 1, layout.align()).unwrap()
+        Layout::from_size_align(layout.size() + layout.align(), layout.align()).unwrap()
     }
 
     #[inline]

src/query/functions-v2/src/aggregates/aggregate_combinator_distinct.rs

@@ -70,6 +70,7 @@ where State: DistinctStateFunc
 
     fn state_layout(&self) -> Layout {
         let layout = Layout::new::<State>();
+
         let netesed = self.nested.state_layout();
         Layout::from_size_align(layout.size() + netesed.size(), layout.align()).unwrap()
     }

src/query/functions/src/aggregates/adaptors/aggregate_null_unary_adaptor.rs

@@ -94,7 +94,7 @@ impl<const NULLABLE_RESULT: bool> AggregateFunction for AggregateNullUnaryAdapto
     #[inline]
     fn state_layout(&self) -> Layout {
         let layout = self.nested.state_layout();
-        let add = usize::from(NULLABLE_RESULT);
+        let add = if NULLABLE_RESULT { layout.align() } else { 0 };
         Layout::from_size_align(layout.size() + add, layout.align()).unwrap()
     }
 

src/query/functions/src/aggregates/adaptors/aggregate_null_variadic_adaptor.rs

@@ -95,7 +95,7 @@ impl<const NULLABLE_RESULT: bool> AggregateFunction
     #[inline]
     fn state_layout(&self) -> Layout {
         let layout = self.nested.state_layout();
-        let add = usize::from(NULLABLE_RESULT);
+        let add = if NULLABLE_RESULT { layout.align() } else { 0 };
         Layout::from_size_align(layout.size() + add, layout.align()).unwrap()
     }
 

src/query/functions/src/aggregates/adaptors/aggregate_ornull_adaptor.rs

@@ -92,7 +92,7 @@ impl AggregateFunction for AggregateFunctionOrNullAdaptor {
     #[inline]
     fn state_layout(&self) -> std::alloc::Layout {
         let layout = self.inner.state_layout();
-        Layout::from_size_align(layout.size() + 1, layout.align()).unwrap()
+        Layout::from_size_align(layout.size() + layout.align(), layout.align()).unwrap()
     }
 
     #[inline]

src/query/functions/src/aggregates/aggregate_combinator_distinct.rs

@@ -66,6 +66,7 @@ where State: DistinctStateFunc
 
     fn state_layout(&self) -> Layout {
         let layout = Layout::new::<State>();
+
         let netesed = self.nested.state_layout();
         Layout::from_size_align(layout.size() + netesed.size(), layout.align()).unwrap()
     }