use async_trait::async_trait;
use futures_core::future::BoxFuture;
use std::future::Future;
use std::time::Duration;
use tokio::runtime::Handle;
use ve_tos_rust_sdk::common::init_tracing_log;
use ve_tos_rust_sdk::asynchronous::bucket::BucketAPI;
use ve_tos_rust_sdk::asynchronous::tos;
use ve_tos_rust_sdk::asynchronous::tos::AsyncRuntime;
use ve_tos_rust_sdk::bucket::{CORSRule, PutBucketCORSInput};
#[derive(Debug, Default)]
pub struct TokioRuntime {}
impl AsyncRuntime for TokioRuntime {
type JoinError = tokio::task::JoinError;
async fn sleep(&self, duration: Duration) {
tokio::time::sleep(duration).await;
fn spawn<'a, F>(&self, future: F) -> BoxFuture<'a, Result<F::Output, Self::JoinError>>
F: Future + Send + 'static,
F::Output: Send + 'static,
Box::pin(Handle::current().spawn(future))
fn block_on<F: Future>(&self, future: F) -> F::Output {
Handle::current().block_on(future)
let _guard = init_tracing_log("info", "temp/logs", "app.log");
let ak = env::var("TOS_ACCESS_KEY").unwrap_or("".to_string());
let sk = env::var("TOS_SECRET_KEY").unwrap_or("".to_string());
let endpoint = "https://tos-cn-beijing.volces.com";
let region = "cn-beijing";
let client = tos::builder::<TokioRuntime>()
.connection_timeout(3000)
let bucket_name = "my-single-bucket";
let mut rule = CORSRule::new();
rule.set_allowed_origins(vec!["*".to_string()]);
rule.set_allowed_methods(vec!["PUT".to_string(), "GET".to_string()]);
rule.set_allowed_headers(vec!["Authorization".to_string()]);
rule.set_expose_headers(vec!["x-tos-request-id".to_string()]);
rule.set_max_age_seconds(1024);
rule.set_response_vary(true);
let input = PutBucketCORSInput::new_with_rules(bucket_name, vec![rule]);
let output = client.put_bucket_cors(&input).await;
Ok(_) => println!("Successfully set CORS rules for bucket: {}", bucket_name),
Err(err) => eprintln!("Failed to set CORS rules: {:?}", err),