From 7d3876dbe56448d973017ceb17aa3a62380662e7 Mon Sep 17 00:00:00 2001
From: Bugen Zhao <i@bugenzhao.com>
Date: Tue, 19 Sep 2023 16:26:23 +0800
Subject: [PATCH] add panic return

Signed-off-by: Bugen Zhao <i@bugenzhao.com>
---
 Cargo.lock                      | 4 ++--
 src/expr/Cargo.toml             | 2 +-
 src/tests/simulation/Cargo.toml | 2 +-
 src/udf/Cargo.toml              | 2 +-
 src/udf/src/external.rs         | 1 +
 5 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 4cb27cf9cd3ca..36a3068229fe5 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -1453,9 +1453,9 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "cfg-or-panic"
-version = "0.1.1"
+version = "0.2.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "cf85d5384815558275789d91d1895d1d9919a6e2534d6144650f036ac65691a6"
+checksum = "bc7cb2538d4ecc42b6c3b57a83094d8c69894e74468d18cd045a09fdea807358"
 dependencies = [
  "proc-macro2",
  "quote",
diff --git a/src/expr/Cargo.toml b/src/expr/Cargo.toml
index 91c68999fd6ac..eab539364b32a 100644
--- 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
index ab9201e6ddd3f..127d40855652d 100644
--- 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
index 2bdf12d44cfda..2d13f39bdddc4 100644
--- 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
index e9ff8652f31ec..08f5c35c21c1f 100644
--- 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,