Skip to content

Commit 35406f3

Browse files
committed
upsert: refactor struct-looking enum into struct
Signed-off-by: Petros Angelatos <[email protected]>
1 parent ae822d2 commit 35406f3

File tree

3 files changed

+113
-165
lines changed

3 files changed

+113
-165
lines changed

src/storage/src/upsert.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ use crate::metrics::upsert::UpsertMetrics;
4848
use crate::storage_state::StorageInstanceContext;
4949
use crate::upsert_continual_feedback;
5050
use types::{
51-
BincodeOpts, StateValue, UpsertState, UpsertStateBackend, Value, consolidating_merge_function,
51+
BincodeOpts, StateValue, UpsertState, UpsertStateBackend, consolidating_merge_function,
5252
upsert_bincode_opts,
5353
};
5454

@@ -607,15 +607,15 @@ async fn drain_staged_input<S, G, T, FromTime, E>(
607607
if let Some(old_value) =
608608
existing_value.replace(StateValue::finalized_value(value.clone()))
609609
{
610-
if let Value::FinalizedValue(old_value) = old_value.into_decoded() {
610+
if let Some(old_value) = old_value.into_decoded().finalized {
611611
output_updates.push((old_value, ts.clone(), Diff::MINUS_ONE));
612612
}
613613
}
614614
output_updates.push((value, ts, Diff::ONE));
615615
}
616616
None => {
617617
if let Some(old_value) = existing_value.take() {
618-
if let Value::FinalizedValue(old_value) = old_value.into_decoded() {
618+
if let Some(old_value) = old_value.into_decoded().finalized {
619619
output_updates.push((old_value, ts, Diff::MINUS_ONE));
620620
}
621621
}

src/storage/src/upsert/memory.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
use std::collections::HashMap;
1818

1919
use itertools::Itertools;
20+
use mz_ore::cast::CastFrom;
2021

2122
use super::UpsertKey;
2223
use super::types::{
@@ -93,7 +94,7 @@ where
9394
stats.processed_gets += 1;
9495
let value = self.state.get(&key).cloned();
9596
let metadata = value.as_ref().map(|v| ValueMetadata {
96-
size: v.memory_size(),
97+
size: u64::cast_from(v.memory_size()),
9798
is_tombstone: v.is_tombstone(),
9899
});
99100
stats.processed_gets_size += metadata.map_or(0, |m| m.size);

0 commit comments

Comments
 (0)