问题所在：
1. 在主循环中，每次遇到 FA 标志时都创建了新的 AuxInfo ，且覆盖了之前的值
2. AuxInfo::parse 方法每次只处理一个字段，而不是累积所有字段
解决方案，我们有两种处理方式：
方案 1 - 修改 ParserFactory::parse 方法，使用 Vec 收集所有 AUX 信息，最后合并：
```rust
impl ParserFactory {
pub fn parse(rdb_context: &[u8]) -> Result<RDBInfo> {
let mut cursor = Cursor::new(rdb_context);
let base_info = BaseInfo::parse(&mut cursor)?;
let rdb_version: usize = base_info.rdb_version.parse().expect("Not a valid number");
let mut aux_infos = Vec::new(); // 存储所有 AUX 信息
let mut db_info = None;
if rdb_version < 7 {
aux_infos = Vec::new();
};
loop {
let mut flag_byte = [0;1];
if cursor.read_exact(&mut flag_byte).is_err() {
break
}
match flag_byte[0] {
FA => {
if let Some(aux_info) = AuxInfo::parse(&mut cursor)? {
aux_infos.push(aux_info);
}
}
FE => {
db_info = Some(DbInfo::parse(&mut cursor)?);
}
FF => {
println!("parse done.");
break
}
_ => {
continue;
}
}
}
// 合并所有 AUX 信息
let final_aux_info = aux_infos.into_iter().fold(
AuxInfo {
redis_server_version: String::new(),
used_mem: 0,
},
|mut acc, curr| {
if !curr.redis_server_version.is_empty() {
acc.redis_server_version = curr.redis_server_version;
}
if curr.used_mem != 0 {
acc.used_mem = curr.used_mem;
}
acc
}
);
let rdb_info = RDBInfo {
base_info,
aux_info: final_aux_info,
};
Ok(rdb_info)
}
}
```
方案 2 - 修改 AuxInfo::parse 方法，让它保持当前状态：
```rust
impl Parser for AuxInfo {
fn parse(cursor: &mut Cursor<&[u8]>) -> Result<Option<AuxInfo>> {
let aux_name = parser_aux_name(cursor)?;
// 根据字段名返回对应的 AuxInfo
let aux_info = match aux_name.as_str() {
"redis-ver" => {
Some(AuxInfo {
redis_server_version: parser_aux_value(cursor)?,
used_mem: 0,
})
}
"used-mem" => {
Some(AuxInfo {
redis_server_version: String::new(),
used_mem: parser_aux_value(cursor)?.parse::<usize>()
.expect("Failed to parse used-mem"),
})
}
_ => None
};
Ok(aux_info)
}
}
```
我建议使用方案 1 ，因为：
1. 更清晰地表达了数据收集的过程
2. 更容易扩展，如果将来需要添加新的 AUX 字段
3. 数据处理逻辑更集中，便于维护
关键改进点：
1. 使用 Vec 收集所有 AUX 信息
2. 使用 fold 方法合并所有 AUX 信息
3. 保留了非空/非零值，避免数据丢失
你需要注意的地方：
1. 确保 parser_aux_name 和 parser_aux_value 正确处理了数据
2. 考虑错误处理机制，可能需要添加更多的错误类型
3. 可以添加日志来跟踪解析过程
Select Language