Início / coding / Perguntas / 79336704
Accepted
Yuki
Asked: 2025-01-08 00:31:22 +0800 CST

&Opção<NomeCabeçalho> para &NomeCabeçalho

  • 772

Estou criando um proxy usando axum para aceitar requisições do navegador e reqwest para enviar requisições para o host de destino em Rust. Então, tenho que mudar a requisição axum para reqwest e remover alguns cabeçalhos como host, origin, etc. Esta é minha função remove.

pub fn remove_headers(request: &reqwest::Request, keys_to_remove: Vec<String>) -> HeaderMap {
    let mut cloned_headers = request.headers().clone();

    // Convert keys_to_remove into a set for fast lookups
    let keys_set: HashSet<String> = keys_to_remove.into_iter().collect();

    // Create a new HeaderMap and only insert headers that are not in the keys_to_remove list
    let filtered_headers = cloned_headers
        .into_iter()
        .filter(|(key, _)| {
            // Dereference the key from &Option<HeaderName> to HeaderName and compare with the set
            if let Some(header_name) = key {
                !keys_set.contains(&header_name.as_str().to_string())
            } else {
                key.is_some()
            }
        })
        .collect::<HeaderMap>(); // Collect back into a HeaderMap

    filtered_headers
}

Espero que o tipo de chave seja &HeaderName, mas agora &Option. Então filtered_headers não é convertido para HeaderMap.

rust

1 respostas

  1. Best Answer

    Além da dica postada nos comentários (usando filter_map), você pode economizar desempenho não copiando todas as coleções/strings envolvidas:

    • Não clone todos os cabeçalhos, itere sobre referências e clone valores-chave quando não filtrados
    • Não converta o nome do cabeçalho para String, você pode fazer a pesquisa usando&str
    • Não converta o vetor em um hashmap sempre (se possível)
    fn remove_headers(request: &Request, keys_to_remove: &HashSet<String>) -> HeaderMap {
    request
        .headers()
        .iter()
        .filter_map(|(key, value)| {
            if !keys_to_remove.contains(key.as_str()) {
                Some((key.clone(), value.clone()))
            } else {
                None
            }
        })
        .collect()
}

    Observe que se você puder mutar request, você pode simplesmente usar HeaderMap::remove:

    fn remove_headers(request: &mut Request, keys_to_remove: &Vec<String>) {
    for key in keys_to_remove {
        request.headers_mut().remove(key);
    }
}
    • 0

relate perguntas