raphtory 0.17.0

raphtory, a temporal graph library
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

use crate::core::state::{compute_state::ComputeState, shuffle_state::ShuffleComputeState};
use std::{borrow::Cow, cell::RefCell, rc::Rc};

#[derive(Debug)]
pub(crate) struct EVState<'a, CS: Clone> {
    pub(crate) shard_state: Cow<'a, ShuffleComputeState<CS>>,
    pub(crate) global_state: Cow<'a, ShuffleComputeState<CS>>,
}

impl<'a, CS: ComputeState> EVState<'a, CS> {
    pub fn rc_from(
        shard_state: Cow<'a, ShuffleComputeState<CS>>,
        global_state: Cow<'a, ShuffleComputeState<CS>>,
    ) -> Rc<RefCell<Self>> {
        Rc::new(RefCell::new(Self {
            shard_state,
            global_state,
        }))
    }

    pub fn restore_states(
        self,
    ) -> (
        Cow<'a, ShuffleComputeState<CS>>,
        Cow<'a, ShuffleComputeState<CS>>,
    ) {
        (self.shard_state, self.global_state)
    }

    pub(crate) fn shard_mut(&mut self) -> &mut ShuffleComputeState<CS> {
        self.shard_state.to_mut()
    }

    pub(crate) fn global_mut(&mut self) -> &mut ShuffleComputeState<CS> {
        self.global_state.to_mut()
    }

    pub(crate) fn shard(&self) -> &ShuffleComputeState<CS> {
        &self.shard_state
    }

    pub(crate) fn global(&self) -> &ShuffleComputeState<CS> {
        &self.global_state
    }
}