diff --git a/src/runtime/driver/op/mod.rs b/src/runtime/driver/op/mod.rs index 32ba3e7a..8d3559b0 100644 --- a/src/runtime/driver/op/mod.rs +++ b/src/runtime/driver/op/mod.rs @@ -4,6 +4,7 @@ use std::marker::PhantomData; use std::pin::Pin; use std::task::{Context, Poll, Waker}; +use io_uring::squeue::Flags; use io_uring::{cqueue, squeue}; mod slab_list; @@ -37,6 +38,12 @@ impl> UnsubmittedOneshot { } } + /// Set the SQE's flags. + pub fn set_flags(mut self, flags: Flags) -> Self { + self.sqe = self.sqe.flags(flags); + self + } + /// Submit an operation to the driver for batched entry to the kernel. pub fn submit(self) -> InFlightOneshot { let handle = CONTEXT