starweaver-session 0.10.0

Durable session contracts for Starweaver
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150

use chrono::Utc;
use starweaver_context::ResumableState;
use starweaver_core::SessionId;

use crate::{
    error::{SessionStoreError, SessionStoreResult},
    records::{EnvironmentStateRef, SessionRecord, SessionStatus},
    store::{SessionFilter, SessionPage, SessionPageKey, SessionPageQuery},
};

use super::{InMemorySessionStore, store_failed};

impl InMemorySessionStore {
    pub(super) fn save_session_record(&self, mut session: SessionRecord) -> SessionStoreResult<()> {
        let mut inner = self.inner.lock().map_err(store_failed)?;
        if let Some(current) = inner.sessions.get(&session.session_id) {
            session.revision = current.revision.saturating_add(1);
            session.updated_at = Utc::now();
        }
        inner.sessions.insert(session.session_id.clone(), session);
        Ok(())
    }

    pub(super) fn load_session_record(
        &self,
        session_id: &SessionId,
    ) -> SessionStoreResult<SessionRecord> {
        let inner = self.inner.lock().map_err(store_failed)?;
        inner
            .sessions
            .get(session_id)
            .cloned()
            .ok_or_else(|| SessionStoreError::NotFound(session_id.as_str().to_string()))
    }

    #[allow(clippy::needless_pass_by_value)]
    pub(super) fn list_session_records(
        &self,
        filter: SessionFilter,
    ) -> SessionStoreResult<Vec<SessionRecord>> {
        let inner = self.inner.lock().map_err(store_failed)?;
        let mut sessions = inner
            .sessions
            .values()
            .filter(|session| filter.status.is_none_or(|status| session.status == status))
            .filter(|session| {
                filter
                    .profile
                    .as_ref()
                    .is_none_or(|profile| session.profile.as_ref() == Some(profile))
            })
            .filter(|session| {
                filter
                    .workspace
                    .as_ref()
                    .is_none_or(|workspace| session.workspace.as_ref() == Some(workspace))
            })
            .cloned()
            .collect::<Vec<_>>();
        sessions.sort_by_key(|session| std::cmp::Reverse(session.updated_at));
        if let Some(limit) = filter.limit {
            sessions.truncate(limit);
        }
        Ok(sessions)
    }

    pub(super) fn list_session_record_page(
        &self,
        query: &SessionPageQuery,
    ) -> SessionStoreResult<SessionPage> {
        let inner = self.inner.lock().map_err(store_failed)?;
        let mut sessions = inner
            .sessions
            .values()
            .filter(|session| {
                query.after().is_none_or(|after| {
                    session.updated_at < after.updated_at
                        || (session.updated_at == after.updated_at
                            && session.session_id < after.session_id)
                })
            })
            .cloned()
            .collect::<Vec<_>>();
        sessions.sort_by(|left, right| {
            right
                .updated_at
                .cmp(&left.updated_at)
                .then_with(|| right.session_id.cmp(&left.session_id))
        });
        let has_more = sessions.len() > query.limit();
        sessions.truncate(query.limit());
        let next_key = sessions
            .last()
            .map(SessionPageKey::from_session)
            .or_else(|| query.after().cloned());
        Ok(SessionPage {
            sessions,
            next_key,
            has_more,
        })
    }

    pub(super) fn set_session_status(
        &self,
        session_id: &SessionId,
        status: SessionStatus,
    ) -> SessionStoreResult<()> {
        let mut inner = self.inner.lock().map_err(store_failed)?;
        let session = inner
            .sessions
            .get_mut(session_id)
            .ok_or_else(|| SessionStoreError::NotFound(session_id.as_str().to_string()))?;
        session.status = status;
        session.revision = session.revision.saturating_add(1);
        session.updated_at = Utc::now();
        Ok(())
    }

    pub(super) fn save_context_state_snapshot(
        &self,
        session_id: &SessionId,
        state: ResumableState,
    ) -> SessionStoreResult<()> {
        let mut inner = self.inner.lock().map_err(store_failed)?;
        let session = inner
            .sessions
            .get_mut(session_id)
            .ok_or_else(|| SessionStoreError::NotFound(session_id.as_str().to_string()))?;
        session.state = state;
        session.revision = session.revision.saturating_add(1);
        session.updated_at = Utc::now();
        Ok(())
    }

    pub(super) fn save_environment_state_ref(
        &self,
        session_id: &SessionId,
        environment_state: EnvironmentStateRef,
    ) -> SessionStoreResult<()> {
        let mut inner = self.inner.lock().map_err(store_failed)?;
        let session = inner
            .sessions
            .get_mut(session_id)
            .ok_or_else(|| SessionStoreError::NotFound(session_id.as_str().to_string()))?;
        session.environment_state = Some(environment_state);
        session.revision = session.revision.saturating_add(1);
        session.updated_at = Utc::now();
        Ok(())
    }
}