i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-02 01:10:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

OAuth refresh 隧道代理统一启用 follow-redirects

Browse Source
This commit is contained in:
fawney19
2026-04-28 17:47:28 +08:00
parent 70f747d406
commit 9194f78e56
2 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
apps/aether-gateway/src/state/oauth.rs
View File

@@ -1279,17 +1279,10 @@ impl AppState {
.await;
let proxy_is_tunnel = local_oauth_proxy_is_tunnel(proxy_snapshot.as_ref());
let mut headers = request.headers.clone();
if !proxy_is_tunnel {
headers.insert(
EXECUTION_REQUEST_FOLLOW_REDIRECTS_HEADER.to_string(),
"true".to_string(),
);
} else {
headers.insert(
EXECUTION_REQUEST_HTTP1_ONLY_HEADER.to_string(),
"true".to_string(),
);
}
headers.insert(
EXECUTION_REQUEST_FOLLOW_REDIRECTS_HEADER.to_string(),
"true".to_string(),
);
let plan = ExecutionPlan {
request_id: request.request_id.to_string(),
candidate_id: None,

13
apps/aether-gateway/src/tests/control/admin/oauth.rs
View File

@@ -4571,8 +4571,7 @@ async fn gateway_refreshes_admin_provider_oauth_key_locally_via_execution_runtim
}
#[tokio::test]
async fn gateway_refreshes_admin_provider_oauth_key_tunnel_proxy_without_follow_redirects_like_master(
) {
async fn gateway_refreshes_admin_provider_oauth_key_tunnel_proxy_with_direct_refresh_controls() {
let execution_plans = Arc::new(Mutex::new(Vec::<ExecutionPlan>::new()));
let execution_plans_clone = Arc::clone(&execution_plans);
let execution_runtime = Router::new().route(
@@ -4706,15 +4705,15 @@ async fn gateway_refreshes_admin_provider_oauth_key_tunnel_proxy_without_follow_
assert_eq!(
refresh_plan
.headers
.get(EXECUTION_REQUEST_FOLLOW_REDIRECTS_HEADER),
None
.get(EXECUTION_REQUEST_FOLLOW_REDIRECTS_HEADER)
.map(String::as_str),
Some("true")
);
assert_eq!(
refresh_plan
.headers
.get(EXECUTION_REQUEST_HTTP1_ONLY_HEADER)
.map(String::as_str),
Some("true")
.get(EXECUTION_REQUEST_HTTP1_ONLY_HEADER),
None
);
gateway_handle.abort();