Add unit test cases and benchmark for push_command()

Add unit test cases and benchmark for push_command().

Signed-off-by: Liu Jiang <[email protected]>

Author: jiangliu

io-uring-bench/src/nop.rs

@@ -41,5 +41,33 @@ fn bench_normal(c: &mut Criterion) {
     });
 }
 
-criterion_group!(squeue, bench_normal);
+fn bench_prepare(c: &mut Criterion) {
+    let mut io_uring = IoUring::new(16).unwrap();
+
+    c.bench_function("prepare", |b| {
+        b.iter(|| {
+            let mut queue = TaskQueue(128);
+            let nop = black_box(opcode::Nop::new());
+
+            while queue.want() {
+                {
+                    let mut sq = io_uring.submission();
+                    while queue.want() {
+                        unsafe {
+                            match sq.push_command(&nop, None) {
+                                Ok(_) => queue.pop(),
+                                Err(_) => break,
+                            }
+                        }
+                    }
+                }
+
+                io_uring.submit_and_wait(16).unwrap();
+
+                io_uring.completion().map(black_box).for_each(drop);
+            }
+        });
+    });
+}
+criterion_group!(squeue, bench_normal, bench_prepare);
 criterion_main!(squeue);

io-uring-test/src/main.rs

@@ -27,6 +27,7 @@ fn main() -> anyhow::Result<()> {
     };
 
     tests::queue::test_nop(&mut ring, &test)?;
+    tests::queue::test_nop_prepare(&mut ring, &test)?;
     tests::queue::test_queue_split(&mut ring, &test)?;
 
     #[cfg(feature = "unstable")]

io-uring-test/src/tests/queue.rs

@@ -1,4 +1,5 @@
 use crate::Test;
+use io_uring::squeue::OptionValues;
 use io_uring::{opcode, IoUring};
 
 pub fn test_nop(ring: &mut IoUring, test: &Test) -> anyhow::Result<()> {
@@ -26,6 +27,32 @@ pub fn test_nop(ring: &mut IoUring, test: &Test) -> anyhow::Result<()> {
     Ok(())
 }
 
+pub fn test_nop_prepare(ring: &mut IoUring, test: &Test) -> anyhow::Result<()> {
+    require! {
+        test;
+    }
+
+    println!("test nop with prepare");
+
+    let nop = opcode::Nop::new();
+    let opt = OptionValues::default().user_data(0x42);
+
+    unsafe {
+        let mut queue = ring.submission();
+        queue.push_command(&nop, Some(&opt)).expect("queue is full");
+    }
+
+    ring.submit_and_wait(1)?;
+
+    let cqes = ring.completion().collect::<Vec<_>>();
+
+    assert_eq!(cqes.len(), 1);
+    assert_eq!(cqes[0].user_data(), 0x42);
+    assert_eq!(cqes[0].result(), 0);
+
+    Ok(())
+}
+
 #[cfg(feature = "unstable")]
 pub fn test_batch(ring: &mut IoUring, test: &Test) -> anyhow::Result<()> {
     use std::mem::MaybeUninit;