Fix uploading (#305)

* Upgrade rust-s3 to fix tokio panics

* Run fmt

* Update deps
This commit is contained in:
Geometrically
2022-02-20 20:16:32 -07:00
committed by GitHub
parent 3ee144459f
commit 9492363b22
6 changed files with 416 additions and 812 deletions

View File

@@ -8,8 +8,6 @@ mod s3_host;
pub use backblaze::BackblazeHost;
use bytes::Bytes;
pub use mock::MockHost;
use s3::creds::AwsCredsError;
use s3::S3Error;
pub use s3_host::S3Host;
#[derive(Error, Debug)]
@@ -19,9 +17,7 @@ pub enum FileHostingError {
#[error("Backblaze error: {0}")]
BackblazeError(serde_json::Value),
#[error("S3 error: {0}")]
S3Error(#[from] S3Error),
#[error("S3 Authentication error: {0}")]
S3CredentialsError(#[from] AwsCredsError),
S3Error(String),
#[error("File system error in file hosting: {0}")]
FileSystemError(#[from] std::io::Error),
#[error("Invalid Filename")]

View File

@@ -24,8 +24,13 @@ impl S3Host {
region: bucket_region.to_string(),
endpoint: url.to_string(),
},
Credentials::new(Some(access_token), Some(secret), None, None, None)?,
)?;
Credentials::new(Some(access_token), Some(secret), None, None, None).map_err(|_| {
FileHostingError::S3Error("Error while creating credentials".to_string())
})?,
)
.map_err(|_| {
FileHostingError::S3Error("Error while creating Bucket instance".to_string())
})?;
bucket.add_header("x-amz-acl", "public-read");
@@ -46,7 +51,10 @@ impl FileHost for S3Host {
self.bucket
.put_object_with_content_type(format!("/{}", file_name), &*file_bytes, content_type)
.await?;
.await
.map_err(|_| {
FileHostingError::S3Error("Error while uploading file to S3".to_string())
})?;
Ok(UploadFileData {
file_id: file_name.to_string(),
@@ -65,7 +73,12 @@ impl FileHost for S3Host {
file_id: &str,
file_name: &str,
) -> Result<DeleteFileData, FileHostingError> {
self.bucket.delete_object(format!("/{}", file_name)).await?;
self.bucket
.delete_object(format!("/{}", file_name))
.await
.map_err(|_| {
FileHostingError::S3Error("Error while deleting file from S3".to_string())
})?;
Ok(DeleteFileData {
file_id: file_id.to_string(),