add panic return

Signed-off-by: Bugen Zhao <[email protected]>
risingwavelabs · Sep 19, 2023 · 7d3876d · 7d3876d
1 parent c1738d6
commit 7d3876d
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/expr/Cargo.toml b/src/expr/Cargo.toml
@@ -22,7 +22,7 @@ arrow-schema = { workspace = true }
 async-trait = "0.1"
 auto_enums = "0.8"
 await-tree = { workspace = true }
-cfg-or-panic = "0.1"
+cfg-or-panic = "0.2"
 chrono = { version = "0.4", default-features = false, features = ["clock", "std"] }
 chrono-tz = { version = "0.8", features = ["case-insensitive"] }
 ctor = "0.2"

diff --git a/src/tests/simulation/Cargo.toml b/src/tests/simulation/Cargo.toml
@@ -14,7 +14,7 @@ normal = ["serde"]
 anyhow = "1.0"
 async-trait = "0.1"
 aws-sdk-s3 = { version = "0.2", package = "madsim-aws-sdk-s3" }
-cfg-or-panic = "0.1"
+cfg-or-panic = "0.2"
 clap = { version = "4", features = ["derive"] }
 console = "0.15"
 etcd-client = { workspace = true }

diff --git a/src/udf/Cargo.toml b/src/udf/Cargo.toml
@@ -15,7 +15,7 @@ arrow-array = { workspace = true }
 arrow-flight = { workspace = true }
 arrow-schema = { workspace = true }
 arrow-select = { workspace = true }
-cfg-or-panic = "0.1"
+cfg-or-panic = "0.2"
 futures-util = "0.3.28"
 static_assertions = "1"
 thiserror = "1"

diff --git a/src/udf/src/external.rs b/src/udf/src/external.rs
@@ -100,6 +100,7 @@ impl ArrowFlightUdfClient {
     }
 
     /// Call a function with streaming input and output.
+    #[panic_return = "Result<stream::Empty<_>>"]
     pub async fn call_stream(
         &self,
         id: &str,