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]]
|
[[example]]
|
||||||
name = "file_uploader"
|
name = "file_uploader"
|
||||||
|
|
||||||
|
[[example]]
|
||||||
|
name = "file_downloader"
|
||||||
|
|
||||||
[[example]]
|
[[example]]
|
||||||
name = "object_prompt"
|
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](examples/file_uploader)
|
||||||
* [Upload a file to MinIO with CLI](examples/put_object)
|
* [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
|
### object_prompt.rs
|
||||||
|
|
||||||
* [Prompt a file on MinIO](examples/object_prompt)
|
* [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()));
|
.join(Path::new(tmp_file_name.as_os_str()));
|
||||||
|
|
||||||
let mut total = 0;
|
let mut total = 0;
|
||||||
{
|
let mut fp = fs::OpenOptions::new()
|
||||||
let mut fp = fs::File::open(&tmp_file_path).await?;
|
.write(true)
|
||||||
|
.create(true)
|
||||||
|
.open(&tmp_file_path)
|
||||||
|
.await?;
|
||||||
let (mut r, _) = self.to_stream().await?;
|
let (mut r, _) = self.to_stream().await?;
|
||||||
while let Some(bytes) = r.next().await {
|
while let Some(bytes) = r.next().await {
|
||||||
let bytes = bytes?;
|
let bytes = bytes?;
|
||||||
@ -213,7 +216,6 @@ impl ObjectContent {
|
|||||||
fp.write_all(&bytes).await?;
|
fp.write_all(&bytes).await?;
|
||||||
}
|
}
|
||||||
fp.flush().await?;
|
fp.flush().await?;
|
||||||
}
|
|
||||||
fs::rename(&tmp_file_path, file_path).await?;
|
fs::rename(&tmp_file_path, file_path).await?;
|
||||||
Ok(total)
|
Ok(total)
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user