Skip to content

Commit 60c1471

Browse files
committed
fix(query): reduce task internal session overhead
1 parent 80d13e2 commit 60c1471

2 files changed

Lines changed: 6 additions & 9 deletions

File tree

src/query/service/src/task/service.rs

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,6 @@ use crate::meta_client_error;
8686
use crate::meta_service_error;
8787
use crate::schedulers::ServiceQueryExecutor;
8888
use crate::sessions::QueryContext;
89-
use crate::sessions::TableContextCluster;
9089
use crate::sessions::TableContextSettings;
9190
use crate::task::meta::TaskMetaHandle;
9291
use crate::task::session::create_session;
@@ -222,13 +221,11 @@ impl TaskService {
222221
..
223222
}) = task_message.warehouse_options()
224223
{
225-
if warehouse
226-
!= &self
227-
.create_context(None)
228-
.await?
229-
.get_cluster()
230-
.get_warehouse_id()?
231-
{
224+
let current_warehouse = ClusterDiscovery::instance()
225+
.single_node_cluster(&GlobalConfig::instance())
226+
.await?
227+
.get_warehouse_id()?;
228+
if warehouse != &current_warehouse {
232229
continue;
233230
}
234231
}

src/query/service/src/task/session.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ pub async fn create_session(
4242
restricted_role: Option<String>,
4343
) -> Result<Arc<Session>> {
4444
let session_manager = SessionManager::instance();
45-
let session = session_manager.create_session(SessionType::MySQL).await?;
45+
let session = session_manager.create_session(SessionType::Dummy).await?;
4646
let session = session_manager.register_session(session)?;
4747
session.set_authed_user(user, restricted_role).await?;
4848
Ok(session)

0 commit comments

Comments
 (0)