io/uring/Operation: add method ReplaceUring()

3 files changed, 27 insertions, 0 deletions

diff --git a/src/io/uring/CancellableOperation.hxx b/src/io/uring/CancellableOperation.hxx
index b55af5867..e7b71958c 100644
--- a/src/io/uring/CancellableOperation.hxx
+++ b/src/io/uring/CancellableOperation.hxx
@@ -67,6 +67,16 @@ public:
 		// TODO: io_uring_prep_cancel()
 	}
 
+	void Replace(Operation &old_operation,
+		     Operation &new_operation) noexcept {
+		assert(operation == &old_operation);
+		assert(old_operation.cancellable == this);
+
+		old_operation.cancellable = nullptr;
+		operation = &new_operation;
+		new_operation.cancellable = this;
+	}
+
 	void OnUringCompletion(int res) noexcept {
 		if (operation == nullptr)
 			return;
diff --git a/src/io/uring/Operation.cxx b/src/io/uring/Operation.cxx
index 120434e97..8fadebcf4 100644
--- a/src/io/uring/Operation.cxx
+++ b/src/io/uring/Operation.cxx
@@ -33,6 +33,7 @@
 #include "Operation.hxx"
 #include "CancellableOperation.hxx"
 
+#include <cassert>
 #include <utility>
 
 namespace Uring {
@@ -46,4 +47,14 @@ Operation::CancelUring() noexcept
 	std::exchange(cancellable, nullptr)->Cancel(*this);
 }
 
+void
+Operation::ReplaceUring(Operation &new_operation) noexcept
+{
+	assert(IsUringPending());
+
+	cancellable->Replace(*this, new_operation);
+
+	assert(cancellable == nullptr);
+}
+
 } // namespace Uring
diff --git a/src/io/uring/Operation.hxx b/src/io/uring/Operation.hxx
index 8f753b16f..a8c3b4adc 100644
--- a/src/io/uring/Operation.hxx
+++ b/src/io/uring/Operation.hxx
@@ -63,6 +63,12 @@ public:
 	void CancelUring() noexcept;
 
 	/**
+	 * Replace this pending operation with a new one.  This method
+	 * is only legal if IsUringPending().
+	 */
+	void ReplaceUring(Operation &new_operation) noexcept;
+
+	/**
 	 * This method is called when the operation completes.
 	 *
 	 * @param res the result code; the meaning is specific to the