Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat(services/azdls): list start from and sas token support #5242

Draft
wants to merge 6 commits into
base: main
Choose a base branch
from
Draft

feat(services/azdls): list start from and sas token support #5242

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
ddfd3ae
feat(services/azdls): list start from and sas token support
alexwilcoxson-rel Oct 24, 2024
384c71e
chore(services/azdls): clippy
alexwilcoxson-rel Oct 24, 2024
aba400e
feat(services-azdls): refactored crc64 and pass list args through fla…
alexwilcoxson-rel Oct 29, 2024
c370249
feat(services-azdls): fmt
alexwilcoxson-rel Oct 29, 2024
df6cf90
Merge branch 'main' into azdls-list-start-from
alexwilcoxson-rel Oct 29, 2024
7be7756
Merge branch 'main' into azdls-list-start-from
alexwilcoxson-rel Oct 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 25 additions & 2 deletions core/src/services/azdls/backend.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -133,6 +133,15 @@ impl AzdlsBuilder {
self
}

/// Set sas_token of this backend.
pub fn sas_token(mut self, sas_token: &str) -> Self {
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Reqsign has sas_token support; we can use it here: https://docs.rs/reqsign/0.16.0/reqsign/struct.AzureStorageConfig.html#structfield.sas_token

Maybe it's worth a separate PR?

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Are you suggesting we use reqsign::AzureStorageConfig on AzdlsBuilder or AzdlsConfig or something else?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, we can construct like s3 does:

opendal/core/src/services/s3/backend.rs

Lines 745 to 783 in 3fe8caa

// This is our current config.
let mut cfg = AwsConfig::default();
if !self.config.disable_config_load {
#[cfg(not(target_arch = "wasm32"))]
{
cfg = cfg.from_profile();
cfg = cfg.from_env();
}
}
if let Some(v) = self.config.region.clone() {
cfg.region = Some(v);
}
if cfg.region.is_none() {
return Err(Error::new(
ErrorKind::ConfigInvalid,
"region is missing. Please find it by S3::detect_region() or set them in env.",
)
.with_operation("Builder::build")
.with_context("service", Scheme::S3));
}
let region = cfg.region.to_owned().unwrap();
debug!("backend use region: {region}");
// Building endpoint.
let endpoint = self.build_endpoint(&region);
debug!("backend use endpoint: {endpoint}");
// Setting all value from user input if available.
if let Some(v) = self.config.access_key_id {
cfg.access_key_id = Some(v)
}
if let Some(v) = self.config.secret_access_key {
cfg.secret_access_key = Some(v)
}
if let Some(v) = self.config.session_token {
cfg.session_token = Some(v)
}

if !sas_token.is_empty() {
self.config.sas_token = Some(sas_token.to_string());
}

self
}

/// Specify the http client that used by this service.
///
/// # Notes
Expand Down Expand Up @@ -188,7 +197,7 @@ impl Builder for AzdlsBuilder {
.clone()
.or_else(|| infer_storage_name_from_endpoint(endpoint.as_str())),
account_key: self.config.account_key.clone(),
sas_token: None,
sas_token: self.config.sas_token.clone(),
..Default::default()
};

Expand All @@ -203,6 +212,7 @@ impl Builder for AzdlsBuilder {
loader: cred_loader,
signer,
}),
has_sas_token: self.config.sas_token.is_some(),
})
}
}
Expand All @@ -211,6 +221,7 @@ impl Builder for AzdlsBuilder {
#[derive(Debug, Clone)]
pub struct AzdlsBackend {
core: Arc<AzdlsCore>,
has_sas_token: bool,
}

impl Access for AzdlsBackend {
Expand Down Expand Up @@ -238,6 +249,12 @@ impl Access for AzdlsBackend {
rename: true,

list: true,
list_with_start_after: true,

presign: self.has_sas_token,
presign_stat: self.has_sas_token,
presign_read: self.has_sas_token,
presign_write: self.has_sas_token,

..Default::default()
});
Expand Down Expand Up @@ -347,7 +364,13 @@ impl Access for AzdlsBackend {
}

async fn list(&self, path: &str, args: OpList) -> Result<(RpList, Self::Lister)> {
let l = AzdlsLister::new(self.core.clone(), path.to_string(), args.limit());
let start_after = args.start_after().map(|s| s.to_string());
let l = AzdlsLister::new(
alexwilcoxson-rel marked this conversation as resolved.
Show resolved Hide resolved
self.core.clone(),
path.to_string(),
args.limit(),
start_after,
);

Ok((RpList::default(), oio::PageLister::new(l)))
}
Expand Down
5 changes: 5 additions & 0 deletions core/src/services/azdls/config.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,8 @@ pub struct AzdlsConfig {
pub account_name: Option<String>,
/// Account key of this backend.
pub account_key: Option<String>,
/// SAS token of this backend.
pub sas_token: Option<String>,
}

impl Debug for AzdlsConfig {
Expand All @@ -50,6 +52,9 @@ impl Debug for AzdlsConfig {
if self.account_key.is_some() {
ds.field("account_key", &"<redacted>");
}
if self.sas_token.is_some() {
ds.field("sas_token", &"<redacted>");
}

ds.finish()
}
Expand Down
15 changes: 15 additions & 0 deletions core/src/services/azdls/core.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ use std::fmt::Debug;
use std::fmt::Formatter;
use std::fmt::Write;

use base64::Engine;
use http::header::CONTENT_DISPOSITION;
use http::header::CONTENT_LENGTH;
use http::header::CONTENT_TYPE;
Expand All @@ -34,6 +35,8 @@ use reqsign::AzureStorageSigner;
use crate::raw::*;
use crate::*;

use super::crc64;

const X_MS_RENAME_SOURCE: &str = "x-ms-rename-source";
const X_MS_VERSION: &str = "x-ms-version";

Expand Down Expand Up @@ -258,6 +261,7 @@ impl AzdlsCore {
pub async fn azdls_list(
&self,
path: &str,
start_after: Option<&str>,
continuation: &str,
limit: Option<usize>,
) -> Result<Response<Buffer>> {
Expand All @@ -276,9 +280,13 @@ impl AzdlsCore {
if let Some(limit) = limit {
write!(url, "&maxResults={limit}").expect("write into string must succeed");
}

if !continuation.is_empty() {
write!(url, "&continuation={}", percent_encode_path(continuation))
.expect("write into string must succeed");
} else if let Some(start_after) = start_after {
write!(url, "&continuation={}", create_continuation(start_after))
.expect("write into string must succeed");
}

let mut req = Request::get(&url)
Expand Down Expand Up @@ -318,3 +326,10 @@ impl AzdlsCore {
}
}
}

fn create_continuation(start_after: &str) -> String {
alexwilcoxson-rel marked this conversation as resolved.
Show resolved Hide resolved
let crc = crc64::compute(format!("{}#$0", start_after).as_bytes());
let token = format!("{} 0 {}", crc, start_after);
let token = base64::prelude::BASE64_STANDARD.encode(token.as_bytes());
token
}
Loading
Loading