obs_sdk/
lib.rs

1//! # 使用说明
2//! 
3//! ## 1. 列举桶内对象列表
4//! 
5//! ```rust
6//! use obs_sdk::{ObsClient, ObsError};
7//! 
8//! static ENDPOINT: &str = "obs.cn-north-4.myhuaweicloud.com";
9//! static AK: &str = "YOUR_AK";
10//! static SK: &str = "YOUR_SK";
11//! static BUCKET_NAME: &str = "bucket_name";
12//! 
13//! #[tokio::test]
14//! async fn test_list_prefix() -> Result<(), ObsError> {
15//!     let client = ObsClient {
16//!         endpoint: ENDPOINT.to_string(),
17//!         ak: AK.to_string(),
18//!         sk: SK.to_string(),
19//!         bucket: BUCKET_NAME.to_string(),
20//!     };
21//!     let res = client.list("tmp", false).await?;
22//!     println!("{:?}", res);
23//!     Ok(())
24//! }
25//! ```
26//! 
27//! ## 2. 上传对象到桶
28//! 
29//! ```rust
30//! use obs_sdk::{ObsClient, ObsError};
31//! 
32//! static ENDPOINT: &str = "obs.cn-north-4.myhuaweicloud.com";
33//! static AK: &str = "YOUR_AK";
34//! static SK: &str = "YOUR_SK";
35//! static BUCKET_NAME: &str = "bucket_name";
36//! 
37//! #[tokio::test]
38//! async fn test_upload_object() -> Result<(), ObsError> {
39//!     let client = ObsClient {
40//!         endpoint: ENDPOINT.to_string(),
41//!         ak: AK.to_string(),
42//!         sk: SK.to_string(),
43//!         bucket: BUCKET_NAME.to_string(),
44//!     };
45//!     let res = client.upload_file("tmp_cargo.txt", "Cargo.txt").await?;
46//!     println!("{:?}", res);
47//!     Ok(())
48//! }
49//! ```
50//! 
51//! ## 3. 下载对象到本地目录
52//! 
53//! ```rust
54//! use obs_sdk::{ObsClient, ObsError};
55//! 
56//! static ENDPOINT: &str = "obs.cn-north-4.myhuaweicloud.com";
57//! static AK: &str = "YOUR_AK";
58//! static SK: &str = "YOUR_SK";
59//! static BUCKET_NAME: &str = "bucket_name";
60//! 
61//! #[tokio::test]
62//! async fn test_download_file02() -> Result<(), ObsError> {
63//!     let client = ObsClient {
64//!         endpoint: ENDPOINT.to_string(),
65//!         ak: AK.to_string(),
66//!         sk: SK.to_string(),
67//!         bucket: BUCKET_NAME.to_string(),
68//!     };
69//!     let res = client.download_file("2hls_stutter-10.mp4", "video/2hls_stutter-10.mp4", false).await;
70//!     res
71//! }
72//! ```
73//! 
74//! ## 4. 下载对象为字节内容
75//! 
76//! ```rust
77//! use obs_sdk::{ObsClient, ObsError};
78//! 
79//! static ENDPOINT: &str = "obs.cn-north-4.myhuaweicloud.com";
80//! static AK: &str = "YOUR_AK";
81//! static SK: &str = "YOUR_SK";
82//! static BUCKET_NAME: &str = "bucket_name";
83//! 
84//! #[tokio::test]
85//! async fn test_download_file01() -> Result<(), ObsError> {
86//!     let client = ObsClient {
87//!         endpoint: ENDPOINT.to_string(),
88//!         ak: AK.to_string(),
89//!         sk: SK.to_string(),
90//!         bucket: BUCKET_NAME.to_string(),
91//!     };
92//!     let data = client.download_object("2hls_stutter-10.mp4").await?;
93//!     let file_path = Path::new("output.mp4");
94//!     match fs::write(file_path, data) {
95//!         Ok(_) => println!("文件保存成功{:?}", file_path),
96//!         Err(e) => eprintln!("文件保存失败:{}", e)
97//!     }
98//!     Ok(())
99//! }
100//! ```
101//! 
102//! ## 5. url鉴权
103//! 
104//! ```rust
105//! use obs_sdk::{ObsClient, ObsError};
106//!
107//! static ENDPOINT: &str = "obs.cn-north-4.myhuaweicloud.com";
108//! static AK: &str = "YOUR_AK";
109//! static SK: &str = "YOUR_SK";
110//! static BUCKET_NAME: &str = "bucket_name";
111//! 
112//! #[test]
113//! fn test_url_sign() -> Result<(), ObsError> {
114//!     let client = ObsClient {
115//!         endpoint: ENDPOINT.to_string(),
116//!         ak: AK.to_string(),
117//!         sk: SK.to_string(),
118//!         bucket: BUCKET_NAME.to_string(),
119//!     };
120//!     let sign_url = client.url_sign("https://ranfs.obs.cn-north-4.myhuaweicloud.com/tmp_cargo.txt")?;
121//!     println!("sign_url = {}", sign_url);
122//!     Ok(())
123//! }
124//! ```
125//! 
126//! ## 6. 批量上传文件
127//! 
128//! ```rust
129//! use obs_sdk::{ObsClient, ObsError};
130//!
131//! static ENDPOINT: &str = "obs.cn-north-4.myhuaweicloud.com";
132//! static AK: &str = "YOUR_AK";
133//! static SK: &str = "YOUR_SK";
134//! static BUCKET_NAME: &str = "bucket_name";
135//! 
136//! #[tokio::test]
137//! async fn test_upload_files() -> Result<(), ObsError> {
138//!     let client = ObsClient {
139//!         endpoint: ENDPOINT.to_string(),
140//!         ak: AK.to_string(),
141//!         sk: SK.to_string(),
142//!         bucket: BUCKET_NAME.to_string(),
143//!     };
144//!     let files = vec![
145//!         ("file1.txt", "local/path/file1.txt"),
146//!         ("file2.txt", "local/path/file2.txt"),
147//!         ("file3.txt", "local/path/file3.txt"),
148//!     ];
149//!     let results = client.upload_files(files).await;
150//!     for (i, result) in results.iter().enumerate() {
151//!         match result {
152//!             Ok(_) => println!("文件{}上传成功", i),
153//!             Err(e) => eprintln!("文件{}上传失败: {:?}", i, e),
154//!         }
155//!     }
156//!     Ok(())
157//! }
158//! ```
159//! 
160//! ## 7. 批量下载文件
161//! 
162//! ```rust
163//! use obs_sdk::{ObsClient, ObsError};
164//!
165//! static ENDPOINT: &str = "obs.cn-north-4.myhuaweicloud.com";
166//! static AK: &str = "YOUR_AK";
167//! static SK: &str = "YOUR_SK";
168//! static BUCKET_NAME: &str = "bucket_name";
169//! 
170//! #[tokio::test]
171//! async fn test_download_files() -> Result<(), ObsError> {
172//!     let client = ObsClient {
173//!         endpoint: ENDPOINT.to_string(),
174//!         ak: AK.to_string(),
175//!         sk: SK.to_string(),
176//!         bucket: BUCKET_NAME.to_string(),
177//!     };
178//!     let files = vec![
179//!         ("remote/file1.txt", "local/path/file1.txt", false),
180//!         ("remote/file2.txt", "local/path/file2.txt", true),
181//!         ("remote/file3.txt", "local/path/file3.txt", false),
182//!     ];
183//!     let results = client.download_files(files).await;
184//!     for (i, result) in results.iter().enumerate() {
185//!         match result {
186//!             Ok(_) => println!("文件{}下载成功", i),
187//!             Err(e) => eprintln!("文件{}下载失败: {:?}", i, e),
188//!         }
189//!     }
190//!     Ok(())
191//! }
192//! ```
193//! 
194mod utils;
195mod algorithm;
196
197use algorithm::HmacSha1;
198use futures::stream::{self, StreamExt};
199use std::{fmt, fs, io};
200use std::path::Path;
201use serde::{Serialize, Deserialize};
202use regex::Regex;
203use std::vec::Vec;
204use url::{Url, form_urlencoded};
205use chrono::Local;
206use reqwest::StatusCode;
207use urlencoding::encode;
208
209/// 华为云OBS客户端
210/// 
211pub struct ObsClient {
212    pub endpoint: String,
213    pub ak: String,
214    pub sk: String,
215    pub bucket: String,
216}
217
218pub enum ObsError {
219    Common(Box<dyn core::error::Error + Send + Sync + 'static>),
220}
221
222// 兼容世界上所有的标准错误
223impl <E> From<E> for ObsError
224where
225    E: core::error::Error + Send + Sync + 'static
226{
227    fn from(err: E) -> ObsError {
228        ObsError::Common(Box::new(err))
229    }
230}
231
232impl fmt::Debug for ObsError {
233    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
234        let mut builder = f.debug_struct("obs-sdk::Error");
235        builder.finish()
236    }
237}
238
239impl fmt::Display for ObsError {
240    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
241        match self {
242            ObsError::Common(e) => f.write_str(&format!("obs sdk error: {:?}", e))?,
243        };
244        Ok(())
245    }
246}
247
248
249
250impl ObsClient {
251
252    /// 列举指定前缀开头的所有对象元数据，方法内部会构造http请求：
253    /// ```plain
254    /// GET / HTTP/1.1
255    /// Host: bucketname.obs.cn-north-4.myhuaweicloud.com
256    /// Date: date
257    /// Authorization: authorization
258    /// ```
259    /// 
260    /// # 参数
261    /// 
262    /// `prefix` - 对象名前缀
263    /// `is_all` - 是否列举所有满足prefix的对象，true表示分页查询所有对象，false表示只查询第一页（最多1000个）
264    /// 
265    pub async fn list(&self, prefix: &str, is_all: bool) -> Result<Vec<ObjectMeta>, ObsError> {
266        let mut all_objects = Vec::new();
267        let mut marker: Option<String> = None;
268
269        loop {
270            let url = match &marker {
271                Some(m) => format!("https://{}.{}/?prefix={}&marker={}", self.bucket, self.endpoint, prefix, m),
272                None => format!("https://{}.{}/?prefix={}", self.bucket, self.endpoint, prefix),
273            };
274
275            let date = utils::now_str_gmt();
276            let hmacsha1 = HmacSha1();
277            let string_to_sign = hmacsha1.header_string_to_sign("GET", "", "", &date, "", &format!("/{}/", self.bucket));
278            let signature = hmacsha1.sign_to_base64string(&string_to_sign, &self.sk);
279            let authorization = format!("OBS {}:{}", self.ak, signature);
280
281            let client = reqwest::Client::new();
282            let res = client.get(&url)
283                .header("Date", &date)
284                .header("Authorization", &authorization)
285                .send()
286                .await?;
287
288            if !res.status().is_success() {
289                return Err(ObsError::Common(Box::new(io::Error::new(
290                    io::ErrorKind::Other,
291                    format!("请求失败，状态码={}", res.status())
292                ))));
293            }
294
295            let xml_content_string = res.text().await?;
296            let result = XmlParser::new(&xml_content_string).parse();
297
298            all_objects.extend(result.objects);
299
300            if !is_all || !result.is_truncated {
301                break;
302            }
303
304            marker = result.next_marker;
305        }
306
307        Ok(all_objects)
308    }
309
310    /// 上传对象
311    /// 
312    /// 方法内部构建请求
313    /// ```plain
314    /// PUT /object01 HTTP/1.1
315    /// User-Agent: curl/7.29.0
316    /// Host: examplebucket.obs.cn-north-4.myhuaweicloud.com
317    /// Accept: */*
318    /// Date: WED, 01 Jul 2015 04:11:15 GMT
319    /// Authorization: OBS H4IPJX0TQTHTHEBQQCEC:gYqplLq30dEX7GMi2qFWyjdFsyw=
320    /// Content-Length: 10240
321    /// Expect: 100-continue
322    /// 
323    /// [1024 Byte data content]
324    /// ```
325    /// 
326    pub async fn upload_object(&self, obj_key: &str, data: Vec<u8>) -> Result<(), ObsError> {
327        let obj_key = &Self::urlencode(obj_key);
328        // 构造完整的url地址
329        let url = format!("https://{}.{}/{}", self.bucket, self.endpoint, obj_key);
330
331        let md5_string = utils::base64_md5_str(&data);
332
333        //  获取GMT格式的时间字符串
334        let date = utils::now_str_gmt();
335
336        // 创建HmacSha1对象
337        let hmacsha1 = HmacSha1();
338
339        let file_type = &utils::get_mime_type_from_extension(obj_key)
340            .unwrap_or(String::from("application/octet-stream"));
341
342        // 构造签名用的原始字符串
343        let string_to_sign = hmacsha1.header_string_to_sign("PUT", &md5_string, file_type, &date, "",  &format!("/{}/{}", self.bucket, obj_key));
344
345        // 根据原始字符串+ak，获取header签名
346        let signature = hmacsha1.sign_to_base64string(&string_to_sign, &self.sk);
347
348        // 构造请求头Authorization的值
349        let authorization = format!("OBS {}:{}", self.ak, signature);
350
351        // 构造http请求
352        let client = reqwest::Client::new();
353        let res = client.put(url)
354            .header("Content-MD5", &md5_string)
355            .header("Date", &date)
356            .header("Content-Type", file_type)
357            .header("Content-Length", data.len())
358            .header("Authorization", authorization)
359            .body(data)
360            .send()
361            .await?;
362        let _status: StatusCode = res.status();
363        Ok(())
364
365    }
366
367    /// 上传文件
368    pub async fn upload_file(&self, obj_key: &str, file_path: &str) -> Result<(), ObsError> {
369        let data = fs::read(file_path)?;
370        self.upload_object(obj_key, data).await
371    }
372
373    /// 批量上传文件
374    /// 
375    /// 使用异步并发上传多个文件，并发数不超过32个
376    /// 
377    /// # 参数
378    /// 
379    /// `files` - 文件列表，每个元素是(obj_key, file_path)元组
380    /// 
381    /// # 返回
382    /// 
383    /// 返回一个Vec<Result<(), ObsError>>，每个元素对应一个文件的上传结果
384    /// 
385    pub async fn upload_files(&self, files: Vec<(&str, &str)>) -> Vec<Result<(), ObsError>> {
386        const MAX_CONCURRENT: usize = 32;
387        
388        stream::iter(files)
389            .map(|(obj_key, file_path)| async move {
390                self.upload_file(obj_key, file_path).await
391            })
392            .buffer_unordered(MAX_CONCURRENT)
393            .collect()
394            .await
395    }
396
397    /// 下载对象，方法内部会构造http请求：
398    /// ```plain
399    /// GET /{obj_key} HTTP/1.1
400    /// Host: {bucket}.obs.cn-north-4.myhuaweicloud.com
401    /// Date: {date}
402    /// ```
403    /// 
404    pub async fn download_object(&self, obj_key: &str) -> Result<Vec<u8>, ObsError> {
405        let obj_key = &Self::urlencode(obj_key);
406        // 构造完整的url地址
407        let url = format!("https://{}.{}/{}", self.bucket, self.endpoint, obj_key);
408
409        //  获取GMT格式的时间字符串
410        let date = utils::now_str_gmt();
411
412        // 创建HmacSha1对象
413        let hmacsha1 = HmacSha1();
414
415        // 构造签名用的原始字符串
416        let string_to_sign = hmacsha1.header_string_to_sign("GET", "", "", &date, "",  &format!("/{}/{}", self.bucket, obj_key));
417
418        // 根据原始字符串+ak，获取header签名
419        let signature = hmacsha1.sign_to_base64string(&string_to_sign, &self.sk);
420
421        // 构造请求头Authorization的值
422        let authorization = format!("OBS {}:{}", self.ak, signature);
423
424        // 构造http请求
425        let client = reqwest::Client::new();
426        let res = client.get(url)
427            .header("Authorization", &authorization)
428            .header("Date", &date)
429            .send()
430            .await?;
431
432        // 如果请求成功，则返回字节内容
433        if res.status().is_success() {
434            return Ok(res.bytes().await?.to_vec());
435        }
436        
437        Err(ObsError::Common(Box::new(io::Error::new(io::ErrorKind::Other, format!("请求失败，状态码={}", res.status())))))
438    }
439
440    /// 删除obs上的对象
441    pub async fn delete_object(&self, obj_key: &str) -> Result<(), ObsError> {
442        let obj_key = &Self::urlencode(obj_key);
443        // 构造完整的url地址
444        let url = format!("https://{}.{}/{}", self.bucket, self.endpoint, obj_key);
445
446        let md5_string = "";
447
448        //  获取GMT格式的时间字符串
449        let date = utils::now_str_gmt();
450
451        // 创建HmacSha1对象
452        let hmacsha1 = HmacSha1();
453
454        // 构造签名用的原始字符串
455        let string_to_sign = hmacsha1.header_string_to_sign("DELETE", &md5_string, "", &date, "",  &format!("/{}/{}", self.bucket, obj_key));
456
457        // 根据原始字符串+ak，获取header签名
458        let signature = hmacsha1.sign_to_base64string(&string_to_sign, &self.sk);
459
460        // 构造请求头Authorization的值
461        let authorization = format!("OBS {}:{}", self.ak, signature);
462
463        // 构造http请求
464        let client = reqwest::Client::new();
465        let res = client.delete(url)
466            .header("Date", &date)
467            .header("Authorization", authorization)
468            .send()
469            .await;
470
471        let res = match res {
472            Ok(response) => response,
473            Err(e) => {
474                return Err(ObsError::Common(Box::new(io::Error::new(io::ErrorKind::Other, e))));
475            },
476        };
477        let status = res.status();
478        println!("status = {}, {}", status, res.text_with_charset("utf-8").await?);
479
480        Ok(())
481    }
482
483    /// 下载文件，并指定本地保存用的文件路径
484    /// 
485    /// # 参数
486    /// 
487    /// `overwrite` - 是否覆盖，true，当文件存在时，覆盖文件，false，当文件存在时，不覆盖文件
488    /// 
489    pub async fn download_file(&self, obj_key: &str, file_path: &str, overwrite: bool) -> Result<(), ObsError> {
490        let file_path = Path::new(file_path);
491
492        // 判断文件是否存在，如果存在，不做任何操作
493        if file_path.exists() && !overwrite {
494            return Err(ObsError::Common(Box::new(io::Error::new(io::ErrorKind::AlreadyExists, "文件已存在，请删除文件或设置覆盖参数"))));
495        }
496
497        // 根据父目录是否存在，选择性创建父目录
498        let parent = file_path.parent().unwrap();
499        if !parent.exists() {
500            fs::create_dir_all(&parent)?;
501        }
502
503        // 下载文件，得到原始文件字节内容
504        let data = self.download_object(obj_key).await?;
505        
506        // 保存文件
507        fs::write(file_path, data)?;
508        Ok(())
509    }
510
511    /// 批量下载文件
512    /// 
513    /// 使用异步并发下载多个文件，并发数不超过32个
514    /// 
515    /// # 参数
516    /// 
517    /// `files` - 文件列表，每个元素是(obj_key, file_path, overwrite)元组
518    /// 
519    /// # 返回
520    /// 
521    /// 返回一个Vec<Result<(), ObsError>>，每个元素对应一个文件的下载结果
522    /// 
523    pub async fn download_files(&self, files: Vec<(&str, &str, bool)>) -> Vec<Result<(), ObsError>> {
524        const MAX_CONCURRENT: usize = 32;
525        
526        stream::iter(files)
527            .map(|(obj_key, file_path, overwrite)| async move {
528                self.download_file(obj_key, file_path, overwrite).await
529            })
530            .buffer_unordered(MAX_CONCURRENT)
531            .collect()
532            .await
533    }
534
535    pub fn url_sign(&self, url_str: &str) -> Result<String, ObsError> {
536        let obs_object_url = Url::parse(url_str)?;
537        let resource_part = obs_object_url.path();
538        let host = obs_object_url.host().unwrap();
539        let domain = match host {
540            url::Host::Domain(domain) => domain.to_string(),
541            _ => format!("{}.{}", self.bucket, self.endpoint)
542        };
543        let parts: Vec<&str> = domain.split(".").collect();
544        let bucket_name = parts[0];
545
546        let timestamp = utils::timestamp(Local::now(), 3600*2);
547
548        //  获取GMT格式的时间字符串
549        let expires = format!("{}", timestamp);
550
551        // 创建HmacSha1对象
552        let hmacsha1 = HmacSha1();
553
554        // 构造签名用的原始字符串
555        let string_to_sign = hmacsha1.url_string_to_sign("GET", "", "", &expires, "",  &format!("/{}{}", bucket_name, resource_part));
556
557        // 根据原始字符串+ak，获取header签名
558        let signature = hmacsha1.sign_to_base64string(&string_to_sign, &self.sk);
559        let signature = form_urlencoded::byte_serialize(signature.as_bytes()).collect::<String>();
560
561        // 构造url
562        let sign_url = format!("{}?AccessKeyId={}&Expires={}&Signature={}", url_str, self.ak, expires, signature);
563        Ok(sign_url)
564    }
565
566    fn urlencode(s: &str) -> String {
567        let tokens: Vec<String> = s.split("/").map(|token| {
568            encode(token).to_string()
569        }).collect();
570        tokens.join("/")
571    }
572
573}
574
575
576/// obs对象的元数据信息
577/// 
578/// 这个结构体用于表示 OBS 对象的元数据，包含对象的各种属性，如名称、修改时间、内容标识、大小以及存储类型。
579#[derive(Serialize, Deserialize, Debug)]
580pub struct ObjectMeta {
581
582    /// 对象名
583    /// 
584    /// 唯一标识 OBS 存储中的对象
585    pub key: String,
586
587    /// 对象最近一次被修改的时间（UTC时间）
588    /// 
589    /// 该时间戳表示对象在 OBS 存储中最后一次被修改的时刻，采用 UTC 时间格式。
590    pub last_modified: String,
591
592    /// 对象的base64编码的128位MD5摘要
593    /// 
594    /// 这个 ETag 值是对象内容的唯一标识，可以通过该值识别对象内容是否有变化。
595    pub etag: String,
596
597    /// 对象的字节数
598    /// 
599    /// 表示对象在存储中占用的字节大小
600    pub size: u64,
601
602    /// 对象的存储类型：STANDARD，WARM，COLD，DEEP_ARCHIVE
603    /// 
604    /// 不同的存储类型对应不同的存储成本和访问性能，用户可以根据对象的访问频率等因素选择合适的存储类型
605    pub storage_class: String,
606}
607
608/// 列举桶内对象的响应结果
609#[derive(Debug)]
610struct ListObjectsResult {
611    objects: Vec<ObjectMeta>,
612    is_truncated: bool,
613    next_marker: Option<String>,
614}
615
616/// XML解析器
617/// 
618/// 用于解析XML格式的响应数据，目前这里面针对obs的接口"列举桶内对象"的响应结果进行解析，没有进行通用的xml解析，其不能作为通用工具使用
619struct XmlParser { 
620    xml: String,
621}
622
623
624impl XmlParser {
625    fn new(xml: &str) -> Self {
626        XmlParser { xml: xml.to_string() }
627    }
628
629    /// 解析obs接口"列举桶内对象"的响应结果
630    /// 
631    /// 该内部采用正则表达式进行解析，因此依赖外部的regex库
632    fn parse(&self) -> ListObjectsResult {
633        let xml = &self.xml;
634
635        // 定义解析需要使用的正则表达式
636        let contents_re = Regex::new(r#"<Contents>(.*?)</Contents>"#).unwrap();
637        let key_regex = Regex::new(r#"<Key>(.*?)</Key>"#).unwrap();
638        let last_modified_regex = Regex::new(r#"<LastModified>(.*?)</LastModified>"#).unwrap();
639        let etag_regex = Regex::new(r#"<ETag>(.*?)</ETag>"#).unwrap();
640        let size_regex = Regex::new(r#"<Size>(.*?)</Size>"#).unwrap();
641        let storage_class_regex = Regex::new(r#"<StorageClass>(.*?)</StorageClass>"#).unwrap();
642        let is_truncated_regex = Regex::new(r#"<IsTruncated>(.*?)</IsTruncated>"#).unwrap();
643        let next_marker_regex = Regex::new(r#"<NextMarker>(.*?)</NextMarker>"#).unwrap();
644
645
646        // 解析Contents标签内的数据
647        let mut contents_vec = Vec::new();
648        for captures in contents_re.captures_iter(xml) {
649            let inner_content = &captures[1];
650
651            let key = key_regex.captures(inner_content).map(|cap| cap[1].to_string()).unwrap_or_default();
652            let last_modified = last_modified_regex.captures(inner_content).map(|cap| cap[1].to_string()).unwrap_or_default();
653            let etag = etag_regex.captures(inner_content).map(|cap| cap[1].to_string()).unwrap_or_default();
654            let size = size_regex.captures(inner_content).and_then(|cap| cap[1].parse().ok()).unwrap_or(0);
655            let storage_class = storage_class_regex.captures(inner_content).map(|cap| cap[1].to_string()).unwrap_or_default();
656            let content = ObjectMeta {
657                key,
658                last_modified,
659                etag,
660                size,
661                storage_class,
662            };
663            contents_vec.push(content);
664        }
665
666        // 解析IsTruncated字段
667        let is_truncated = is_truncated_regex.captures(xml)
668            .and_then(|cap| cap[1].to_string().parse::<bool>().ok())
669            .unwrap_or(false);
670
671        // 解析NextMarker字段
672        let next_marker = next_marker_regex.captures(xml)
673            .map(|cap| cap[1].to_string());
674
675        ListObjectsResult {
676            objects: contents_vec,
677            is_truncated,
678            next_marker,
679        }
680    }
681}
682
683
684#[cfg(test)]
685mod tests {
686    use super::*;
687    use chrono::{Duration, Local};
688    use std::time::{SystemTime, UNIX_EPOCH};
689
690    #[test]
691    fn test_parse_xml() {
692        let xml = r#"<?xml version="1.0" encoding="UTF-8" standalone="yes"?><ListBucketResult xmlns="http://obs.myhwclouds.com/doc/2015-06-30/"><Name>obs-products</Name><Prefix>tmp</Prefix><Marker></Marker><MaxKeys>1000</MaxKeys><IsTruncated>false</IsTruncated><Contents><Key>tmp/</Key><LastModified>2024-12-03T12:01:48.020Z</LastModified><ETag>"d41d8cd98f00b204e9800998ecf8427e"</ETag><Size>0</Size><Owner><ID>74df55bf376f41d48959d2aa9deaaf38</ID></Owner><StorageClass>STANDARD</StorageClass></Contents><Contents><Key>tmp/index001.png</Key><LastModified>2025-08-20T07:42:59.813Z</LastModified><ETag>"de317c0b7b6e02b42ef2b9e29bb5906a"</ETag><Size>12082</Size><Owner><ID>74df55bf376f41d48959d2aa9deaaf38</ID></Owner><StorageClass>STANDARD</StorageClass></Contents><Contents><Key>tmp/index002.png</Key><LastModified>2025-08-20T07:52:10.204Z</LastModified><ETag>"de317c0b7b6e02b42ef2b9e29bb5906a"</ETag><Size>12082</Size><Owner><ID>74df55bf376f41d48959d2aa9deaaf38</ID></Owner><StorageClass>STANDARD</StorageClass></Contents></ListBucketResult>"#;
693        let parser = XmlParser::new(xml);
694        let result = parser.parse();
695        let json_data = serde_json::to_string_pretty(&result.objects).unwrap();
696        println!("{}", json_data);
697        println!("is_truncated: {}", result.is_truncated);
698        println!("next_marker: {:?}", result.next_marker);
699    }
700
701    #[test]
702    fn test_parse_xml_with_next_marker() {
703        let xml = r#"<?xml version="1.0" encoding="UTF-8" standalone="yes"?><ListBucketResult xmlns="http://obs.myhwclouds.com/doc/2015-06-30/"><Name>obs-products</Name><Prefix>tmp</Prefix><Marker></Marker><MaxKeys>1000</MaxKeys><IsTruncated>true</IsTruncated><NextMarker>tmp/index999.png</NextMarker><Contents><Key>tmp/index001.png</Key><LastModified>2025-08-20T07:42:59.813Z</LastModified><ETag>"de317c0b7b6e02b42ef2b9e29bb5906a"</ETag><Size>12082</Size><Owner><ID>74df55bf376f41d48959d2aa9deaaf38</ID></Owner><StorageClass>STANDARD</StorageClass></Contents></ListBucketResult>"#;
704        let parser = XmlParser::new(xml);
705        let result = parser.parse();
706        assert_eq!(result.is_truncated, true);
707        assert_eq!(result.next_marker, Some("tmp/index999.png".to_string()));
708        println!("is_truncated: {}", result.is_truncated);
709        println!("next_marker: {:?}", result.next_marker);
710    }
711
712    #[test]
713    fn test_timestamp() {
714        let now = Local::now();
715        let two_hours = Duration::hours(2);
716        let future_time = now + two_hours;
717
718        let system_time: SystemTime = future_time.into();
719        let duration = system_time.duration_since(UNIX_EPOCH).unwrap();
720        let timestamp = duration.as_secs();
721        println!("timestamp = {}", timestamp);
722    }
723
724}
obs_sdk/lib.rs

obs_sdk/
lib.rs
