tfhe 1.6.1

TFHE-rs is a fully homomorphic encryption (FHE) library that implements Zama's variant of TFHE.
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
35
36
37
38
39
40
41
42
43
44
45
46
47

use std::collections::HashMap;
use std::hash::Hash;
use std::sync::{Arc, RwLock};

pub struct GenericPlanMap<Key, Value>(pub RwLock<HashMap<Key, Arc<Value>>>);

impl<Key: Eq + Hash + Copy, Value> GenericPlanMap<Key, Value> {
    pub fn new() -> Self {
        Self(RwLock::new(HashMap::new()))
    }

    pub fn set(&self, key: Key, new_value: Arc<Value>) {
        let mut values = self.0.write().unwrap();

        let _old_value = values.insert(key, new_value);

        drop(values);
    }

    pub fn get_or_init(&self, key: Key, new_value: impl Fn(Key) -> Value) -> Arc<Value> {
        let values = self.0.read().unwrap();

        let value = values.get(&key).cloned();
        drop(values);

        value.unwrap_or_else(|| {
            // If we don't find a plan for the given polynomial size and modulus, we insert a
            // new one (if we still don't find it after getting a write lock)

            let new_value = Arc::new(new_value(key));

            let mut values = self.0.write().unwrap();

            let value = Arc::clone(values.entry(key).or_insert(new_value));

            drop(values);

            value
        })
    }
}

impl<Key: Eq + Hash + Copy, Value> Default for GenericPlanMap<Key, Value> {
    fn default() -> Self {
        Self::new()
    }
}