File size: 3,831 Bytes
a21c316 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | // 错误分类模块 - 将底层错误转换为用户友好的消息
/// 分类流式响应错误并返回错误类型、英文消息和 i18n key
///
/// 返回值: (错误类型, 英文错误消息, i18n_key)
/// - 错误类型: 用于日志和错误码
/// - 英文消息: fallback 消息,供非浏览器客户端使用
/// - i18n_key: 前端翻译键,供浏览器客户端本地化
/// 分类流式响应错误并返回错误类型、英文消息和 i18n key
///
/// 返回值: (错误类型, 英文错误消息, i18n_key)
pub fn classify_stream_error<E: std::fmt::Display>(error: &E) -> (&'static str, &'static str, &'static str) {
let error_str = error.to_string().to_lowercase();
if error_str.contains("timeout") || error_str.contains("deadline") {
(
"timeout_error",
"Request timeout, please check your network connection",
"errors.stream.timeout_error"
)
} else if error_str.contains("connection") || error_str.contains("connect") || error_str.contains("dns") {
(
"connection_error",
"Connection failed, please check your network or proxy settings",
"errors.stream.connection_error"
)
} else if error_str.contains("decode") || error_str.contains("parse") {
(
"decode_error",
"Network unstable, data transmission interrupted. Try: 1) Check network 2) Switch proxy 3) Retry",
"errors.stream.decode_error"
)
} else if error_str.contains("stream") || error_str.contains("body") {
(
"stream_error",
"Stream transmission error, please retry later",
"errors.stream.stream_error"
)
} else {
(
"unknown_error",
"Unknown error occurred",
"errors.stream.unknown_error"
)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_classify_timeout_error() {
// 使用简单的字符串错误进行模拟测试
let error = "Connection timed out after 30s";
let (error_type, message, i18n_key) = classify_stream_error(&error);
assert_eq!(error_type, "timeout_error");
assert!(message.contains("timeout"));
assert_eq!(i18n_key, "errors.stream.timeout_error");
}
#[test]
fn test_error_message_format() {
// 测试错误消息格式
// 模拟一个 DNS 错误
let error = "error trying to connect: dns error: failed to lookup address information";
let (error_type, message, i18n_key) = classify_stream_error(&error);
// 错误类型应该是已知的类型之一
assert!(
error_type == "timeout_error" ||
error_type == "connection_error" ||
error_type == "decode_error" ||
error_type == "stream_error" ||
error_type == "unknown_error"
);
// 消息不应该为空
assert!(!message.is_empty());
// i18n_key 应该以 errors.stream. 开头
assert!(i18n_key.starts_with("errors.stream."));
}
#[test]
fn test_i18n_keys_format() {
// 验证所有错误类型都有正确的 i18n_key 格式
let test_cases = vec![
("timeout_error", "errors.stream.timeout_error"),
("connection_error", "errors.stream.connection_error"),
("decode_error", "errors.stream.decode_error"),
("stream_error", "errors.stream.stream_error"),
("unknown_error", "errors.stream.unknown_error"),
];
// 这里我们只验证 i18n_key 格式
for (expected_type, expected_key) in test_cases {
assert_eq!(format!("errors.stream.{}", expected_type), expected_key);
}
}
}
|