refactor: use fetch_chunk to fetch generation

Also, add GenerationChunk::from_data_chunk function.

1 files changed, 6 insertions, 0 deletions

diff --git a/src/chunk.rs b/src/chunk.rs
index 7fdeccb..4917b60 100644
--- a/src/chunk.rs
+++ b/src/chunk.rs
@@ -35,6 +35,12 @@ impl GenerationChunk {
         Self { chunk_ids }
     }
 
+    pub fn from_data_chunk(chunk: &DataChunk) -> anyhow::Result<Self> {
+        let data = chunk.data();
+        let data = std::str::from_utf8(data)?;
+        Ok(serde_json::from_str(data)?)
+    }
+
     pub fn is_empty(&self) -> bool {
         self.chunk_ids.is_empty()
     }