mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-27 16:12:56 +00:00
b38e841f2d
The tokio futures added some overhead, so switch to plain non-blocking I/O with poll(). In a simple pgbench test on my laptop (select-only queries, scale-factor 1 `pgbench -P1 -T50 -S`), this gives about 10% improvement, from about 4300 TPS to 4800 TPS.
18 lines
586 B
Rust
18 lines
586 B
Rust
use nix::fcntl::{fcntl, OFlag, F_GETFL, F_SETFL};
|
|
use std::os::unix::io::RawFd;
|
|
|
|
/// Put a file descriptor into non-blocking mode
|
|
pub fn set_nonblock(fd: RawFd) -> Result<(), std::io::Error> {
|
|
let bits = fcntl(fd, F_GETFL)?;
|
|
|
|
// Safety: If F_GETFL returns some unknown bits, they should be valid
|
|
// for passing back to F_SETFL, too. If we left them out, the F_SETFL
|
|
// would effectively clear them, which is not what we want.
|
|
let mut flags = unsafe { OFlag::from_bits_unchecked(bits) };
|
|
flags |= OFlag::O_NONBLOCK;
|
|
|
|
fcntl(fd, F_SETFL(flags))?;
|
|
|
|
Ok(())
|
|
}
|