astelm/telemt
1
0
Fork 0
mirror of https://github.com/telemt/telemt.git synced 2026-04-18 11:04:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

ME Hardswap Generation stability

Browse Source 
Co-Authored-By: brekotis <93345790+brekotis@users.noreply.github.com>
This commit is contained in:
Alexey
2026-03-02 00:39:18 +03:00
parent ac453638b8
commit 49f4a7bb22
6 changed files with 165 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/transport/middle_proxy/pool_reinit.rs
View File

@@ -266,8 +266,26 @@ impl MePool {
}
let previous_generation = self.current_generation();
let generation = self.generation.fetch_add(1, Ordering::Relaxed) + 1;
let hardswap = self.hardswap.load(Ordering::Relaxed);
let generation = if hardswap {
let pending_generation = self.pending_hardswap_generation.load(Ordering::Relaxed);
if pending_generation != 0 && pending_generation >= previous_generation {
debug!(
previous_generation,
generation = pending_generation,
"ME hardswap continues with pending generation"
);
pending_generation
} else {
let next_generation = self.generation.fetch_add(1, Ordering::Relaxed) + 1;
self.pending_hardswap_generation
.store(next_generation, Ordering::Relaxed);
next_generation
}
} else {
self.pending_hardswap_generation.store(0, Ordering::Relaxed);
self.generation.fetch_add(1, Ordering::Relaxed) + 1
};
if hardswap {
self.warmup_generation_for_all_dcs(rng, generation, &desired_by_dc)
@@ -354,6 +372,9 @@ impl MePool {
drop(writers);
if stale_writer_ids.is_empty() {
if hardswap {
self.pending_hardswap_generation.store(0, Ordering::Relaxed);
}
debug!("ME reinit cycle completed with no stale writers");
return;
}
@@ -375,6 +396,9 @@ impl MePool {
self.mark_writer_draining_with_timeout(writer_id, drain_timeout, !hardswap)
.await;
}
if hardswap {
self.pending_hardswap_generation.store(0, Ordering::Relaxed);
}
}
pub async fn zero_downtime_reinit_periodic(self: &Arc<Self>, rng: &SecureRandom) {