mirror of
https://github.com/chatmail/core.git
synced 2026-05-08 09:26:29 +03:00
fix: create_send_msg_jobs: Don't pass message ID to smtp
Follow-up to 970222f. When resending to new broadcast members, we shouldn't update the message db
entry and also share the message id with existing SMTP jobs:
- Otherwise the behavior would depend on timings which isn't good.
- If the message sending failed, resending it to a new member shalln't make it `OutDelivered`.
This commit is contained in:
36
src/chat.rs
36
src/chat.rs
@@ -2753,8 +2753,8 @@ async fn prepare_send_msg(
|
|||||||
chat_id.unarchive_if_not_muted(context, msg.state).await?;
|
chat_id.unarchive_if_not_muted(context, msg.state).await?;
|
||||||
}
|
}
|
||||||
chat.prepare_msg_raw(context, msg, update_msg_id).await?;
|
chat.prepare_msg_raw(context, msg, update_msg_id).await?;
|
||||||
|
let resend_to_new = false;
|
||||||
let row_ids = create_send_msg_jobs(context, msg)
|
let row_ids = create_send_msg_jobs(context, msg, resend_to_new)
|
||||||
.await
|
.await
|
||||||
.context("Failed to create send jobs")?;
|
.context("Failed to create send jobs")?;
|
||||||
if !row_ids.is_empty() {
|
if !row_ids.is_empty() {
|
||||||
@@ -2824,7 +2824,11 @@ async fn render_mime_message_and_pre_message(
|
|||||||
/// Returns row ids if `smtp` table jobs were created or an empty `Vec` otherwise.
|
/// Returns row ids if `smtp` table jobs were created or an empty `Vec` otherwise.
|
||||||
///
|
///
|
||||||
/// The caller has to interrupt SMTP loop or otherwise process new rows.
|
/// The caller has to interrupt SMTP loop or otherwise process new rows.
|
||||||
pub(crate) async fn create_send_msg_jobs(context: &Context, msg: &mut Message) -> Result<Vec<i64>> {
|
async fn create_send_msg_jobs(
|
||||||
|
context: &Context,
|
||||||
|
msg: &mut Message,
|
||||||
|
resend_to_new: bool,
|
||||||
|
) -> Result<Vec<i64>> {
|
||||||
let cmd = msg.param.get_cmd();
|
let cmd = msg.param.get_cmd();
|
||||||
if cmd == SystemMessage::GroupNameChanged || cmd == SystemMessage::GroupDescriptionChanged {
|
if cmd == SystemMessage::GroupNameChanged || cmd == SystemMessage::GroupDescriptionChanged {
|
||||||
msg.chat_id
|
msg.chat_id
|
||||||
@@ -2940,12 +2944,13 @@ pub(crate) async fn create_send_msg_jobs(context: &Context, msg: &mut Message) -
|
|||||||
msg.param.remove(Param::GuaranteeE2ee);
|
msg.param.remove(Param::GuaranteeE2ee);
|
||||||
}
|
}
|
||||||
msg.subject.clone_from(&rendered_msg.subject);
|
msg.subject.clone_from(&rendered_msg.subject);
|
||||||
|
if !resend_to_new {
|
||||||
// Sort the message to the bottom. Employ `msgs_index7` to compute `timestamp`.
|
// Sort the message to the bottom. Employ `msgs_index7` to compute `timestamp`.
|
||||||
context
|
context
|
||||||
.sql
|
.sql
|
||||||
.execute(
|
.execute(
|
||||||
"
|
"
|
||||||
UPDATE msgs SET
|
UPDATE msgs SET
|
||||||
timestamp=(
|
timestamp=(
|
||||||
SELECT MAX(timestamp) FROM msgs INDEXED BY msgs_index7 WHERE
|
SELECT MAX(timestamp) FROM msgs INDEXED BY msgs_index7 WHERE
|
||||||
-- From `InFresh` to `OutMdnRcvd` inclusive except `OutDraft`.
|
-- From `InFresh` to `OutMdnRcvd` inclusive except `OutDraft`.
|
||||||
@@ -2955,7 +2960,7 @@ UPDATE msgs SET
|
|||||||
id<=?
|
id<=?
|
||||||
),
|
),
|
||||||
pre_rfc724_mid=?, subject=?, param=?
|
pre_rfc724_mid=?, subject=?, param=?
|
||||||
WHERE id=?
|
WHERE id=?
|
||||||
",
|
",
|
||||||
(
|
(
|
||||||
msg.chat_id,
|
msg.chat_id,
|
||||||
@@ -2967,8 +2972,13 @@ WHERE id=?
|
|||||||
),
|
),
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
let chunk_size = context.get_max_smtp_rcpt_to().await?;
|
let chunk_size = context.get_max_smtp_rcpt_to().await?;
|
||||||
|
let msg_id = match resend_to_new {
|
||||||
|
true => MsgId::new(u32::MAX),
|
||||||
|
false => msg.id,
|
||||||
|
};
|
||||||
let trans_fn = |t: &mut rusqlite::Transaction| {
|
let trans_fn = |t: &mut rusqlite::Transaction| {
|
||||||
let mut row_ids = Vec::<i64>::new();
|
let mut row_ids = Vec::<i64>::new();
|
||||||
|
|
||||||
@@ -2989,7 +2999,7 @@ WHERE id=?
|
|||||||
&pre_msg.rfc724_mid,
|
&pre_msg.rfc724_mid,
|
||||||
&recipients_chunk,
|
&recipients_chunk,
|
||||||
&pre_msg.message,
|
&pre_msg.message,
|
||||||
msg.id,
|
msg_id,
|
||||||
))?;
|
))?;
|
||||||
row_ids.push(row_id.try_into()?);
|
row_ids.push(row_id.try_into()?);
|
||||||
}
|
}
|
||||||
@@ -2997,7 +3007,7 @@ WHERE id=?
|
|||||||
&rendered_msg.rfc724_mid,
|
&rendered_msg.rfc724_mid,
|
||||||
&recipients_chunk,
|
&recipients_chunk,
|
||||||
&rendered_msg.message,
|
&rendered_msg.message,
|
||||||
msg.id,
|
msg_id,
|
||||||
))?;
|
))?;
|
||||||
row_ids.push(row_id.try_into()?);
|
row_ids.push(row_id.try_into()?);
|
||||||
}
|
}
|
||||||
@@ -4609,7 +4619,11 @@ pub async fn forward_msgs_2ctx(
|
|||||||
chat.prepare_msg_raw(ctx_dst, &mut msg, None).await?;
|
chat.prepare_msg_raw(ctx_dst, &mut msg, None).await?;
|
||||||
|
|
||||||
curr_timestamp += 1;
|
curr_timestamp += 1;
|
||||||
if !create_send_msg_jobs(ctx_dst, &mut msg).await?.is_empty() {
|
let resend_to_new = false;
|
||||||
|
if !create_send_msg_jobs(ctx_dst, &mut msg, resend_to_new)
|
||||||
|
.await?
|
||||||
|
.is_empty()
|
||||||
|
{
|
||||||
ctx_dst.scheduler.interrupt_smtp().await;
|
ctx_dst.scheduler.interrupt_smtp().await;
|
||||||
}
|
}
|
||||||
created_msgs.push(msg.id);
|
created_msgs.push(msg.id);
|
||||||
@@ -4767,7 +4781,11 @@ pub(crate) async fn resend_msgs_ex(
|
|||||||
if let Some(to_fingerprint) = &to_fingerprint {
|
if let Some(to_fingerprint) = &to_fingerprint {
|
||||||
msg.param.set(Param::Arg4, to_fingerprint.clone());
|
msg.param.set(Param::Arg4, to_fingerprint.clone());
|
||||||
}
|
}
|
||||||
if create_send_msg_jobs(context, &mut msg).await?.is_empty() {
|
let resend_to_new = to_fingerprint.is_some();
|
||||||
|
if create_send_msg_jobs(context, &mut msg, resend_to_new)
|
||||||
|
.await?
|
||||||
|
.is_empty()
|
||||||
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -3049,6 +3049,31 @@ async fn test_broadcast_resend_to_new_member() -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
||||||
|
async fn test_broadcast_resend_failed_msg_to_new_member() -> Result<()> {
|
||||||
|
let mut tcm = TestContextManager::new();
|
||||||
|
let alice = &tcm.alice().await;
|
||||||
|
let bob = &tcm.bob().await;
|
||||||
|
let fiona = &tcm.fiona().await;
|
||||||
|
let alice_bc_id = create_broadcast(alice, "bc".to_string()).await?;
|
||||||
|
let qr = get_securejoin_qr(alice, Some(alice_bc_id)).await.unwrap();
|
||||||
|
|
||||||
|
tcm.exec_securejoin_qr(bob, alice, &qr).await;
|
||||||
|
let alice_msg_id = alice.send_text(alice_bc_id, "text").await.sender_msg_id;
|
||||||
|
let mut msg = Message::load_from_db(alice, alice_msg_id).await?;
|
||||||
|
message::set_msg_failed(alice, &mut msg, "error").await?;
|
||||||
|
let fiona_bc_id = tcm.exec_securejoin_qr(fiona, alice, &qr).await;
|
||||||
|
let resent_msg = alice.pop_sent_msg().await;
|
||||||
|
let fiona_msg = fiona.recv_msg(&resent_msg).await;
|
||||||
|
assert_eq!(fiona_msg.chat_id, fiona_bc_id);
|
||||||
|
assert_eq!(fiona_msg.text, "text");
|
||||||
|
assert_eq!(
|
||||||
|
alice_msg_id.get_state(alice).await?,
|
||||||
|
MessageState::OutFailed
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
/// - Alice has multiple devices
|
/// - Alice has multiple devices
|
||||||
/// - Alice creates a broadcast and sends a message into it
|
/// - Alice creates a broadcast and sends a message into it
|
||||||
/// - Alice's second device sees the broadcast
|
/// - Alice's second device sees the broadcast
|
||||||
|
|||||||
Reference in New Issue
Block a user