mercuryworkshop-mirror/epoxy-tls
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/epoxy-tls.git synced 2025-05-13 06:20:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

switch to jemalloc and add memory stats

Browse source
This commit is contained in:
Toshit Chawda 2024-08-24 21:53:24 -07:00
parent 0f05588b8f
commit 6f0e1e7feb
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
3 changed files with 81 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

41
server/src/main.rs
View file

@ -40,9 +40,45 @@ fn format_stream_type(stream_type: StreamType) -> &'static str {
}
}
fn generate_stats() -> Result<String, std::fmt::Error> {
fn generate_stats() -> anyhow::Result<String> {
let mut out = String::new();
let len = CLIENTS.len();
use tikv_jemalloc_ctl::stats::{active, allocated, mapped, metadata, resident, retained};
tikv_jemalloc_ctl::epoch::advance()?;
writeln!(&mut out, "Memory usage:")?;
writeln!(
&mut out,
"\tActive: {:?} MiB",
active::read()? as f64 / (1024 * 1024) as f64
)?;
writeln!(
&mut out,
"\tAllocated: {:?} MiB",
allocated::read()? as f64 / (1024 * 1024) as f64
)?;
writeln!(
&mut out,
"\tMapped: {:?} MiB",
mapped::read()? as f64 / (1024 * 1024) as f64
)?;
writeln!(
&mut out,
"\tMetadata: {:?} MiB",
metadata::read()? as f64 / (1024 * 1024) as f64
)?;
writeln!(
&mut out,
"\tResident: {:?} MiB",
resident::read()? as f64 / (1024 * 1024) as f64
)?;
writeln!(
&mut out,
"\tRetained: {:?} MiB",
retained::read()? as f64 / (1024 * 1024) as f64
)?;
writeln!(
&mut out,
"{} clients connected{}",
@ -108,6 +144,9 @@ fn handle_stream(stream: ServerRouteResult, id: String) {
});
}
#[global_allocator]
static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
#[tokio::main(flavor = "multi_thread")]
async fn main() -> anyhow::Result<()> {
if CLI.default_config {