Import upstream v0.16.22, stripped

Upstream commit: 474dd0229cb20cf513036619781ed97bd8073c3f Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 117 in 50 files Dangling module declarations removed: 5 Cargo edits turning enterprise off: 14 Verification: clean Enterprise feature gates left for rebuilt features: 19 in 18 files Produced by tools/fork/strip.py. The full report is in docs/fork/strip-reports/ on main.
2026-09-18 10:21:56 -07:00
commit 7dae9b29fd
1650 changed files with 485521 additions and 0 deletions
@@ -0,0 +1,52 @@
+/*
+ * SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
+ */
+
+use crate::task_manager::*;
+
+pub trait TaskLockManager: Sync + Send {
+    fn try_lock_task(&self, task: u64) -> impl Future<Output = bool> + Send;
+    fn remove_index_lock(&self, id: u64) -> impl Future<Output = ()> + Send;
+}
+
+impl TaskLockManager for Server {
+    async fn try_lock_task(&self, id: u64) -> bool {
+        match self
+            .in_memory_store()
+            .try_lock(KV_LOCK_TASK, &id.to_be_bytes(), DEFAULT_LOCK_EXPIRY)
+            .await
+        {
+            Ok(result) => {
+                if !result {
+                    trc::event!(
+                        TaskManager(TaskManagerEvent::TaskLocked),
+                        Id = id,
+                        Details = "Task details not available",
+                    );
+                }
+                result
+            }
+            Err(err) => {
+                trc::error!(err.id(id).details("Failed to lock task"));
+
+                false
+            }
+        }
+    }
+
+    async fn remove_index_lock(&self, id: u64) {
+        if let Err(err) = self
+            .in_memory_store()
+            .remove_lock(KV_LOCK_TASK, &id.to_be_bytes())
+            .await
+        {
+            trc::error!(
+                err.details("Failed to unlock task")
+                    .ctx(trc::Key::Id, id)
+                    .caused_by(trc::location!())
+            );
+        }
+    }
+}