mirror of
https://github.com/AppFlowy-IO/AppFlowy.git
synced 2025-07-23 17:11:23 +00:00
42 lines
1.3 KiB
Rust
42 lines
1.3 KiB
Rust
use std::fmt::Display;
|
|
|
|
pub enum StreamMessage {
|
|
MessageId(i64),
|
|
IndexStart,
|
|
IndexEnd,
|
|
Text(String),
|
|
OnData(String),
|
|
OnError(String),
|
|
Metadata(String),
|
|
Done,
|
|
StartIndexFile { file_name: String },
|
|
EndIndexFile { file_name: String },
|
|
IndexFileError { file_name: String },
|
|
}
|
|
|
|
impl Display for StreamMessage {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
match self {
|
|
StreamMessage::MessageId(message_id) => write!(f, "message_id:{}", message_id),
|
|
StreamMessage::IndexStart => write!(f, "index_start:"),
|
|
StreamMessage::IndexEnd => write!(f, "index_end"),
|
|
StreamMessage::Text(text) => {
|
|
write!(f, "data:{}", text)
|
|
},
|
|
StreamMessage::OnData(message) => write!(f, "data:{message}"),
|
|
StreamMessage::OnError(message) => write!(f, "error:{message}"),
|
|
StreamMessage::Done => write!(f, "done:"),
|
|
StreamMessage::Metadata(s) => write!(f, "metadata:{s}"),
|
|
StreamMessage::StartIndexFile { file_name } => {
|
|
write!(f, "start_index_file:{}", file_name)
|
|
},
|
|
StreamMessage::EndIndexFile { file_name } => {
|
|
write!(f, "end_index_file:{}", file_name)
|
|
},
|
|
StreamMessage::IndexFileError { file_name } => {
|
|
write!(f, "index_file_error:{}", file_name)
|
|
},
|
|
}
|
|
}
|
|
}
|