Refactor: Tight up unix::Client::new impl

Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
rust-lang · Jun 29, 2022 · ff1c741 · ff1c741
1 parent 0345ff6
commit ff1c741
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/unix.rs b/src/unix.rs
@@ -35,13 +35,11 @@ impl Client {
         // 128 cores.
         const BUFFER: [u8; 128] = [b'|'; 128];
 
-        while limit >= BUFFER.len() {
-            (&client.write).write_all(&BUFFER)?;
-            limit -= BUFFER.len();
-        }
+        while limit > 0 {
+            let n = limit.min(BUFFER.len());
 
-        if limit > 0 {
-            (&client.write).write_all(&BUFFER[..limit])?;
+            (&client.write).write_all(&BUFFER[..n])?;
+            limit -= n;
         }
 
         Ok(client)