rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-01-07 05:42:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: fix panic in TimeSeriesParquetReader

Browse Source
This commit is contained in:
evenyag
2025-03-10 14:13:37 +08:00
parent 63bc544514
commit 79d249f5fa
1 changed files with 25 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
src/sst-convert/src/reader/remote_write.rs
View File

@@ -243,32 +243,35 @@ impl TimeSeriesParquetReader {
if let Some(array) = &self.array {
if self.row_index >= array.len() {
let Some(record_batch) = self
.stream
.next()
.await
.transpose()
.context(ReadParquetSnafu { path: &self.path })
.context(MitoSnafu)?
else {
self.eof = true;
return Ok(None);
};
self.row_index = 0;
self.array = Some(
record_batch
.column(0)
.as_any()
.downcast_ref::<StringArray>()
.context(MissingColumnSnafu {
column_name: "remote write json column",
})?
.clone(),
);
self.array = None;
}
}
if self.array.is_none() {
let Some(record_batch) = self
.stream
.next()
.await
.transpose()
.context(ReadParquetSnafu { path: &self.path })
.context(MitoSnafu)?
else {
self.eof = true;
return Ok(None);
};
self.array = Some(
record_batch
.column(0)
.as_any()
.downcast_ref::<StringArray>()
.context(MissingColumnSnafu {
column_name: "remote write json column",
})?
.clone(),
);
}
let array = self.array.as_ref().unwrap();
let value = array.value(self.row_index);
let time_series = serde_json::from_str(value).context(JsonSnafu)?;