rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-01-08 14:22:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add user_info extension to prom_store handler (#2212)

Browse Source 
chore: add user_info extention to prom_store auth
This commit is contained in:
shuiyisong
2023-08-21 12:55:34 +08:00
committed by GitHub
parent 033b650d0d
commit c5f507c20e
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/servers/src/http/prom_store.rs
View File

@@ -13,9 +13,11 @@
// limitations under the License.
use api::prom_store::remote::{ReadRequest, WriteRequest};
use auth::UserInfoRef;
use axum::extract::{Query, RawBody, State};
use axum::http::{header, StatusCode};
use axum::response::IntoResponse;
use axum::Extension;
use common_catalog::consts::DEFAULT_SCHEMA_NAME;
use common_telemetry::timer;
use hyper::Body;
@@ -46,6 +48,7 @@ impl Default for DatabaseQuery {
pub async fn remote_write(
State(handler): State<PromStoreProtocolHandlerRef>,
Query(params): Query<DatabaseQuery>,
user_info: Extension<UserInfoRef>,
RawBody(body): RawBody,
) -> Result<(StatusCode, ())> {
let request = decode_remote_write_request(body).await?;
@@ -59,6 +62,7 @@ pub async fn remote_write(
);
let ctx = QueryContext::with_db_name(params.db.as_ref());
ctx.set_current_user(Some(user_info.0));
handler.write(request, ctx).await?;
Ok((StatusCode::NO_CONTENT, ()))
}
@@ -80,6 +84,7 @@ impl IntoResponse for PromStoreResponse {
pub async fn remote_read(
State(handler): State<PromStoreProtocolHandlerRef>,
Query(params): Query<DatabaseQuery>,
user_info: Extension<UserInfoRef>,
RawBody(body): RawBody,
) -> Result<PromStoreResponse> {
let request = decode_remote_read_request(body).await?;
@@ -93,6 +98,7 @@ pub async fn remote_read(
);
let ctx = QueryContext::with_db_name(params.db.as_ref());
ctx.set_current_user(Some(user_info.0));
handler.read(request, ctx).await
}