Update error.rs

longportapp · Feb 5, 2025 · dc0199f · dc0199f
1 parent 125012c
commit dc0199f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/rust/src/error.rs b/rust/src/error.rs
@@ -61,6 +61,17 @@ impl Error {
         }
     }
 
+    /// Returns the OpenAPI error code
+    pub fn openapi_error_code(&self) -> Option<i64> {
+        match self {
+            Error::HttpClient(HttpClientError::OpenApi { code, .. }) => Some(*code as i64),
+            Error::WsClient(WsClientError::ResponseError { detail, .. }) => {
+                detail.as_ref().map(|detail| detail.code as i64)
+            }
+            _ => None,
+        }
+    }
+
     /// Consumes this error and returns a simple error
     pub fn into_simple_error(self) -> SimpleError {
         match self {