commonware-resolver 2026.4.0

Resolve data identified by a fixed-length key.
Documentation 
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
33
34

use crate::Span;
use bytes::Bytes;
use commonware_utils::channel::{fallible::OneshotExt, oneshot};
use std::collections::HashMap;

/// A producer that can be used for testing
#[derive(Clone, Default)]
pub struct Producer<K: Span, V> {
    /// The data that the producer produces upon request
    data: HashMap<K, V>,
}

impl<K: Span, V> Producer<K, V> {
    /// Adds a key-value pair to the producer
    pub fn insert(&mut self, key: K, value: V) {
        self.data.insert(key, value);
    }
}

impl<K: Span, V: Into<Bytes> + Clone + Send + 'static> crate::p2p::Producer for Producer<K, V> {
    type Key = K;

    /// Produce a value for the given key.
    ///
    /// If the key is not found, the returned receiver will resolve with an error since the sender
    /// is dropped.
    async fn produce(&mut self, key: Self::Key) -> oneshot::Receiver<Bytes> {
        let (sender, receiver) = oneshot::channel();
        if let Some(value) = self.data.get(&key) {
            sender.send_lossy(value.clone().into());
        }
        receiver
    }
}