notedthat_api_http/
state.rs

1//! Shared application state for the axum router.
2
3use notedthat_core::{KbSlug, Storage};
4use notedthat_indexer::Searcher;
5use std::collections::BTreeMap;
6use std::sync::Arc;
7
8/// Application state shared across all axum handlers.
9///
10/// This is cloned cheaply for each request (all fields are behind [`Arc`]).
11#[derive(Clone)]
12pub struct AppState {
13    /// The backing storage implementation (injected at startup).
14    pub storage: Arc<dyn Storage>,
15    /// Canonical map of slug string → [`KbSlug`] for declared knowledge bases.
16    pub declared_kbs: Arc<BTreeMap<String, KbSlug>>,
17    /// Static Bearer token for authenticating API requests.
18    pub bearer_token: Arc<String>,
19    /// Maximum accepted PUT body size in bytes (16 MiB in M2).
20    pub max_body_size: u64,
21    /// Maximum object size eligible for patch operations, in bytes.
22    pub max_patchable_size: u64,
23    /// Sender half of the async indexing queue.
24    pub indexer_tx: tokio::sync::mpsc::Sender<notedthat_indexer::IndexEvent>,
25    /// The search implementation (injected at startup).
26    pub searcher: Arc<dyn Searcher>,
27}
28
29#[cfg(test)]
30mod tests {
31    use super::*;
32    use crate::testing::InMemoryStorage;
33    use std::collections::BTreeMap;
34    use std::sync::Arc;
35
36    fn minimal_state(tx: tokio::sync::mpsc::Sender<notedthat_indexer::IndexEvent>) -> AppState {
37        AppState {
38            storage: Arc::new(InMemoryStorage::default()),
39            declared_kbs: Arc::new(BTreeMap::new()),
40            bearer_token: Arc::new("token".to_string()),
41            max_body_size: 1024,
42            max_patchable_size: 1024,
43            indexer_tx: tx,
44            searcher: Arc::new(crate::testing::NoopSearcher),
45        }
46    }
47
48    #[tokio::test]
49    async fn clone_shares_same_channel() {
50        let (tx, mut rx) = tokio::sync::mpsc::channel(10);
51        let state = minimal_state(tx);
52        let cloned = state.clone();
53
54        let event = notedthat_indexer::IndexEvent::Tombstone {
55            kb: notedthat_core::KbSlug::try_new("test").expect("valid kb slug"),
56            object_key: notedthat_core::ObjectPath::try_from("a.md").expect("valid path"),
57        };
58        cloned
59            .indexer_tx
60            .send(event.clone())
61            .await
62            .expect("send on cloned tx");
63
64        let received = rx.recv().await.expect("receive on original rx");
65        assert_eq!(
66            received, event,
67            "cloned Sender must share the same underlying channel"
68        );
69    }
70}
notedthat_api_http/state.rs

notedthat_api_http/
state.rs
