mirror of
https://github.com/minio/minio-rs.git
synced 2025-12-06 15:26:51 +08:00
Fix object download to file (#105)
The ObjectContent::to_file() function was not working properly, as it could not write in a temporary file (created with tokio::fs::File::open(), which opens a file in read-only mode). To solve this issue, modified the way temporary files are opened to be able to write inside of it. * Add an example to download a file This new example demonstrates how to download an object and store its content into a file. --------- Co-authored-by: Henk-Jan Lebbink <henkjan.lebbink@gmail.com> Co-authored-by: Jean Sainctavit <jean.sainctavit@qarnot-computing.com>
This commit is contained in:
parent
0438f044ff
commit
26d67b80df
@ -63,5 +63,8 @@ quickcheck = "1.0.3"
|
||||
[[example]]
|
||||
name = "file_uploader"
|
||||
|
||||
[[example]]
|
||||
name = "file_downloader"
|
||||
|
||||
[[example]]
|
||||
name = "object_prompt"
|
||||
|
||||
@ -15,6 +15,10 @@ Run the examples from the command line with:
|
||||
* [Upload a file to MinIO](examples/file_uploader)
|
||||
* [Upload a file to MinIO with CLI](examples/put_object)
|
||||
|
||||
### file_downloader.rs
|
||||
|
||||
* [Download af file from MinIO](examples/file_downloader.rs)
|
||||
|
||||
### object_prompt.rs
|
||||
|
||||
* [Prompt a file on MinIO](examples/object_prompt)
|
||||
|
||||
75
examples/file_downloader.rs
Normal file
75
examples/file_downloader.rs
Normal file
@ -0,0 +1,75 @@
|
||||
use minio::s3::args::{BucketExistsArgs, MakeBucketArgs};
|
||||
use minio::s3::builders::ObjectContent;
|
||||
use minio::s3::client::ClientBuilder;
|
||||
use minio::s3::creds::StaticProvider;
|
||||
use minio::s3::http::BaseUrl;
|
||||
use minio::s3::types::S3Api;
|
||||
use std::path::Path;
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
||||
env_logger::init(); // Note: set environment variable RUST_LOG="INFO" to log info and higher
|
||||
|
||||
let base_url = "https://play.min.io".parse::<BaseUrl>()?;
|
||||
log::info!("Trying to connect to MinIO at: `{:?}`", base_url);
|
||||
|
||||
let static_provider = StaticProvider::new(
|
||||
"Q3AM3UQ867SPQQA43P2F",
|
||||
"zuf+tfteSlswRu7BJ86wekitnifILbZam1KYY3TG",
|
||||
None,
|
||||
);
|
||||
|
||||
let client = ClientBuilder::new(base_url.clone())
|
||||
.provider(Some(Box::new(static_provider)))
|
||||
.build()?;
|
||||
|
||||
let bucket_name: &str = "file-download-rust-bucket";
|
||||
let object_name: &str = "cat.png";
|
||||
|
||||
// File we are going to upload to the bucket
|
||||
let filename: &Path = Path::new("./examples/cat.png");
|
||||
let download_path: &str = &format!("/tmp/downloads/{object_name}");
|
||||
|
||||
// Check 'bucket_name' bucket exist or not.
|
||||
let exists: bool = client
|
||||
.bucket_exists(&BucketExistsArgs::new(bucket_name).unwrap())
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
// Make 'bucket_name' bucket if not exist.
|
||||
if !exists {
|
||||
client
|
||||
.make_bucket(&MakeBucketArgs::new(bucket_name).unwrap())
|
||||
.await
|
||||
.unwrap();
|
||||
}
|
||||
|
||||
if filename.exists() {
|
||||
log::info!("File '{}' exists.", &filename.to_str().unwrap());
|
||||
} else {
|
||||
log::error!("File '{}' does not exist.", &filename.to_str().unwrap());
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
let content = ObjectContent::from(filename);
|
||||
client
|
||||
.put_object_content(bucket_name, object_name, content)
|
||||
.send()
|
||||
.await?;
|
||||
|
||||
log::info!(
|
||||
"file '{}' is successfully uploaded as object '{object_name}' to bucket '{bucket_name}'.",
|
||||
filename.display()
|
||||
);
|
||||
|
||||
let get_object = client.get_object(bucket_name, object_name).send().await?;
|
||||
|
||||
get_object
|
||||
.content
|
||||
.to_file(&Path::new(download_path))
|
||||
.await?;
|
||||
|
||||
log::info!("Object '{object_name}' is successfully downloaded to file '{download_path}'.");
|
||||
|
||||
Ok(())
|
||||
}
|
||||
@ -201,8 +201,11 @@ impl ObjectContent {
|
||||
.join(Path::new(tmp_file_name.as_os_str()));
|
||||
|
||||
let mut total = 0;
|
||||
{
|
||||
let mut fp = fs::File::open(&tmp_file_path).await?;
|
||||
let mut fp = fs::OpenOptions::new()
|
||||
.write(true)
|
||||
.create(true)
|
||||
.open(&tmp_file_path)
|
||||
.await?;
|
||||
let (mut r, _) = self.to_stream().await?;
|
||||
while let Some(bytes) = r.next().await {
|
||||
let bytes = bytes?;
|
||||
@ -213,7 +216,6 @@ impl ObjectContent {
|
||||
fp.write_all(&bytes).await?;
|
||||
}
|
||||
fp.flush().await?;
|
||||
}
|
||||
fs::rename(&tmp_file_path, file_path).await?;
|
||||
Ok(total)
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user