sz_orm_storage/
huawei.rs

1use crate::error::StorageError;
2use crate::storage::Storage;
3use async_trait::async_trait;
4use std::collections::HashMap;
5use std::sync::Arc;
6use tokio::sync::RwLock;
7
8pub struct HuaweiObsStorage {
9    pub bucket: String,
10    pub endpoint: String,
11    store: Arc<RwLock<HashMap<String, Vec<u8>>>>,
12}
13
14impl HuaweiObsStorage {
15    pub fn new(bucket: impl Into<String>, endpoint: impl Into<String>) -> Self {
16        Self {
17            bucket: bucket.into(),
18            endpoint: endpoint.into(),
19            store: Arc::new(RwLock::new(HashMap::new())),
20        }
21    }
22
23    pub fn url_for(&self, key: &str) -> String {
24        format!("obs://{}.{}/{}", self.bucket, self.endpoint, key)
25    }
26}
27
28#[async_trait]
29impl Storage for HuaweiObsStorage {
30    async fn put(
31        &self,
32        key: &str,
33        data: &[u8],
34        _content_type: &str,
35    ) -> Result<String, StorageError> {
36        let mut store = self.store.write().await;
37        store.insert(key.to_string(), data.to_vec());
38        Ok(self.url_for(key))
39    }
40
41    async fn get(&self, key: &str) -> Result<Vec<u8>, StorageError> {
42        let store = self.store.read().await;
43        store
44            .get(key)
45            .cloned()
46            .ok_or_else(|| StorageError::NotFound(format!("obs://{}/{}", self.bucket, key)))
47    }
48
49    async fn delete(&self, key: &str) -> Result<(), StorageError> {
50        let mut store = self.store.write().await;
51        store.remove(key);
52        Ok(())
53    }
54
55    async fn exists(&self, key: &str) -> Result<bool, StorageError> {
56        let store = self.store.read().await;
57        Ok(store.contains_key(key))
58    }
59}
60
61#[cfg(test)]
62mod tests {
63    use super::*;
64
65    #[tokio::test]
66    async fn test_huawei_put_and_get() {
67        let storage = HuaweiObsStorage::new("my-bucket", "obs.cn-north-4.myhuaweicloud.com");
68        let url = storage
69            .put("a.txt", b"huawei-data", "text/plain")
70            .await
71            .unwrap();
72        assert!(url.starts_with("obs://my-bucket.obs.cn-north-4.myhuaweicloud.com/"));
73
74        let data = storage.get("a.txt").await.unwrap();
75        assert_eq!(data, b"huawei-data");
76    }
77
78    #[tokio::test]
79    async fn test_huawei_get_not_found() {
80        let storage = HuaweiObsStorage::new("bucket", "obs.cn-north-4.myhuaweicloud.com");
81        let result = storage.get("missing").await;
82        assert!(result.is_err());
83        assert!(matches!(result.unwrap_err(), StorageError::NotFound(_)));
84    }
85
86    #[tokio::test]
87    async fn test_huawei_delete_and_exists() {
88        let storage = HuaweiObsStorage::new("bucket", "obs.cn-north-4.myhuaweicloud.com");
89        storage.put("key", b"data", "text/plain").await.unwrap();
90        assert!(storage.exists("key").await.unwrap());
91
92        storage.delete("key").await.unwrap();
93        assert!(!storage.exists("key").await.unwrap());
94    }
95
96    #[tokio::test]
97    async fn test_huawei_overwrite() {
98        let storage = HuaweiObsStorage::new("bucket", "obs.cn-north-4.myhuaweicloud.com");
99        storage.put("key", b"v1", "text/plain").await.unwrap();
100        storage.put("key", b"v2", "text/plain").await.unwrap();
101        assert_eq!(storage.get("key").await.unwrap(), b"v2");
102    }
103
104    #[tokio::test]
105    async fn test_huawei_multiple_keys() {
106        let storage = HuaweiObsStorage::new("bucket", "obs.cn-north-4.myhuaweicloud.com");
107        storage.put("a", b"a-data", "text/plain").await.unwrap();
108        storage.put("b", b"b-data", "text/plain").await.unwrap();
109
110        assert_eq!(storage.get("a").await.unwrap(), b"a-data");
111        assert_eq!(storage.get("b").await.unwrap(), b"b-data");
112
113        storage.delete("a").await.unwrap();
114        assert!(!storage.exists("a").await.unwrap());
115        assert!(storage.exists("b").await.unwrap());
116    }
117}
sz_orm_storage/huawei.rs

sz_orm_storage/
huawei.rs
