Skip to content

Commit 6033112

Browse files
committed
chore: fix comment
1 parent eb81346 commit 6033112

File tree

7 files changed

+7
-17
lines changed

7 files changed

+7
-17
lines changed

Cargo.lock

Lines changed: 1 addition & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

src/common/storage/src/config.rs

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -500,20 +500,17 @@ impl Debug for StorageRedisConfig {
500500
pub struct ShareTableConfig {
501501
pub share_endpoint_address: Option<String>,
502502
pub share_endpoint_token: RefreshableToken,
503-
pub requester_tenant_id: String,
504503
}
505504

506505
impl ShareTableConfig {
507506
pub fn init(
508507
share_endpoint_address: &str,
509508
token_file: &str,
510-
requester_tenant_id: String,
511509
default_token: String,
512510
) -> common_exception::Result<()> {
513511
GlobalInstance::set(Self::try_create(
514512
share_endpoint_address,
515513
token_file,
516-
requester_tenant_id,
517514
default_token,
518515
)?);
519516

@@ -523,7 +520,6 @@ impl ShareTableConfig {
523520
pub fn try_create(
524521
share_endpoint_address: &str,
525522
token_file: &str,
526-
requester_tenant_id: String,
527523
default_token: String,
528524
) -> common_exception::Result<ShareTableConfig> {
529525
let share_endpoint_address = if share_endpoint_address.is_empty() {
@@ -541,7 +537,6 @@ impl ShareTableConfig {
541537
Ok(ShareTableConfig {
542538
share_endpoint_address,
543539
share_endpoint_token,
544-
requester_tenant_id,
545540
})
546541
}
547542

@@ -553,10 +548,6 @@ impl ShareTableConfig {
553548
ShareTableConfig::instance().share_endpoint_token
554549
}
555550

556-
pub fn requester_tenant_id() -> String {
557-
ShareTableConfig::instance().requester_tenant_id
558-
}
559-
560551
pub fn instance() -> ShareTableConfig {
561552
GlobalInstance::get()
562553
}

src/query/service/src/global_services.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,6 @@ impl GlobalServices {
7171
&config.query.share_endpoint_address,
7272
&config.query.share_endpoint_auth_token_file,
7373
config.query.tenant_id.clone(),
74-
config.query.tenant_id.clone(),
7574
)?;
7675

7776
CacheManager::init(&config.query)?;

src/query/sharing/Cargo.toml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@ backon = "0.2"
1616
base64 = "0.13.0"
1717
bytes = "1"
1818
common-auth = { path = "../../common/auth" }
19+
common-config = { path = "../config" }
20+
1921
http = "0.2"
2022
log = "0.4"
2123
moka = "0.9"

src/query/sharing/src/layer.rs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,6 @@ use crate::SharedSigner;
5656
pub fn create_share_table_operator(
5757
share_endpoint_address: Option<String>,
5858
share_endpoint_token: RefreshableToken,
59-
requester_tenant_id: String,
6059
share_tenant_id: &str,
6160
share_name: &str,
6261
table_name: &str,
@@ -69,7 +68,6 @@ pub fn create_share_table_operator(
6968
share_endpoint_address, share_tenant_id, share_name, table_name
7069
),
7170
share_endpoint_token,
72-
requester_tenant_id,
7371
);
7472
Operator::new(apply_wrapper(SharedAccessor {
7573
signer,

src/query/sharing/src/signer.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ use anyhow::anyhow;
2222
use anyhow::Result;
2323
use bytes::Bytes;
2424
use common_auth::RefreshableToken;
25+
use common_config::GlobalConfig;
2526
use http::header::AUTHORIZATION;
2627
use http::header::CONTENT_LENGTH;
2728
use http::Method;
@@ -45,7 +46,6 @@ pub struct SharedSigner {
4546
cache: Cache<PresignRequest, PresignedRequest>,
4647
client: HttpClient,
4748
token: RefreshableToken,
48-
requester_tenant_id: String,
4949
}
5050

5151
impl Debug for SharedSigner {
@@ -58,7 +58,7 @@ impl Debug for SharedSigner {
5858

5959
impl SharedSigner {
6060
/// Create a new SharedSigner.
61-
pub fn new(endpoint: &str, token: RefreshableToken, req_tenant_id: String) -> Self {
61+
pub fn new(endpoint: &str, token: RefreshableToken) -> Self {
6262
let cache = Cache::builder()
6363
// Databend Cloud Presign will expire after 3600s (1 hour).
6464
// We will expire them 10 minutes before to avoid edge cases.
@@ -70,7 +70,6 @@ impl SharedSigner {
7070
cache,
7171
client: HttpClient::new(),
7272
token,
73-
requester_tenant_id: req_tenant_id,
7473
}
7574
}
7675

@@ -153,12 +152,13 @@ impl SharedSigner {
153152
.collect();
154153
let bs = Bytes::from(serde_json::to_vec(&reqs)?);
155154
let auth = self.token.to_header().await?;
155+
let requester = GlobalConfig::instance().as_ref().query.tenant_id.clone();
156156
let req = Request::builder()
157157
.method(Method::POST)
158158
.uri(&self.endpoint)
159159
.header(AUTHORIZATION, auth)
160160
.header(CONTENT_LENGTH, bs.len())
161-
.header(TENANT_HEADER, self.requester_tenant_id.clone())
161+
.header(TENANT_HEADER, requester)
162162
.body(AsyncBody::Bytes(bs))?;
163163
let resp = self.client.send_async(req).await?;
164164
let bs = resp.into_body().bytes().await?;

src/query/storages/fuse/fuse/src/fuse_table.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,6 @@ impl FuseTable {
9898
Some(ref from_share) => create_share_table_operator(
9999
ShareTableConfig::share_endpoint_address(),
100100
ShareTableConfig::share_endpoint_token(),
101-
ShareTableConfig::requester_tenant_id(),
102101
&from_share.tenant,
103102
&from_share.share_name,
104103
&table_info.name,

0 commit comments

Comments
 (0)