Fix table id when columns updated (#25)

helium · Oct 30, 2024 · ec09e02 · ec09e02
1 parent 55ed28e
commit ec09e02
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/protobuf-delta-lake-sink/src/main.rs b/protobuf-delta-lake-sink/src/main.rs
@@ -1,3 +1,8 @@
+use std::{
+    collections::{HashMap, HashSet},
+    sync::Arc,
+};
+
 use anyhow::{anyhow, Context, Result};
 use chrono::{NaiveDateTime, Utc};
 use clap::Parser;
@@ -16,10 +21,6 @@ use file_store::Settings;
 use futures::stream::{self, StreamExt};
 use protobuf::CodedInputStream;
 use serde_json::Value;
-use std::{
-    collections::{HashMap, HashSet},
-    sync::Arc,
-};
 pub use store::*;
 
 use crate::{
@@ -189,7 +190,8 @@ async fn main() -> Result<()> {
             vec!["date".to_string()],
             HashMap::new(),
         );
-        let meta = action::MetaData::try_from(metadata)?;
+        let mut meta = action::MetaData::try_from(metadata)?;
+        meta.id = table.get_metadata()?.id.clone();
         let actions = vec![Action::metaData(meta)];
         let storage = table.object_store();
         commit(storage.as_ref(), &actions, &table.state).await?;