Skip to content

Commit 79ee41e

Browse files
committed
read api key from header
1 parent 7807b20 commit 79ee41e

1 file changed

Lines changed: 9 additions & 1 deletion

File tree

be/src/api.rs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -454,7 +454,15 @@ impl FromRequestParts<Config> for Arc<gafe::AccountLimit> {
454454
let params = parts.uri.query().unwrap_or_default();
455455
let decoded =
456456
serde_urlencoded::from_str::<HashMap<String, String>>(params).unwrap_or_default();
457-
let client_id = decoded.get("api-key").cloned().unwrap_or_default();
457+
458+
let header_key = parts
459+
.headers
460+
.get("api-key")
461+
.and_then(|v| v.to_str().ok())
462+
.map(String::from);
463+
let url_key = decoded.get("api-key").cloned();
464+
465+
let client_id = url_key.or(header_key).unwrap_or_default();
458466
match config.account_limits.lock().unwrap().get(&client_id) {
459467
Some(limit) => Ok(limit.clone()),
460468
None => Ok(config.free_limit.clone()),

0 commit comments

Comments
 (0)