sz_orm_storage/
aliyun.rs

1//! # Aliyun OSS Storage（**MOCK-ONLY，非生产可用**）
2//!
3//! ⚠️ **重要警告：本模块为内存 Mock 实现，未集成真实阿里云 OSS SDK。**
4//!
5//! - 所有数据存储在进程内 `HashMap`，重启即丢失
6//! - 不执行任何 HTTP 请求，不与真实 OSS 服务交互
7//! - 不支持认证、签名、TLS、分片上传等任何 OSS 生产特性
8//! - **请勿用于生产环境**——仅适用于单元测试与本地开发
9//!
10//! 如需真实 OSS 集成，请基于 [`crate::storage::Storage`] trait
11//! 接入阿里云官方 SDK（如 `aliyun-oss` crate）实现。
12
13use crate::error::StorageError;
14use crate::storage::Storage;
15use async_trait::async_trait;
16use std::collections::HashMap;
17use std::sync::Arc;
18use tokio::sync::RwLock;
19
20/// 阿里云 OSS 存储后端（**Mock 实现**）
21///
22/// ⚠️ 仅用于测试。所有数据存储在内存 `HashMap`，不与真实 OSS 服务交互。
23/// 如需生产使用，请实现 `Storage` trait 接入官方 SDK。
24pub struct AliyunOssStorage {
25    pub bucket: String,
26    pub endpoint: String,
27    store: Arc<RwLock<HashMap<String, Vec<u8>>>>,
28}
29
30impl AliyunOssStorage {
31    pub fn new(bucket: impl Into<String>, endpoint: impl Into<String>) -> Self {
32        Self {
33            bucket: bucket.into(),
34            endpoint: endpoint.into(),
35            store: Arc::new(RwLock::new(HashMap::new())),
36        }
37    }
38
39    pub fn url_for(&self, key: &str) -> String {
40        format!("oss://{}.{}/{}", self.bucket, self.endpoint, key)
41    }
42}
43
44#[async_trait]
45impl Storage for AliyunOssStorage {
46    async fn put(
47        &self,
48        key: &str,
49        data: &[u8],
50        _content_type: &str,
51    ) -> Result<String, StorageError> {
52        let mut store = self.store.write().await;
53        store.insert(key.to_string(), data.to_vec());
54        Ok(self.url_for(key))
55    }
56
57    async fn get(&self, key: &str) -> Result<Vec<u8>, StorageError> {
58        let store = self.store.read().await;
59        store
60            .get(key)
61            .cloned()
62            .ok_or_else(|| StorageError::NotFound(format!("oss://{}/{}", self.bucket, key)))
63    }
64
65    async fn delete(&self, key: &str) -> Result<(), StorageError> {
66        let mut store = self.store.write().await;
67        store.remove(key);
68        Ok(())
69    }
70
71    async fn exists(&self, key: &str) -> Result<bool, StorageError> {
72        let store = self.store.read().await;
73        Ok(store.contains_key(key))
74    }
75}
76
77#[cfg(test)]
78mod tests {
79    use super::*;
80
81    #[tokio::test]
82    async fn test_aliyun_put_and_get() {
83        let storage = AliyunOssStorage::new("bucket", "oss-cn-hangzhou.aliyuncs.com");
84        let url = storage
85            .put("a.txt", b"aliyun-data", "text/plain")
86            .await
87            .unwrap();
88        assert!(url.starts_with("oss://bucket.oss-cn-hangzhou.aliyuncs.com/"));
89
90        let data = storage.get("a.txt").await.unwrap();
91        assert_eq!(data, b"aliyun-data");
92    }
93
94    #[tokio::test]
95    async fn test_aliyun_get_not_found() {
96        let storage = AliyunOssStorage::new("bucket", "oss-cn-hangzhou.aliyuncs.com");
97        let result = storage.get("missing").await;
98        assert!(result.is_err());
99        assert!(matches!(result.unwrap_err(), StorageError::NotFound(_)));
100    }
101
102    #[tokio::test]
103    async fn test_aliyun_delete_and_exists() {
104        let storage = AliyunOssStorage::new("bucket", "oss-cn-hangzhou.aliyuncs.com");
105        storage.put("key", b"data", "text/plain").await.unwrap();
106        assert!(storage.exists("key").await.unwrap());
107
108        storage.delete("key").await.unwrap();
109        assert!(!storage.exists("key").await.unwrap());
110    }
111
112    #[tokio::test]
113    async fn test_aliyun_overwrite() {
114        let storage = AliyunOssStorage::new("bucket", "oss-cn-hangzhou.aliyuncs.com");
115        storage.put("key", b"v1", "text/plain").await.unwrap();
116        storage.put("key", b"v2", "text/plain").await.unwrap();
117        assert_eq!(storage.get("key").await.unwrap(), b"v2");
118    }
119}
sz_orm_storage/aliyun.rs

sz_orm_storage/
aliyun.rs
