error handling changes

GitOrigin-RevId: 184aaa1ac8ee197bc82817b84d6b97296d39c6f8

Author: gautamg795

crates/common/src/retriable_stream.rs

@@ -64,6 +64,10 @@ where
         }
         anyhow::bail!("Attempted to clone RetriableStream after it has been polled")
     }
+
+    pub fn is_fresh(&self) -> bool {
+        self.shared_inner.lock().is_some()
+    }
 }
 
 impl<T: TryStream + std::marker::Unpin> Stream for RetriableStream<T>

crates/errors/src/lib.rs

@@ -347,6 +347,10 @@ impl ErrorMetadata {
         self.code == ErrorCode::Forbidden
     }
 
+    pub fn is_misdirected_request(&self) -> bool {
+        self.code == ErrorCode::MisdirectedRequest
+    }
+
     /// Return true if this error is deterministically caused by user. If so,
     /// we can propagate it into JS out of a syscall, and cache it if it is the
     /// full UDF result.
@@ -499,7 +503,7 @@ impl ErrorCode {
             ErrorCode::PaginationLimit => tonic::Code::InvalidArgument,
             ErrorCode::OutOfRetention => tonic::Code::OutOfRange,
             ErrorCode::OperationalInternalServerError => tonic::Code::Internal,
-            ErrorCode::MisdirectedRequest => tonic::Code::InvalidArgument,
+            ErrorCode::MisdirectedRequest => tonic::Code::FailedPrecondition,
         }
     }
 
@@ -531,6 +535,7 @@ pub trait ErrorMetadataAnyhowExt {
     fn is_forbidden(&self) -> bool;
     fn should_report_to_sentry(&self) -> Option<(sentry::Level, Option<f64>)>;
     fn is_deterministic_user_error(&self) -> bool;
+    fn is_misdirected_request(&self) -> bool;
     fn user_facing_message(&self) -> String;
     fn short_msg(&self) -> &str;
     fn msg(&self) -> &str;
@@ -618,6 +623,13 @@ impl ErrorMetadataAnyhowExt for anyhow::Error {
         false
     }
 
+    fn is_misdirected_request(&self) -> bool {
+        if let Some(e) = self.downcast_ref::<ErrorMetadata>() {
+            return e.is_misdirected_request();
+        }
+        false
+    }
+
     /// Returns the level at which the given error should report to sentry
     /// INFO -> it's a client-at-fault error
     /// WARNING -> it's a server-at-fault error that is expected