Skip to content

Commit a6bfc7f

Browse files
committed
fix(api): add middleware to redis interface
1 parent 94797a9 commit a6bfc7f

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

netmito/src/api/mod.rs

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ use crate::{
3030
config::{InfraPool, REDIS_CONNECTION_INFO},
3131
error::ApiError,
3232
schema::RedisConnectionInfo,
33-
service::auth::{user_auth_with_name_middleware, AuthUser},
33+
service::auth::{user_auth_middleware, user_auth_with_name_middleware, AuthUser},
3434
};
3535

3636
pub fn router(st: InfraPool, cancel_token: CancellationToken) -> Router {
@@ -49,7 +49,13 @@ pub fn router(st: InfraPool, cancel_token: CancellationToken) -> Router {
4949
get(|| async { (StatusCode::OK, Json(json!({"status": "ok"}))) }),
5050
)
5151
.route("/login", post(users::user_login))
52-
.route("/redis", get(query_redis_connection_info))
52+
.route(
53+
"/redis",
54+
get(query_redis_connection_info).layer(middleware::from_fn_with_state(
55+
st.clone(),
56+
user_auth_middleware,
57+
)),
58+
)
5359
.nest("/users", users::users_router(st.clone()))
5460
.nest("/admin", admin::admin_router(st.clone(), cancel_token))
5561
.nest("/groups", groups::groups_router(st.clone()))
@@ -73,7 +79,13 @@ pub fn router(st: InfraPool, cancel_token: CancellationToken) -> Router {
7379
get(|| async { (StatusCode::OK, Json(json!({"status": "ok"}))) }),
7480
)
7581
.route("/login", post(users::user_login))
76-
.route("/redis", get(query_redis_connection_info))
82+
.route(
83+
"/redis",
84+
get(query_redis_connection_info).layer(middleware::from_fn_with_state(
85+
st.clone(),
86+
user_auth_middleware,
87+
)),
88+
)
7789
.nest("/users", users::users_router(st.clone()))
7890
.nest("/admin", admin::admin_router(st.clone(), cancel_token))
7991
.nest("/groups", groups::groups_router(st.clone()))

0 commit comments

Comments
 (0)