backups/grapevine
1
0
Fork 0
mirror of https://gitlab.computer.surgery/matrix/grapevine.git synced 2025-12-17 07:41:23 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

sending.rs: add RequestData to replace big tuple

Browse source
This commit is contained in:
Lambda 2024-05-22 18:40:44 +00:00
parent e294543ddb
commit 9961f1465f
1 changed files with 47 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

72
src/service/sending.rs
View file

@ -101,16 +101,19 @@ impl RequestKey {
}
}
pub(crate) struct RequestData {
outgoing_kind: OutgoingKind,
event_type: SendingEventType,
key: RequestKey,
}
pub(crate) struct Service {
db: &'static dyn Data,
/// The state for a given state hash.
pub(super) maximum_requests: Arc<Semaphore>,
pub(crate) sender:
mpsc::UnboundedSender<(OutgoingKind, SendingEventType, RequestKey)>,
receiver: Mutex<
mpsc::UnboundedReceiver<(OutgoingKind, SendingEventType, RequestKey)>,
>,
pub(crate) sender: mpsc::UnboundedSender<RequestData>,
receiver: Mutex<mpsc::UnboundedReceiver<RequestData>>,
}
#[derive(Debug)]
@ -202,12 +205,10 @@ impl Service {
{
futures.push(Self::handle_events(kind, events));
},
Some((outgoing_kind, event, key)) = receiver.recv() =>
Some(data) = receiver.recv() =>
if let Some(HandlerInputs { kind, events }) =
self.handle_receiver(
outgoing_kind,
event,
key,
data,
&mut current_transaction_status,
)
{
@ -288,21 +289,23 @@ impl Service {
}
#[tracing::instrument(
skip(self, event, key, current_transaction_status),
skip(self, event_type, key, current_transaction_status),
fields(
current_status = ?current_transaction_status.get(&outgoing_kind),
),
)]
fn handle_receiver(
&self,
outgoing_kind: OutgoingKind,
event: SendingEventType,
key: RequestKey,
RequestData {
outgoing_kind,
event_type,
key,
}: RequestData,
current_transaction_status: &mut TransactionStatusMap,
) -> Option<HandlerInputs> {
if let Ok(Some(events)) = self.select_events(
&outgoing_kind,
vec![(event, key)],
vec![(event_type, key)],
current_transaction_status,
) {
Some(HandlerInputs {
@ -531,11 +534,15 @@ impl Service {
pushkey: String,
) -> Result<()> {
let outgoing_kind = OutgoingKind::Push(user.to_owned(), pushkey);
let event = SendingEventType::Pdu(pdu_id.to_owned());
let event_type = SendingEventType::Pdu(pdu_id.to_owned());
let keys =
self.db.queue_requests(&[(&outgoing_kind, event.clone())])?;
self.db.queue_requests(&[(&outgoing_kind, event_type.clone())])?;
self.sender
.send((outgoing_kind, event, keys.into_iter().next().unwrap()))
.send(RequestData {
outgoing_kind,
event_type,
key: keys.into_iter().next().unwrap(),
})
.unwrap();
Ok(())
@ -559,8 +566,15 @@ impl Service {
let keys = self.db.queue_requests(
&requests.iter().map(|(o, e)| (o, e.clone())).collect::<Vec<_>>(),
)?;
for ((outgoing_kind, event), key) in requests.into_iter().zip(keys) {
self.sender.send((outgoing_kind.clone(), event, key)).unwrap();
for ((outgoing_kind, event_type), key) in requests.into_iter().zip(keys)
{
self.sender
.send(RequestData {
outgoing_kind: outgoing_kind.clone(),
event_type,
key,
})
.unwrap();
}
Ok(())
@ -574,11 +588,15 @@ impl Service {
id: u64,
) -> Result<()> {
let outgoing_kind = OutgoingKind::Normal(server.to_owned());
let event = SendingEventType::Edu(serialized);
let event_type = SendingEventType::Edu(serialized);
let keys =
self.db.queue_requests(&[(&outgoing_kind, event.clone())])?;
self.db.queue_requests(&[(&outgoing_kind, event_type.clone())])?;
self.sender
.send((outgoing_kind, event, keys.into_iter().next().unwrap()))
.send(RequestData {
outgoing_kind,
event_type,
key: keys.into_iter().next().unwrap(),
})
.unwrap();
Ok(())
@ -591,11 +609,15 @@ impl Service {
pdu_id: Vec<u8>,
) -> Result<()> {
let outgoing_kind = OutgoingKind::Appservice(appservice_id);
let event = SendingEventType::Pdu(pdu_id);
let event_type = SendingEventType::Pdu(pdu_id);
let keys =
self.db.queue_requests(&[(&outgoing_kind, event.clone())])?;
self.db.queue_requests(&[(&outgoing_kind, event_type.clone())])?;
self.sender
.send((outgoing_kind, event, keys.into_iter().next().unwrap()))
.send(RequestData {
outgoing_kind,
event_type,
key: keys.into_iter().next().unwrap(),
})
.unwrap();
Ok(())