mirror of
https://github.com/farion1231/cc-switch.git
synced 2026-07-24 21:30:17 +08:00
eaf83f4fbe
* fix(proxy): handle SSE fields with or without spaces * refactor(proxy): deduplicate SSE field parsing
32 lines
891 B
Rust
32 lines
891 B
Rust
#[inline]
|
|
pub(crate) fn strip_sse_field<'a>(line: &'a str, field: &str) -> Option<&'a str> {
|
|
line.strip_prefix(&format!("{field}: "))
|
|
.or_else(|| line.strip_prefix(&format!("{field}:")))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::strip_sse_field;
|
|
|
|
#[test]
|
|
fn strip_sse_field_accepts_optional_space() {
|
|
assert_eq!(
|
|
strip_sse_field("data: {\"ok\":true}", "data"),
|
|
Some("{\"ok\":true}")
|
|
);
|
|
assert_eq!(
|
|
strip_sse_field("data:{\"ok\":true}", "data"),
|
|
Some("{\"ok\":true}")
|
|
);
|
|
assert_eq!(
|
|
strip_sse_field("event: message_start", "event"),
|
|
Some("message_start")
|
|
);
|
|
assert_eq!(
|
|
strip_sse_field("event:message_start", "event"),
|
|
Some("message_start")
|
|
);
|
|
assert_eq!(strip_sse_field("id:1", "data"), None);
|
|
}
|
|
}
|