filip/Stalwart
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix queue event migration (closes #1833)

Browse Source
This commit is contained in:
mdecimus
2025-07-16 15:55:20 +02:00
parent 89b561b5ca
commit c6904e0f5d
4 changed files with 148 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

246
crates/migration/src/queue.rs
View File

@@ -7,7 +7,7 @@
use crate::LegacyBincode;
use common::{
Server,
config::smtp::queue::{QueueExpiry, QueueName},
config::smtp::queue::{DEFAULT_QUEUE_NAME, QueueExpiry, QueueName},
};
use smtp::queue::{
Error, ErrorDetails, HostResponse, Message, QueueId, QuotaKey, Recipient, Schedule, Status,
@@ -15,64 +15,22 @@ use smtp::queue::{
};
use std::net::{IpAddr, Ipv4Addr};
use store::{
IterateParams, Serialize, U64_LEN, ValueKey,
ahash::AHashSet,
IterateParams, SUBSPACE_QUEUE_EVENT, Serialize, U64_LEN, ValueKey,
ahash::AHashMap,
write::{
AlignedBytes, Archive, Archiver, BatchBuilder, QueueClass, ValueClass,
key::DeserializeBigEndian, now,
AlignedBytes, AnyClass, Archive, Archiver, BatchBuilder, QueueClass, ValueClass,
key::{DeserializeBigEndian, KeySerializer},
now,
},
};
use trc::AddContext;
use utils::BlobHash;
pub(crate) async fn migrate_queue_v011(server: &Server) -> trc::Result<()> {
let from_key = ValueKey::from(ValueClass::Queue(QueueClass::MessageEvent(
store::write::QueueEvent {
due: 0,
queue_id: 0,
queue_name: [0; 8],
},
)));
let to_key = ValueKey::from(ValueClass::Queue(QueueClass::MessageEvent(
store::write::QueueEvent {
due: u64::MAX,
queue_id: u64::MAX,
queue_name: [u8::MAX; 8],
},
)));
let mut queue_ids = AHashSet::new();
server
.store()
.iterate(
IterateParams::new(from_key, to_key).ascending().no_values(),
|key, _| {
queue_ids.insert(key.deserialize_be_u64(U64_LEN)?);
Ok(true)
},
)
.await
.caused_by(trc::location!())?;
let from_key = ValueKey::from(ValueClass::Queue(QueueClass::Message(0)));
let to_key = ValueKey::from(ValueClass::Queue(QueueClass::Message(u64::MAX)));
server
.store()
.iterate(
IterateParams::new(from_key, to_key).ascending().no_values(),
|key, _| {
queue_ids.insert(key.deserialize_be_u64(0)?);
Ok(true)
},
)
.await
.caused_by(trc::location!())?;
let mut count = 0;
let now = now();
for queue_id in queue_ids {
for (queue_id, due) in get_queue_events(server).await? {
match server
.store()
.get_value::<LegacyBincode<MessageV011>>(ValueKey::from(ValueClass::Queue(
@@ -82,12 +40,28 @@ pub(crate) async fn migrate_queue_v011(server: &Server) -> trc::Result<()> {
{
Ok(Some(bincoded)) => {
let mut batch = BatchBuilder::new();
batch.set(
ValueClass::Queue(QueueClass::Message(queue_id)),
Archiver::new(Message::from(bincoded.inner))
.serialize()
.caused_by(trc::location!())?,
);
let message = Message::from(bincoded.inner);
if let Some(due) = due {
batch.clear(ValueClass::Any(AnyClass {
subspace: SUBSPACE_QUEUE_EVENT,
key: KeySerializer::new(16).write(due).write(queue_id).finalize(),
}));
}
batch
.set(
ValueClass::Queue(QueueClass::MessageEvent(store::write::QueueEvent {
due: due.unwrap_or(now),
queue_id,
queue_name: DEFAULT_QUEUE_NAME.into_inner(),
})),
vec![],
)
.set(
ValueClass::Queue(QueueClass::Message(queue_id)),
Archiver::new(message)
.serialize()
.caused_by(trc::location!())?,
);
count += 1;
server
.store()
@@ -95,7 +69,20 @@ pub(crate) async fn migrate_queue_v011(server: &Server) -> trc::Result<()> {
.await
.caused_by(trc::location!())?;
}
Ok(None) => (),
Ok(None) => {
if let Some(due) = due {
let mut batch = BatchBuilder::new();
batch.clear(ValueClass::Any(AnyClass {
subspace: SUBSPACE_QUEUE_EVENT,
key: KeySerializer::new(16).write(due).write(queue_id).finalize(),
}));
server
.store()
.write(batch.build_all())
.await
.caused_by(trc::location!())?;
}
}
Err(err) => {
if server
.store()
@@ -124,53 +111,10 @@ pub(crate) async fn migrate_queue_v011(server: &Server) -> trc::Result<()> {
}
pub(crate) async fn migrate_queue_v012(server: &Server) -> trc::Result<()> {
let from_key = ValueKey::from(ValueClass::Queue(QueueClass::MessageEvent(
store::write::QueueEvent {
due: 0,
queue_id: 0,
queue_name: [0; 8],
},
)));
let to_key = ValueKey::from(ValueClass::Queue(QueueClass::MessageEvent(
store::write::QueueEvent {
due: u64::MAX,
queue_id: u64::MAX,
queue_name: [u8::MAX; 8],
},
)));
let mut queue_ids = AHashSet::new();
server
.store()
.iterate(
IterateParams::new(from_key, to_key).ascending().no_values(),
|key, _| {
queue_ids.insert(key.deserialize_be_u64(U64_LEN)?);
Ok(true)
},
)
.await
.caused_by(trc::location!())?;
let from_key = ValueKey::from(ValueClass::Queue(QueueClass::Message(0)));
let to_key = ValueKey::from(ValueClass::Queue(QueueClass::Message(u64::MAX)));
server
.store()
.iterate(
IterateParams::new(from_key, to_key).ascending().no_values(),
|key, _| {
queue_ids.insert(key.deserialize_be_u64(0)?);
Ok(true)
},
)
.await
.caused_by(trc::location!())?;
let mut count = 0;
let now = now();
for queue_id in queue_ids {
for (queue_id, due) in get_queue_events(server).await? {
match server
.store()
.get_value::<Archive<AlignedBytes>>(ValueKey::from(ValueClass::Queue(
@@ -185,13 +129,29 @@ pub(crate) async fn migrate_queue_v012(server: &Server) -> trc::Result<()> {
}
}) {
Ok(Some(archive)) => {
let message = Message::from(archive);
let mut batch = BatchBuilder::new();
batch.set(
ValueClass::Queue(QueueClass::Message(queue_id)),
Archiver::new(Message::from(archive))
.serialize()
.caused_by(trc::location!())?,
);
if let Some(due) = due {
batch.clear(ValueClass::Any(AnyClass {
subspace: SUBSPACE_QUEUE_EVENT,
key: KeySerializer::new(16).write(due).write(queue_id).finalize(),
}));
}
batch
.set(
ValueClass::Queue(QueueClass::MessageEvent(store::write::QueueEvent {
due: due.unwrap_or(now),
queue_id,
queue_name: DEFAULT_QUEUE_NAME.into_inner(),
})),
vec![],
)
.set(
ValueClass::Queue(QueueClass::Message(queue_id)),
Archiver::new(message)
.serialize()
.caused_by(trc::location!())?,
);
count += 1;
server
.store()
@@ -199,7 +159,20 @@ pub(crate) async fn migrate_queue_v012(server: &Server) -> trc::Result<()> {
.await
.caused_by(trc::location!())?;
}
Ok(None) => (),
Ok(None) => {
if let Some(due) = due {
let mut batch = BatchBuilder::new();
batch.clear(ValueClass::Any(AnyClass {
subspace: SUBSPACE_QUEUE_EVENT,
key: KeySerializer::new(16).write(due).write(queue_id).finalize(),
}));
server
.store()
.write(batch.build_all())
.await
.caused_by(trc::location!())?;
}
}
Err(err) => {
if server
.store()
@@ -234,6 +207,61 @@ pub(crate) async fn migrate_queue_v012(server: &Server) -> trc::Result<()> {
Ok(())
}
async fn get_queue_events(server: &Server) -> trc::Result<AHashMap<u64, Option<u64>>> {
let from_key = ValueKey::from(ValueClass::Queue(QueueClass::MessageEvent(
store::write::QueueEvent {
due: 0,
queue_id: 0,
queue_name: [0; 8],
},
)));
let to_key = ValueKey::from(ValueClass::Queue(QueueClass::MessageEvent(
store::write::QueueEvent {
due: u64::MAX,
queue_id: u64::MAX,
queue_name: [u8::MAX; 8],
},
)));
let mut queue_ids: AHashMap<u64, Option<u64>> = AHashMap::new();
server
.store()
.iterate(
IterateParams::new(from_key, to_key).ascending().no_values(),
|key, _| {
queue_ids.insert(
key.deserialize_be_u64(U64_LEN)?,
Some(key.deserialize_be_u64(0)?),
);
Ok(true)
},
)
.await
.caused_by(trc::location!())?;
let from_key = ValueKey::from(ValueClass::Queue(QueueClass::Message(0)));
let to_key = ValueKey::from(ValueClass::Queue(QueueClass::Message(u64::MAX)));
server
.store()
.iterate(
IterateParams::new(from_key, to_key).ascending().no_values(),
|key, _| {
let queue_id = key.deserialize_be_u64(0)?;
if !queue_ids.contains_key(&queue_id) {
queue_ids.insert(queue_id, None);
}
Ok(true)
},
)
.await
.caused_by(trc::location!())?;
Ok(queue_ids)
}
impl<SIZE, IDX> From<LegacyMessage<SIZE, IDX>> for Message
where
SIZE: AsU64,