quelch 0.10.1

Ingest data from Jira, Confluence, and more directly into Azure AI Search
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

use serde_json::Value;

use crate::cosmos::CosmosError;

/// A thin envelope helper for Cosmos DB documents.
///
/// All documents stored via `CosmosBackend` must carry:
/// - `id`            — unique within the container+partition
/// - `_partition_key` — logical partition; always equal to `deployment_name`
///
/// This module provides utility functions for extracting those fields from a
/// `serde_json::Value` without fully deserialising the document.
pub struct CosmosDocument;

impl CosmosDocument {
    /// Extract the `id` string field from a JSON document value.
    ///
    /// Returns `CosmosError::Validation` if the field is missing or not a string.
    pub fn extract_id(doc: &Value) -> Result<&str, CosmosError> {
        doc.get("id")
            .and_then(|v| v.as_str())
            .ok_or_else(|| CosmosError::Validation("document missing string `id` field".into()))
    }

    /// Extract the `_partition_key` string field from a JSON document value.
    ///
    /// Returns `CosmosError::Validation` if the field is missing or not a string.
    pub fn extract_partition_key(doc: &Value) -> Result<&str, CosmosError> {
        doc.get("_partition_key")
            .and_then(|v| v.as_str())
            .ok_or_else(|| {
                CosmosError::Validation("document missing string `_partition_key` field".into())
            })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use serde_json::json;

    #[test]
    fn extract_id_and_partition_key_happy_path() {
        let doc = json!({
            "id": "abc::def::ghi",
            "_partition_key": "abc",
            "some_field": 42
        });
        assert_eq!(CosmosDocument::extract_id(&doc).unwrap(), "abc::def::ghi");
        assert_eq!(CosmosDocument::extract_partition_key(&doc).unwrap(), "abc");
    }

    #[test]
    fn extract_id_missing_returns_validation_error() {
        let doc = json!({ "_partition_key": "abc" });
        assert!(matches!(
            CosmosDocument::extract_id(&doc),
            Err(CosmosError::Validation(_))
        ));
    }

    #[test]
    fn extract_partition_key_missing_returns_validation_error() {
        let doc = json!({ "id": "abc" });
        assert!(matches!(
            CosmosDocument::extract_partition_key(&doc),
            Err(CosmosError::Validation(_))
        ));
    }
}