backoff incase of unknown error (#35)

* backoff incase of unknown error * update log to print error * tidy up error log
tembo-io · May 30, 2024 · 65f6d7d · 65f6d7d
1 parent 9132b59
commit 65f6d7d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 15 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -23,7 +23,6 @@ serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 supabase-wrappers = { git = "/~https://github.com/supabase/wrappers.git", default-features = false }
 tokio = { version = "1", features = ["full"] }
-reqwest = "0.11"
 
 [dev-dependencies]
 pgrx-tests = "=0.11.3"
diff --git a/src/lib.rs b/src/lib.rs
@@ -13,6 +13,7 @@ use supabase_wrappers::prelude::*;
 pgrx::pg_module_magic!();
 
 use clerk_rs::{
+    apis::organization_memberships_api::ListOrganizationMembershipsError,
     apis::organization_memberships_api::OrganizationMembership,
     apis::organizations_api::Organization, apis::users_api::User, clerk::Clerk, ClerkConfiguration,
 };
@@ -248,21 +249,13 @@ impl ForeignDataWrapper<ClerkFdwError> for ClerkFdw {
                                     )
                                     .await
                                     .map_err(|e| match e {
-                                        clerk_rs::apis::Error::Reqwest(ref reqwest_error) => {
-                                            if let Some(status_code) = reqwest_error.status() {
-                                                match status_code {
-                                                    reqwest::StatusCode::TOO_MANY_REQUESTS |
-                                                    reqwest::StatusCode::BAD_GATEWAY |
-                                                    reqwest::StatusCode::SERVICE_UNAVAILABLE |
-                                                    reqwest::StatusCode::GATEWAY_TIMEOUT |
-                                                    reqwest::StatusCode::INTERNAL_SERVER_ERROR => {
-                                                        info!("clerk_fdw: received {} error, backing off", status_code);
-                                                        backoff::Error::transient(e)
-                                                    },
-                                                    _ => backoff::Error::Permanent(e),
+                                        clerk_rs::apis::Error::ResponseError(ref response_error) => {
+                                            match response_error.entity {
+                                                Some(ListOrganizationMembershipsError::UnknownValue(_)) => {
+                                                    info!("clerk_fdw: received {:#?}, backing off", response_error.content);
+                                                    backoff::Error::transient(e)
                                                 }
-                                            } else {
-                                                backoff::Error::Permanent(e)
+                                                _ => backoff::Error::Permanent(e),
                                             }
                                         }
                                         _ => backoff::Error::Permanent(e),