magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

integrate trim-on-park memory reclamation to runtime

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2025-01-16 19:08:54 +00:00
parent 7a8ca8842a
commit 77d8e26efe
2 changed files with 33 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/main/runtime.rs
View file

@ -9,6 +9,7 @@ use std::{
};
use conduwuit::{
result::LogErr,
utils::sys::compute::{nth_core_available, set_affinity},
Result,
};
@ -22,12 +23,17 @@ const WORKER_KEEPALIVE: u64 = 36;
const MAX_BLOCKING_THREADS: usize = 1024;
static WORKER_AFFINITY: OnceLock<bool> = OnceLock::new();
static GC_ON_PARK: OnceLock<Option<bool>> = OnceLock::new();
pub(super) fn new(args: &Args) -> Result<tokio::runtime::Runtime> {
WORKER_AFFINITY
.set(args.worker_affinity)
.expect("set WORKER_AFFINITY from program argument");
GC_ON_PARK
.set(args.gc_on_park)
.expect("set GC_ON_PARK from program argument");
let mut builder = Builder::new_multi_thread();
builder
.enable_io()
@ -138,7 +144,21 @@ fn thread_unpark() {}
name = %thread::current().name().unwrap_or("None"),
),
)]
fn thread_park() {}
fn thread_park() {
match GC_ON_PARK
.get()
.as_ref()
.expect("GC_ON_PARK initialized by runtime::new()")
{
| Some(true) | None if cfg!(feature = "jemalloc_conf") => gc_on_park(),
| _ => (),
}
}
fn gc_on_park() {
#[cfg(feature = "jemalloc")]
conduwuit::alloc::je::this_thread::decay().log_err().ok();
}
#[cfg(tokio_unstable)]
#[tracing::instrument(