oss/continuwuity
oss/continuwuity
1
0
Fork 0
mirror of https://forgejo.ellis.link/continuwuation/continuwuity.git synced 2025-07-27 18:28:31 +00:00
Code Wiki Activity

feat(policy-server): Optimise policy server lookups

Browse source
This commit is contained in:
nexy7574 2025-07-19 23:50:32 +01:00
parent cc9202b0c4
commit d7fa624fd2
No known key found for this signature in database
GPG key ID: 0FA334385D0B689F
2 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/service/rooms/event_handler/call_policyserv.rs
View file

@ -11,6 +11,10 @@ use ruma::{
#[implement(super::Service)] #[implement(super::Service)]
#[tracing::instrument(skip_all, level = "debug")] #[tracing::instrument(skip_all, level = "debug")]
pub async fn policyserv_check(&self, pdu: &PduEvent, room_id: &RoomId) -> Result { pub async fn policyserv_check(&self, pdu: &PduEvent, room_id: &RoomId) -> Result {
if pdu.event_type().to_owned() == StateEventType::RoomPolicy.into() {
debug!("Skipping spam check for policy server meta-event in room {room_id}");
return Ok(());
}
let Ok(policyserver) = self let Ok(policyserver) = self
.services .services
.state_accessor .state_accessor
@ -28,6 +32,14 @@ pub async fn policyserv_check(&self, pdu: &PduEvent, room_id: &RoomId) -> Result
return Ok(()); return Ok(());
}, },
}; };
if via.is_empty() {
debug!("Policy server is empty for room {room_id}, skipping spam check");
return Ok(());
}
if !self.services.state_cache.server_in_room(via, room_id).await {
debug!("Policy server {via} is not in the room {room_id}, skipping spam check");
return Ok(());
}
let outgoing = self let outgoing = self
.services .services
.sending .sending

2
src/service/rooms/event_handler/mod.rs
View file

@ -38,6 +38,7 @@ struct Services {
server_keys: Dep<server_keys::Service>, server_keys: Dep<server_keys::Service>,
short: Dep<rooms::short::Service>, short: Dep<rooms::short::Service>,
state: Dep<rooms::state::Service>, state: Dep<rooms::state::Service>,
state_cache: Dep<rooms::state_cache::Service>,
state_accessor: Dep<rooms::state_accessor::Service>, state_accessor: Dep<rooms::state_accessor::Service>,
state_compressor: Dep<rooms::state_compressor::Service>, state_compressor: Dep<rooms::state_compressor::Service>,
timeline: Dep<rooms::timeline::Service>, timeline: Dep<rooms::timeline::Service>,
@ -63,6 +64,7 @@ impl crate::Service for Service {
pdu_metadata: args.depend::<rooms::pdu_metadata::Service>("rooms::pdu_metadata"), pdu_metadata: args.depend::<rooms::pdu_metadata::Service>("rooms::pdu_metadata"),
short: args.depend::<rooms::short::Service>("rooms::short"), short: args.depend::<rooms::short::Service>("rooms::short"),
state: args.depend::<rooms::state::Service>("rooms::state"), state: args.depend::<rooms::state::Service>("rooms::state"),
state_cache: args.depend::<rooms::state_cache::Service>("rooms::state_cache"),
state_accessor: args state_accessor: args
.depend::<rooms::state_accessor::Service>("rooms::state_accessor"), .depend::<rooms::state_accessor::Service>("rooms::state_accessor"),
state_compressor: args state_compressor: args