Fix flight-sql

apache · Nov 9, 2022 · f02d62c · f02d62c
1 parent 86927b3
commit f02d62c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/arrow-flight/src/sql/mod.rs b/arrow-flight/src/sql/mod.rs
@@ -122,18 +122,18 @@ pub trait ProstAnyExt {
     ///
     /// * `Ok(None)` when message type mismatch
     /// * `Err` when parse failed
-    fn unpack<M: ProstMessageExt>(&self) -> ArrowResult<Option<M>>;
+    fn unpack<M: ProstMessageExt>(&self) -> Result<Option<M>, ArrowError>;
 
     /// Pack any message into `prost_types::Any` value.
-    fn pack<M: ProstMessageExt>(message: &M) -> ArrowResult<prost_types::Any>;
+    fn pack<M: ProstMessageExt>(message: &M) -> Result<prost_types::Any, ArrowError>;
 }
 
 impl ProstAnyExt for prost_types::Any {
     fn is<M: ProstMessageExt>(&self) -> bool {
         M::type_url() == self.type_url
     }
 
-    fn unpack<M: ProstMessageExt>(&self) -> ArrowResult<Option<M>> {
+    fn unpack<M: ProstMessageExt>(&self) -> Result<Option<M>, ArrowError> {
         if !self.is::<M>() {
             return Ok(None);
         }
@@ -143,7 +143,7 @@ impl ProstAnyExt for prost_types::Any {
         Ok(Some(m))
     }
 
-    fn pack<M: ProstMessageExt>(message: &M) -> ArrowResult<prost_types::Any> {
+    fn pack<M: ProstMessageExt>(message: &M) -> Result<prost_types::Any, ArrowError> {
         Ok(message.as_any())
     }
 }