Function build_context

benches/queries.rs:100–147

(cg: &TraceDecay)

}

pub async fn build_context(cg: &TraceDecay) -> QueryContext {
    let all = cg.get_all_nodes().await.unwrap_or_default();

    let mut function_ids = Vec::new();
    let mut struct_ids = Vec::new();
    let mut function_qnames = Vec::new();
    let mut any_ids = Vec::new();

    for n in &all {
        any_ids.push(n.id.clone());
        match n.kind {
            NodeKind::Function | NodeKind::Method => {
                function_ids.push(n.id.clone());
                if !n.qualified_name.is_empty() && function_qnames.len() < 64 {
                    function_qnames.push(n.qualified_name.clone());
                }
            }
            NodeKind::Struct | NodeKind::Class => {
                struct_ids.push(n.id.clone());
            }
            _ => {}
        }
    }

    function_ids.truncate(64);
    struct_ids.truncate(64);
    any_ids.truncate(64);

    let files = cg.get_all_files().await.unwrap_or_default();

    // Collect first-segment directory prefixes from the sample files (so
    // `path_prefix` queries are valid for *this* repo regardless of layout).
    let mut dir_prefixes: Vec<String> = files
        .iter()
        .filter_map(|f| f.path.split('/').next().map(str::to_string))
        .collect();
    dir_prefixes.sort();
    dir_prefixes.dedup();
    dir_prefixes.truncate(5);

    QueryContext {
        function_ids,
        struct_ids,
        any_ids,
        function_qnames,
        dir_prefixes,
    }
}

fn five<F: FnMut(usize) -> Query>(mut f: F) -> Vec<Query> {
    (0..5).map(&mut f).collect()