Function create_node_edges

src/vector_store.rs:976–1109

(
    hnsw_index: &HNSWIndex,
    lazy_item_latest_ptr: SharedLatestNode,
    neighbors: Vec<(SharedLatestNode, MetricResult)>,
    version: VersionNumber,
    file_id: IndexFileId,
    max_edges: usi

#[allow(clippy::too_many_arguments)]
fn create_node_edges(
    hnsw_index: &HNSWIndex,
    lazy_item_latest_ptr: SharedLatestNode,
    neighbors: Vec<(SharedLatestNode, MetricResult)>,
    version: VersionNumber,
    file_id: IndexFileId,
    max_edges: usize,
    offset_counter: &HNSWIndexFileOffsetCounter,
    distance_metric: DistanceMetric,
) -> Result<(), WaCustomError> {
    let mut successful_edges = 0;
    let mut neighbors_to_update = Vec::new();

    let lazy_item = unsafe { &*lazy_item_latest_ptr }.latest;
    let node = unsafe { &*lazy_item }.try_get_data(&hnsw_index.cache)?;

    let node_id = node.get_id();

    // First loop: Handle neighbor connections and collect updates
    for (neighbor_lazy_item_latest_ptr, dist) in neighbors {
        if successful_edges >= max_edges {
            break;
        }

        let (neighbor_lazy_item, newly_created) = LatestNode::get_or_create_version(
            neighbor_lazy_item_latest_ptr,
            version,
            &hnsw_index.versions_synchronization_map,
            &hnsw_index.cache,
            file_id,
            offset_counter,
        )?;

        let neighbor_node = unsafe { &*neighbor_lazy_item }.try_get_data(&hnsw_index.cache)?;
        let neighbor_node_id = neighbor_node.get_id();

        assert_eq!(neighbor_node.version.load(Ordering::Relaxed), *version);

        // Ensure that a metadata node gets connected to a pseudo node
        // only if there's a perfect match
        match (
            neighbor_node.replica_node_kind(),
            node.replica_node_kind(),
            &dist,
        ) {
            (
                ReplicaNodeKind::Pseudo,
                ReplicaNodeKind::Metadata,
                MetricResult::CosineSimilarity(cs),
            ) => {
                if cs.0 != 1.0 {
                    continue;
                }
            }
            (
                ReplicaNodeKind::Metadata,
                ReplicaNodeKind::Metadata,
                MetricResult::CosineSimilarity(cs),