Skip to content

Commit 626c8c5

Browse files
committed
refactor(query): rm IDMConfig.
1 parent 5f1af5e commit 626c8c5

File tree

2 files changed

+20
-31
lines changed

2 files changed

+20
-31
lines changed

src/query/config/src/outer_v0.rs

Lines changed: 18 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -823,8 +823,7 @@ pub struct QueryConfig {
823823
pub async_insert_stale_timeout: u64,
824824

825825
#[clap(skip)]
826-
#[serde(flatten)]
827-
pub idm: IDMConfig,
826+
users: Vec<UserConfig>,
828827
}
829828

830829
impl Default for QueryConfig {
@@ -877,7 +876,9 @@ impl TryInto<InnerQueryConfig> for QueryConfig {
877876
async_insert_max_data_size: self.async_insert_max_data_size,
878877
async_insert_busy_timeout: self.async_insert_busy_timeout,
879878
async_insert_stale_timeout: self.async_insert_stale_timeout,
880-
idm: self.idm.try_into()?,
879+
idm: InnerIDMConfig {
880+
users: users_to_inner(self.users)?,
881+
},
881882
})
882883
}
883884
}
@@ -931,7 +932,7 @@ impl From<InnerQueryConfig> for QueryConfig {
931932
async_insert_max_data_size: inner.async_insert_max_data_size,
932933
async_insert_busy_timeout: inner.async_insert_busy_timeout,
933934
async_insert_stale_timeout: inner.async_insert_stale_timeout,
934-
idm: inner.idm.into(),
935+
users: users_from_inner(inner.idm.users),
935936
}
936937
}
937938
}
@@ -1230,36 +1231,22 @@ impl Debug for MetaConfig {
12301231
}
12311232
}
12321233

1233-
#[derive(Clone, Debug, Default, PartialEq, Eq, Serialize, Deserialize)]
1234-
pub struct IDMConfig {
1235-
users: Vec<UserConfig>,
1236-
}
1237-
1238-
impl TryInto<InnerIDMConfig> for IDMConfig {
1239-
type Error = ErrorCode;
1240-
1241-
fn try_into(self) -> Result<InnerIDMConfig> {
1242-
let mut users = HashMap::new();
1243-
for c in self.users.into_iter() {
1244-
users.insert(c.name.clone(), c.auth.try_into()?);
1245-
}
1246-
Ok(InnerIDMConfig { users })
1247-
}
1234+
fn users_from_inner(inner: HashMap<String, AuthInfo>) -> Vec<UserConfig> {
1235+
inner
1236+
.into_iter()
1237+
.map(|(name, auth)| UserConfig {
1238+
name,
1239+
auth: auth.into(),
1240+
})
1241+
.collect()
12481242
}
12491243

1250-
impl From<InnerIDMConfig> for IDMConfig {
1251-
fn from(inner: InnerIDMConfig) -> Self {
1252-
Self {
1253-
users: inner
1254-
.users
1255-
.into_iter()
1256-
.map(|(name, auth)| UserConfig {
1257-
name,
1258-
auth: auth.into(),
1259-
})
1260-
.collect(),
1261-
}
1244+
fn users_to_inner(outer: Vec<UserConfig>) -> Result<HashMap<String, AuthInfo>> {
1245+
let mut inner = HashMap::new();
1246+
for c in outer.into_iter() {
1247+
inner.insert(c.name.clone(), c.auth.try_into()?);
12621248
}
1249+
Ok(inner)
12631250
}
12641251

12651252
#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]

src/query/service/tests/it/storages/system/configs_table.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ async fn test_configs_table() -> Result<()> {
104104
"| query | table_engine_memory_enabled | true | |",
105105
"| query | table_memory_cache_mb_size | 256 | |",
106106
"| query | tenant_id | test | |",
107+
"| query | users | | |",
107108
"| query | wait_timeout_mills | 5000 | |",
108109
"| storage | allow_insecure | false | |",
109110
"| storage | azblob.account_key | | |",
@@ -236,6 +237,7 @@ async fn test_configs_table_redact() -> Result<()> {
236237
"| query | table_engine_memory_enabled | true | |",
237238
"| query | table_memory_cache_mb_size | 256 | |",
238239
"| query | tenant_id | test | |",
240+
"| query | users | | |",
239241
"| query | wait_timeout_mills | 5000 | |",
240242
"| storage | allow_insecure | false | |",
241243
"| storage | azblob.account_key | | |",

0 commit comments

Comments
 (0)