Function partition_put_cells

nodedb-cluster/src/distributed_array/partition.rs:60–87

Group `(hilbert_prefix, cell_msgpack)` pairs by tile bucket. `cells` — each element is `(hilbert_prefix, zerompk-encoded single cell)`. `prefix_bits` — routing granularity (1–16). Returns one `PutBucket` per unique owning vShard. The order of buckets in the output is unspecified but deterministic within a call.

(cells: &[(u64, Vec<u8>)], prefix_bits: u8)

/// Returns one `PutBucket` per unique owning vShard. The order of buckets
/// in the output is unspecified but deterministic within a call.
pub fn partition_put_cells(cells: &[(u64, Vec<u8>)], prefix_bits: u8) -> Result<Vec<PutBucket>> {
    if cells.is_empty() {
        return Ok(Vec::new());
    }

    // vshard_id → (representative_hilbert_prefix, per-cell byte blobs)
    let mut buckets: HashMap<u32, (u64, Vec<Vec<u8>>)> = HashMap::new();

    for (hilbert_prefix, cell_bytes) in cells {
        let vshard_id = array_vshard_for_tile(*hilbert_prefix, prefix_bits)?;
        let entry = buckets
            .entry(vshard_id)
            .or_insert((*hilbert_prefix, Vec::new()));
        entry.1.push(cell_bytes.clone());
    }

    buckets
        .into_iter()
        .map(|(vshard_id, (representative, cell_blobs))| {
            let cells_msgpack = encode_blob_vec(&cell_blobs)?;
            Ok(PutBucket {
                vshard_id,
                representative_hilbert_prefix: representative,
                cells_msgpack,
            })
        })
        .collect()
}

/// Group `(hilbert_prefix, coord_msgpack)` pairs by tile bucket.
///