fix(error): use status.code instead of error_code for Milvus > 2.3 compatibility

Author: Marlon Costa

src/error.rs

@@ -16,6 +16,8 @@
 // For custom error handling
 // TODO
 
+use std::convert::TryFrom;
+
 use crate::collection::Error as CollectionError;
 use crate::proto::common::{ErrorCode, Status};
 use crate::schema::Error as SchemaError;
@@ -73,7 +75,13 @@ pub enum Error {
 
 impl From<Status> for Error {
     fn from(s: Status) -> Self {
-        Error::Server(ErrorCode::from_i32(s.error_code).unwrap(), s.reason)
+        let code = ErrorCode::try_from(s.code).unwrap_or(ErrorCode::UnexpectedError);
+        let reason = if code == ErrorCode::UnexpectedError && s.code != code as i32 {
+            format!("{} (code {})", s.reason, s.code)
+        } else {
+            s.reason
+        };
+        Error::Server(code, reason)
     }
 }
 

src/utils.rs

@@ -14,6 +14,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+use std::convert::TryFrom;
+
 use crate::{
     error::Error,
     proto::common::{ErrorCode, Status},
@@ -24,14 +26,12 @@ pub fn status_to_result(status: &Option<Status>) -> Result<(), Error> {
         .clone()
         .ok_or(Error::Unexpected("no status".to_owned()))?;
 
-    match ErrorCode::from_i32(status.error_code) {
-        Some(i) => match i {
-            ErrorCode::Success => Ok(()),
-            _ => Err(Error::from(status)),
-        },
-        None => Err(Error::Unexpected(format!(
-            "unknown error code {}",
-            status.error_code
-        ))),
+    match ErrorCode::try_from(status.code) {
+        Ok(ErrorCode::Success) => Ok(()),
+        Ok(_) => Err(Error::from(status)),
+        Err(_) => Err(Error::Server(
+            ErrorCode::UnexpectedError,
+            format!("{} (code {})", status.reason, status.code),
+        )),
     }
 }