test: add bench

Author: Berrysoft

Cargo.toml

@@ -46,11 +46,23 @@ futures-util = { workspace = true, optional = true }
 compio = { workspace = true, default-features = false, features = [
     "fs",
     "net",
+    "criterion",
 ] }
 
-tokio = { workspace = true, features = ["rt", "macros", "net", "io-util"] }
+tokio = { workspace = true, features = [
+    "rt",
+    "macros",
+    "fs",
+    "net",
+    "io-util",
+] }
 futures-executor = { workspace = true }
 
+criterion = { version = "0.8.2", features = ["async_tokio"] }
+futures-util = { workspace = true }
+rand = "0.10.1"
+tempfile = "3.27.0"
+
 [features]
 io-uring = ["compio/io-uring"]
 polling = ["compio/polling"]
@@ -61,3 +73,13 @@ enable_log = ["compio-log/enable_log"]
 [[test]]
 name = "net"
 required-features = ["tokio"]
+
+[[bench]]
+name = "net"
+required-features = ["tokio"]
+harness = false
+
+[[bench]]
+name = "fs"
+required-features = ["tokio"]
+harness = false

benches/compat/mod.rs

@@ -0,0 +1,31 @@
+use compio_compat::{RuntimeCompat, TokioAdapter};
+use criterion::async_executor::AsyncExecutor;
+
+pub struct CompioInTokio {
+    truntime: tokio::runtime::Runtime,
+    cruntime: RuntimeCompat<TokioAdapter>,
+}
+
+impl Default for CompioInTokio {
+    fn default() -> Self {
+        let truntime = tokio::runtime::Builder::new_current_thread()
+            .enable_all()
+            .build()
+            .unwrap();
+        let cruntime =
+            RuntimeCompat::<TokioAdapter>::new(compio::runtime::Runtime::new().unwrap()).unwrap();
+        Self { truntime, cruntime }
+    }
+}
+
+impl AsyncExecutor for CompioInTokio {
+    fn block_on<T>(&self, future: impl Future<Output = T>) -> T {
+        (&self).block_on(future)
+    }
+}
+
+impl AsyncExecutor for &CompioInTokio {
+    fn block_on<T>(&self, future: impl Future<Output = T>) -> T {
+        self.truntime.block_on(self.cruntime.execute(future))
+    }
+}