codanna 0.9.19

Code Intelligence for Large Language Models
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

use parking_lot::RwLock;
use std::collections::HashMap;
use std::sync::Arc;

use super::Sha256Hash;

/// Simple thread-safe in-memory memoization map keyed by `Sha256Hash`.
/// Value type must be `Clone` to allow cheap reads without exposing interior mutability.
pub struct ResolutionMemo<V> {
    inner: RwLock<HashMap<Sha256Hash, Arc<V>>>,
}

impl<V> Default for ResolutionMemo<V> {
    fn default() -> Self {
        Self::new()
    }
}

impl<V> ResolutionMemo<V> {
    pub fn new() -> Self {
        Self {
            inner: RwLock::new(HashMap::new()),
        }
    }

    pub fn insert(&self, key: Sha256Hash, value: V) {
        let mut map = self.inner.write();
        map.insert(key, Arc::new(value));
    }

    pub fn get(&self, key: &Sha256Hash) -> Option<Arc<V>> {
        let map = self.inner.read();
        map.get(key).cloned()
    }

    pub fn clear(&self) {
        let mut map = self.inner.write();
        map.clear();
    }
}