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

c

Browse Source 
Signed-off-by: discord9 <discord9@163.com>
This commit is contained in:
discord9
2025-12-22 18:39:14 +08:00
parent 5b50b4824d
commit 709ccd3e31
1 changed files with 23 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
src/query/src/part_sort.rs
View File

@@ -1282,33 +1282,30 @@ mod test {
.unwrap(),
)
}
} else {
if chain_windowed_sort {
// if chain_windowed_sort is true, we expect a single output batch sorted across all ranges
let mut sorted = output_data.into_iter().flatten().collect_vec();
if descending {
sorted.sort_by(|a, b| b.cmp(a));
} else {
sorted.sort();
}
Some(
DfRecordBatch::try_new(schema.clone(), vec![new_ts_array(unit, sorted)])
.unwrap(),
)
} else if chain_windowed_sort {
// if chain_windowed_sort is true, we expect a single output batch sorted across all ranges
let mut sorted = output_data.into_iter().flatten().collect_vec();
if descending {
sorted.sort_by(|a, b| b.cmp(a));
} else {
let batches = output_data
.into_iter()
.map(|a| {
DfRecordBatch::try_new(schema.clone(), vec![new_ts_array(unit, a)])
.unwrap()
})
.collect_vec();
if batches.is_empty() {
None
} else {
Some(concat_batches(&schema, &batches).unwrap())
}
sorted.sort();
}
Some(
DfRecordBatch::try_new(schema.clone(), vec![new_ts_array(unit, sorted)])
.unwrap(),
)
} else {
let batches = output_data
.into_iter()
.map(|a| {
DfRecordBatch::try_new(schema.clone(), vec![new_ts_array(unit, a)]).unwrap()
})
.collect_vec();
if batches.is_empty() {
None
} else {
Some(concat_batches(&schema, &batches).unwrap())
}
};