fud v3

heliaxdev · Nov 3, 2023 · e05962e · e05962e
1 parent bfea2f7
commit e05962e
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/server.rs b/src/server.rs
@@ -180,7 +180,16 @@ where
                         match listener_clone.accept().await {
                             Ok((socket, _addr)) => {
                                 if let Err(e) = conn.clone().run(socket).await {
-                                    Err(backoff::Error::Permanent(e))
+                                    match e.downcast::<tower::load_shed::error::Overloaded>() {
+                                        Err(e) => {
+                                            tracing::error!({ %e }, "error in a connection handler");
+                                            return Err(backoff::Error::Permanent(e))
+                                        }
+                                        Ok(e) => {
+                                            tracing::warn!("Service overloaded - backing off");
+                                            return Err(backoff::Error::transient(e));
+                                        }
+                                    }
                                 } else {
                                     Ok(())
                                 }
@@ -191,7 +200,7 @@ where
                             }
                         }
                     }
-                );
+                ).await;
             });
         }
     }