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
use reqwest::header::{HeaderMap, DATE, CONTENT_TYPE};
use chrono::prelude::*;
use crate::auth::oss_sign_header;
use bytes::Bytes;
#[doc(alias = "oss")]
pub struct OSS<'a> {
pub endpoint: &'a str,
pub http_client: &'a reqwest::Client,
}
impl<'a> OSS<'a> {
pub async fn get_object(&self, bucket: &str, object: &str) -> reqwest::Result<Bytes> {
let mut headers = HeaderMap::new();
self.add_date_and_sign(&mut headers, bucket, object);
let resp = self.http_client
.get(&self.request_url(bucket, object))
.headers(headers)
.send().await?;
resp.bytes().await
}
pub async fn put_object(&self, bucket: &str, object: &str,
content_type: &str, content: &[u8]) -> reqwest::Result<bool> {
let mut headers = HeaderMap::new();
headers.insert(CONTENT_TYPE, content_type.parse().unwrap());
self.add_date_and_sign(&mut headers, bucket, object);
let resp = self.http_client
.put(&self.request_url(bucket, object))
.headers(headers)
.body(content.to_owned())
.send().await?;
return resp.error_for_status().map(|response| response.status().is_success());
}
pub async fn delete_object(&self, bucket: &str, object: &str) -> reqwest::Result<bool> {
let mut headers = HeaderMap::new();
self.add_date_and_sign(&mut headers, bucket, object);
let resp = self.http_client
.delete(&self.request_url(bucket, object))
.headers(headers)
.send().await?;
return resp.error_for_status().map(|response| response.status().is_success());
}
fn request_url(&self, bucket: &str, object: &str) -> String {
format!("https://{}.{}/{}", bucket, self.endpoint, object)
}
fn add_date_and_sign(&self, headers: &mut HeaderMap, bucket: &str, object: &str) {
let now = Utc::now().format("%a, %d %b %Y %T GMT").to_string();
headers.insert(DATE, now.parse().unwrap());
let authorization = oss_sign_header("GET", bucket, object, &headers);
headers.insert("Authorization", authorization.parse().unwrap());
}
}
#[cfg(test)]
mod tests {
use crate::oss::OSS;
use std::str::from_utf8;
use bytes::Buf;
#[tokio::test]
async fn test_get_object() -> Result<(), Box<dyn std::error::Error>> {
let endpoint = "oss-cn-hangzhou.aliyuncs.com";
let ref http_client = reqwest::Client::new();
let oss = OSS { endpoint, http_client };
let bytes = oss.get_object("eren-assets", "hello.txt").await?;
print!("object: {}", from_utf8(bytes.bytes()).unwrap());
Ok(())
}
#[tokio::test]
async fn test_put_object() -> Result<(), Box<dyn std::error::Error>> {
let endpoint = "oss-cn-hangzhou.aliyuncs.com";
let ref http_client = reqwest::Client::new();
let oss = OSS { endpoint, http_client };
let result = oss.put_object("eren-assets", "hello.txt", "text/plain", "hello".as_bytes()).await?;
print!("result: {}", result);
Ok(())
}
}