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

refactor admin outputs to asyncwrite

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2025-01-04 16:57:07 +00:00
parent abf33013e3
commit 8141ca3444
24 changed files with 877 additions and 227 deletions
Download patch file Download diff file Expand all files Collapse all files

82
src/admin/query/resolver.rs
View file

@ -28,56 +28,66 @@ async fn destinations_cache(
) -> Result<RoomMessageEventContent> {
use service::resolver::cache::CachedDest;
writeln!(self, "| Server Name | Destination | Hostname | Expires |").await?;
writeln!(self, "| ----------- | ----------- | -------- | ------- |").await?;
let mut out = String::new();
writeln!(out, "| Server Name | Destination | Hostname | Expires |")?;
writeln!(out, "| ----------- | ----------- | -------- | ------- |")?;
let row = |(name, &CachedDest { ref dest, ref host, expire })| {
let expire = time::format(expire, "%+");
writeln!(out, "| {name} | {dest} | {host} | {expire} |").expect("wrote line");
};
{
let map = self
.services
.resolver
.cache
.destinations
.read()
.expect("locked");
let map = self
.services
.resolver
.cache
.destinations
.read()
.expect("locked");
for (name, &CachedDest { ref dest, ref host, expire }) in map.iter() {
if let Some(server_name) = server_name.as_ref() {
if name != server_name {
continue;
}
}
if let Some(server_name) = server_name.as_ref() {
map.get_key_value(server_name).map(row);
} else {
map.iter().for_each(row);
let expire = time::format(expire, "%+");
writeln!(out, "| {name} | {dest} | {host} | {expire} |")?;
}
}
Ok(RoomMessageEventContent::notice_markdown(out))
self.write_str(out.as_str()).await?;
Ok(RoomMessageEventContent::notice_plain(""))
}
#[admin_command]
async fn overrides_cache(&self, server_name: Option<String>) -> Result<RoomMessageEventContent> {
use service::resolver::cache::CachedOverride;
writeln!(self, "| Server Name | IP | Port | Expires |").await?;
writeln!(self, "| ----------- | --- | ----:| ------- |").await?;
let mut out = String::new();
writeln!(out, "| Server Name | IP | Port | Expires |")?;
writeln!(out, "| ----------- | --- | ----:| ------- |")?;
let row = |(name, &CachedOverride { ref ips, port, expire })| {
let expire = time::format(expire, "%+");
writeln!(out, "| {name} | {ips:?} | {port} | {expire} |").expect("wrote line");
};
{
let map = self
.services
.resolver
.cache
.overrides
.read()
.expect("locked");
let map = self
.services
.resolver
.cache
.overrides
.read()
.expect("locked");
for (name, &CachedOverride { ref ips, port, expire }) in map.iter() {
if let Some(server_name) = server_name.as_ref() {
if name != server_name {
continue;
}
}
if let Some(server_name) = server_name.as_ref() {
map.get_key_value(server_name).map(row);
} else {
map.iter().for_each(row);
let expire = time::format(expire, "%+");
writeln!(out, "| {name} | {ips:?} | {port} | {expire} |")?;
}
}
Ok(RoomMessageEventContent::notice_markdown(out))
self.write_str(out.as_str()).await?;
Ok(RoomMessageEventContent::notice_plain(""))
}