ipc: Read only a single line on the client

Allow extensibility.
This commit is contained in:
Ivan Molodetskikh
2024-06-20 09:22:02 +03:00
parent 598b27f83c
commit 397b7e4bb9
2 changed files with 10 additions and 7 deletions
+8 -6
View File
@@ -1,7 +1,7 @@
//! Helper for blocking communication over the niri socket.
use std::env;
use std::io::{self, Read, Write};
use std::io::{self, BufRead, BufReader, Write};
use std::net::Shutdown;
use std::os::unix::net::UnixStream;
use std::path::Path;
@@ -50,14 +50,16 @@ impl Socket {
pub fn send(self, request: Request) -> io::Result<Reply> {
let Self { mut stream } = self;
let mut buf = serde_json::to_vec(&request).unwrap();
stream.write_all(&buf)?;
let mut buf = serde_json::to_string(&request).unwrap();
stream.write_all(buf.as_bytes())?;
stream.shutdown(Shutdown::Write)?;
buf.clear();
stream.read_to_end(&mut buf)?;
let mut reader = BufReader::new(stream);
let reply = serde_json::from_slice(&buf)?;
buf.clear();
reader.read_line(&mut buf)?;
let reply = serde_json::from_str(&buf)?;
Ok(reply)
}
}