qql-cli 0.4.0

Command-line interface, REPL, converter, and migration tools for QQL
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

//! Cursor extraction and point ID parsing for scroll pagination.

use qql::executor::BackendResponse;
use qql_plan::semantic::PlanPointId;
use serde_json::Value;

/// Extract JSON points and the next-page cursor from a typed scroll response.
///
/// [`ExecData::Hits`](qql::executor::ExecData::Hits) does not carry Qdrant's
/// `next_page_offset`, so the cursor always comes from the last point id;
/// [`ScrollPages`](super::ScrollPages) then removes the inclusive-offset repeat.
pub fn extract_scroll_page(response: &BackendResponse) -> (Vec<Value>, Option<PlanPointId>) {
    let points = response
        .data
        .hits()
        .map(|hits| {
            hits.iter()
                .filter_map(|hit| serde_json::to_value(hit).ok())
                .collect()
        })
        .unwrap_or_default();
    (points, None)
}

/// Convert a JSON number or string point identifier to typed `PlanPointId`.
pub fn json_to_plan_point_id(v: &Value) -> Option<PlanPointId> {
    match v {
        Value::Number(n) => n.as_u64().map(PlanPointId::Number),
        Value::String(s) => Some(PlanPointId::String(s.clone())),
        _ => None,
    }
}