rustberg 0.0.3

A production-grade, cross-platform, single-binary Apache Iceberg REST Catalog
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
151
152
153
154
155
156
157
158

use axum::{
    extract::{Query, State},
    response::Json as AxumJson,
};
use serde::{Deserialize, Serialize};

use crate::app::AppState;

#[derive(Debug, Deserialize)]
pub struct ConfigQuery {
    pub warehouse: Option<String>,
}

#[derive(Debug, Serialize)]
pub struct Overrides {
    pub warehouse: String,
}

#[derive(Debug, Serialize)]
pub struct Defaults {
    pub clients: String,
}

#[derive(Debug, Serialize)]
pub struct ConfigResponse {
    pub overrides: Overrides,
    pub defaults: Defaults,
    pub endpoints: Vec<String>,
}

pub async fn get_config(
    State(state): State<AppState>,
    Query(query): Query<ConfigQuery>,
) -> AxumJson<ConfigResponse> {
    // Check if the `warehouse` query parameter is provided
    let warehouse_uri = query
        .warehouse
        .or(Some(state.warehouse_location)) // Use the provided warehouse or fallback to catalog's warehouse
        .unwrap_or_else(|| "file:///default-warehouse/".to_string()); // Default value

    AxumJson(ConfigResponse {
        overrides: Overrides {
            warehouse: warehouse_uri,
        },
        defaults: Defaults {
            clients: "4".to_string(),
        },
        endpoints: vec![
            "GET /v1/namespaces".to_string(),
            "POST /v1/namespaces".to_string(),
            "GET /v1/namespaces/{namespace}".to_string(),
            "HEAD /v1/namespaces/{namespace}".to_string(),
            "GET /v1/namespaces/{namespace}/tables/{table}".to_string(),
            "GET /v1/namespaces/{namespace}/views/{view}".to_string(),
        ],
    })
}

// ============================================================================
// Unit Tests
// ============================================================================

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

    // ========================================================================
    // ConfigQuery Tests
    // ========================================================================

    #[test]
    fn test_config_query_empty() {
        let json = "{}";
        let query: ConfigQuery = serde_json::from_str(json).unwrap();
        assert!(query.warehouse.is_none());
    }

    #[test]
    fn test_config_query_with_warehouse() {
        let json = r#"{"warehouse": "s3://my-bucket/warehouse"}"#;
        let query: ConfigQuery = serde_json::from_str(json).unwrap();
        assert_eq!(
            query.warehouse,
            Some("s3://my-bucket/warehouse".to_string())
        );
    }

    // ========================================================================
    // Overrides Tests
    // ========================================================================

    #[test]
    fn test_overrides_serialization() {
        let overrides = Overrides {
            warehouse: "s3://bucket/path".to_string(),
        };
        let json = serde_json::to_value(&overrides).unwrap();
        assert_eq!(json["warehouse"], "s3://bucket/path");
    }

    // ========================================================================
    // Defaults Tests
    // ========================================================================

    #[test]
    fn test_defaults_serialization() {
        let defaults = Defaults {
            clients: "8".to_string(),
        };
        let json = serde_json::to_value(&defaults).unwrap();
        assert_eq!(json["clients"], "8");
    }

    // ========================================================================
    // ConfigResponse Tests
    // ========================================================================

    #[test]
    fn test_config_response_serialization() {
        let response = ConfigResponse {
            overrides: Overrides {
                warehouse: "/tmp/warehouse".to_string(),
            },
            defaults: Defaults {
                clients: "4".to_string(),
            },
            endpoints: vec![
                "GET /v1/namespaces".to_string(),
                "POST /v1/namespaces".to_string(),
            ],
        };
        let json = serde_json::to_value(&response).unwrap();
        assert_eq!(json["overrides"]["warehouse"], "/tmp/warehouse");
        assert_eq!(json["defaults"]["clients"], "4");
        assert_eq!(json["endpoints"].as_array().unwrap().len(), 2);
    }

    #[test]
    fn test_config_response_endpoints_format() {
        let response = ConfigResponse {
            overrides: Overrides {
                warehouse: "test".to_string(),
            },
            defaults: Defaults {
                clients: "4".to_string(),
            },
            endpoints: vec![
                "GET /v1/namespaces".to_string(),
                "HEAD /v1/namespaces/{namespace}".to_string(),
            ],
        };
        let json = serde_json::to_value(&response).unwrap();
        let endpoints = json["endpoints"].as_array().unwrap();
        // Verify endpoints have correct format (METHOD path)
        assert!(endpoints[0].as_str().unwrap().starts_with("GET "));
        assert!(endpoints[1].as_str().unwrap().starts_with("HEAD "));
    }
}