Serde:加速自定义枚举反序列化

Question

我的程序解析了足够大的json文档(30MB(，在CPU速度较慢的机器上需要70ms，我想加快进程，我发现 27% 的解析发生在我的foo_document_type_deserialize中，是否可以改进此功能，也许有没有办法跳过这里的String分配：let s = String::deserialize(deserializer)?;？

我完全确定表示枚举值的字符串不包含像 b f n r t " \ 这样的特殊 json 字符，因此使用未转义的字符串应该是安全的。

use serde::{Deserialize, Deserializer};
#[derive(Deserialize, Debug, Clone)]
#[serde(rename_all = "camelCase")]
pub struct FooDocument {
    // other fields...
    #[serde(rename = "type")]
    #[serde(deserialize_with = "foo_document_type_deserialize")]
    doc_type: FooDocumentType,
}
fn foo_document_type_deserialize<'de, D>(deserializer: D) -> Result<FooDocumentType, D::Error>
where
    D: Deserializer<'de>,
{
    use self::FooDocumentType::*;
    let s = String::deserialize(deserializer)?;
    match s.as_str() {
        "tir lim bom bom" => Ok(Var1),
        "hgga;hghau" => Ok(Var2),
        "hgueoqtyhit4t" => Ok(Var3),
        "Text" | "Type not detected" | "---" => Ok(Unknown),
        _ => Err(serde::de::Error::custom(format!(
            "Unsupported foo document type '{}'",
            s
        ))),
    }
}
#[derive(Debug, Clone, Copy)]
pub enum FooDocumentType {
    Unknown,
    Var1,
    Var2,
    Var3,
}

Answer 1

您编写的自定义 impl 采用 serde_derive 可以生成的形式：

#[derive(Deserialize, Debug)]
pub enum FooDocumentType {
    #[serde(rename = "Text", alias = "Type not detected", alias = "---")]
    Unknown,
    #[serde(rename = "tir lim bom bom")]
    Var1,
    #[serde(rename = "hgga;hghau")]
    Var2,
    #[serde(rename = "hgueoqtyhit4t")]
    Var3,
}

当我测量以下内容时，生成的派生代码不会分配内存，并且在快速微基准测试中比您的代码快约 2×：

serde_json::from_str::<FooDocument>(r#"{"type":"hgga;hghau"}"#).unwrap()