Rust 集成 B2

197
2023/08/06 22:36:32

本章将讨论如何将 B2 集成到 Rust中。

依赖

  • rust-s3 即是我们需要的依赖项
  • 由于我们使用的是 tokio 异步库,所以加上其依赖

接下来,我们看一下它的基本操作。

授权

首先,你需要按照《使用 B2 存储》章节所述,注册并添加应用程序密钥。

rust-s3 的授权封装在 s3::creds::Credentials结构体中,对于 B2 而言,该结构体的以下两个字段必须填写:

  • access_key:对应B2应用程序的 keyID
  • secret_key:对应B2应用程序的 applicationKey

所以,它最终是这样的:

let cred =  s3::creds::Credentials {
    access_key: Some("你的keyID".to_string()),
    secret_key: Some("你的applicationKey".to_string()),
    session_token: None,
    security_token: None,
    expiration: None,
};

实例化桶

授权是实例化桶时的参数之一,它需要以下参数:

  • 桶名:你创建的应用程序密钥对应的桶名称
  • 区域:该桶的 Endpoint
  • 授权:上一步骤创建的授权
let bucket = s3::Bucket::new("桶名称", "区域".parse().unwrap, cred).unwrap();

实例化桶之后,就可以进行一系列操作了。

上传文件

上传文件常用的有两个方法:

  • put_object(远程路径, 数据)
  • put_object_with_content_type(远程路径, 数据, 内容类型)

两者区别在于,无论什么类型的文件,put_object() 的*content type(内容类型)*均为 application/octet-stream,而 put_object_with_content_type() 可以手动为不同的文件指定对应的内容类型。

下载文件

通过 get_object_to_writer() 可以下载文件并写入到指定目标。

let mut output_file = tokio::fs::File::create("/tmp/axum.rs.file").await.expect("创建文件失败");

let status_code = bucket.get_object_to_writer("axum.rs.file", &mut output_file).await?

删除文件

使用 delete_object() 可以删除指定的文件。

let resp = bucket.delete_object("axum.rs.file").await?;
assert_eq!(resp.status_code(), 204);
println!("{:?}", resp);