refactor for cooperative cancellation

Cargo.lock

@@ -4410,6 +4410,18 @@ version = "1.1.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "a2eb9349b6444b326872e140eb1cf5e7c522154d69e7a0ffb0fb81c06b37543f"
 
+[[package]]
+name = "stop-token"
+version = "0.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "af91f480ee899ab2d9f8435bfdfc14d08a5754bd9d3fef1f1a1c23336aad6c8b"
+dependencies = [
+ "async-channel",
+ "cfg-if 1.0.0",
+ "futures-core",
+ "pin-project-lite",
+]
+
 [[package]]
 name = "strsim"
 version = "0.10.0"
@@ -5209,6 +5221,7 @@ dependencies = [
  "simplelog",
  "socket2",
  "static_assertions",
+ "stop-token",
  "thiserror",
  "tracing",
  "tracing-error",