rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2025-12-23 06:30:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: convert to influxdb values directly from arrow (#7163)

Browse Source 
* refactor: convert to influxdb values directly from arrow

Signed-off-by: luofucong <luofc@foxmail.com>

* resolve PR comments

Signed-off-by: luofucong <luofc@foxmail.com>

* resolve PR comments

Signed-off-by: luofucong <luofc@foxmail.com>

---------

Signed-off-by: luofucong <luofc@foxmail.com>
This commit is contained in:
LFC
2025-11-03 15:52:37 +08:00
committed by GitHub
parent 5eab9a1be3
commit b7e834ab92
5 changed files with 303 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests-integration/src/tests/instance_kafka_wal_test.rs
View File

@@ -18,6 +18,8 @@ use std::sync::atomic::{AtomicU64, Ordering};
use client::DEFAULT_CATALOG_NAME;
use common_query::{Output, OutputData};
use datatypes::arrow::array::AsArray;
use datatypes::arrow::datatypes::TimestampMillisecondType;
use datatypes::vectors::{TimestampMillisecondVector, VectorRef};
use frontend::instance::Instance;
use itertools::Itertools;
@@ -224,9 +226,9 @@ async fn ensure_data_exists(tables: &[Table], instance: &Arc<Instance>) {
let queried = record_batches
.into_iter()
.flat_map(|rb| {
rb.rows()
.map(|row| row[0].as_timestamp().unwrap().value() as u64)
.collect::<Vec<_>>()
let array = rb.column(0).to_arrow_array();
let array = array.as_primitive::<TimestampMillisecondType>();
array.iter().flatten().map(|x| x as u64).collect::<Vec<_>>()
})
.collect::<Vec<_>>();
let inserted = table