refactor: allow gateway to use "select 1" to verify token in request …

…header.
databendlabs · Sep 25, 2024 · 98637be · 98637be
1 parent e665183
commit 98637be
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/query/service/src/servers/http/middleware.rs b/src/query/service/src/servers/http/middleware.rs
@@ -89,7 +89,12 @@ impl EndpointKind {
             EndpointKind::Verify => Ok(None),
             EndpointKind::Refresh => Ok(Some(TokenType::Refresh)),
             EndpointKind::StartQuery | EndpointKind::PollQuery | EndpointKind::Logout => {
-                Ok(Some(TokenType::Session))
+                if GlobalConfig::instance().query.management_mode {
+                    // allow gateway to use "select 1" to verify token in request header.
+                    Ok(None)
+                } else {
+                    Ok(Some(TokenType::Session))
+                }
             }
             _ => Err(ErrorCode::AuthenticateFailure(format!(
                 "should not use databend token for {self:?}",