|
1 | | -//! 提供可定制实现的 `print!`、`println!` 和 `log::Log`。 |
| 1 | +//! 提供可定制实现的 `print!`、`println!` 和 `log::Log`。 |
2 | 2 |
|
3 | 3 | #![no_std] |
4 | 4 | #![deny(warnings, missing_docs)] |
@@ -31,6 +31,66 @@ pub trait Console: Sync { |
31 | 31 | /// 库找到输出的方法：保存一个对象引用，这是一种单例。 |
32 | 32 | static CONSOLE: Once<&'static dyn Console> = Once::new(); |
33 | 33 |
|
| 34 | +/// 打印缓冲区大小。 |
| 35 | +const BUFFER_SIZE: usize = 64; |
| 36 | + |
| 37 | +/// 打印缓冲区，用于收集格式化输出后一次性输出，避免抢占导致输出交错。 |
| 38 | +struct PrintBuffer { |
| 39 | + buffer: [u8; BUFFER_SIZE], |
| 40 | + pos: usize, |
| 41 | +} |
| 42 | + |
| 43 | +impl PrintBuffer { |
| 44 | + /// 创建一个新的打印缓冲区。 |
| 45 | + const fn new() -> Self { |
| 46 | + Self { |
| 47 | + buffer: [0u8; BUFFER_SIZE], |
| 48 | + pos: 0, |
| 49 | + } |
| 50 | + } |
| 51 | + |
| 52 | + /// 刷新缓冲区，将内容输出到控制台。 |
| 53 | + fn flush(&mut self) { |
| 54 | + if self.pos > 0 { |
| 55 | + if let Some(console) = CONSOLE.get() { |
| 56 | + // SAFETY: buffer 中的内容是从 str 写入的，保证是有效的 UTF-8 |
| 57 | + let s = unsafe { core::str::from_utf8_unchecked(&self.buffer[..self.pos]) }; |
| 58 | + console.put_str(s); |
| 59 | + } |
| 60 | + self.pos = 0; |
| 61 | + } |
| 62 | + } |
| 63 | + |
| 64 | + /// 写入字符串到缓冲区，必要时刷新。 |
| 65 | + fn write(&mut self, s: &str) { |
| 66 | + let bytes = s.as_bytes(); |
| 67 | + let mut offset = 0; |
| 68 | + |
| 69 | + while offset < bytes.len() { |
| 70 | + let remaining_buffer = BUFFER_SIZE - self.pos; |
| 71 | + let remaining_input = bytes.len() - offset; |
| 72 | + let to_copy = remaining_buffer.min(remaining_input); |
| 73 | + |
| 74 | + self.buffer[self.pos..self.pos + to_copy] |
| 75 | + .copy_from_slice(&bytes[offset..offset + to_copy]); |
| 76 | + self.pos += to_copy; |
| 77 | + offset += to_copy; |
| 78 | + |
| 79 | + // 缓冲区满了，刷新 |
| 80 | + if self.pos == BUFFER_SIZE { |
| 81 | + self.flush(); |
| 82 | + } |
| 83 | + } |
| 84 | + } |
| 85 | +} |
| 86 | + |
| 87 | +impl Write for PrintBuffer { |
| 88 | + fn write_str(&mut self, s: &str) -> fmt::Result { |
| 89 | + self.write(s); |
| 90 | + Ok(()) |
| 91 | + } |
| 92 | +} |
| 93 | + |
34 | 94 | /// 用户调用这个函数设置输出的方法。 |
35 | 95 | pub fn init_console(console: &'static dyn Console) { |
36 | 96 | CONSOLE.call_once(|| console); |
@@ -65,10 +125,13 @@ pub fn test_log() { |
65 | 125 | /// 打印。 |
66 | 126 | /// |
67 | 127 | /// 给宏用的，用户不会直接调它。 |
| 128 | +/// 使用栈上缓冲区收集格式化输出，然后一次性输出到控制台。 |
68 | 129 | #[doc(hidden)] |
69 | 130 | #[inline] |
70 | 131 | pub fn _print(args: fmt::Arguments) { |
71 | | - Logger.write_fmt(args).unwrap(); |
| 132 | + let mut buffer = PrintBuffer::new(); |
| 133 | + buffer.write_fmt(args).unwrap(); |
| 134 | + buffer.flush(); |
72 | 135 | } |
73 | 136 |
|
74 | 137 | /// 格式化打印。 |
|
0 commit comments