fix: Add better warning for rate limit (fix #67) and increase burst t…

…o 20
ixhbinphoenix · Mar 20, 2024 · 09bab9a · 09bab9a
1 parent d186481
commit 09bab9a
Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/backend/src/main.rs b/backend/src/main.rs
@@ -159,14 +159,14 @@ async fn main() -> io::Result<()> {
         let governor_config = GovernorConfigBuilder::default()
                 .key_extractor(NginxIpKeyExctrator)
                 .per_second(10)
-                .burst_size(5)
+                .burst_size(20)
                 .use_headers()
                 .finish()
                 .unwrap();
         #[cfg(not(feature = "proxy"))] 
         let governor_config = GovernorConfigBuilder::default()
                 .per_second(10)
-                .burst_size(5)
+                .burst_size(20)
                 .use_headers()
                 .finish()
                 .unwrap();

diff --git a/frontend/src/api/frontmatter.ts b/frontend/src/api/frontmatter.ts
@@ -14,6 +14,9 @@ class Request {
       if (!result.body) {
         return Promise.reject({ status: 500, message: "Server Connection Failed" });
       }
+      if (result.status == 429) {
+        return Promise.reject(new Error("Too many requests. Try again later"));
+      }
       let stream = await Request.readStream(result.body);
       let body = JSON.parse(stream);
       if (!body.success) {
@@ -31,6 +34,9 @@ class Request {
         method: "GET",
         credentials: "include"
       });
+      if (result.status == 429) {
+        return Promise.reject(new Error("Too many requests. Try again later"));
+      }
       const body = await result.json();
       if (!body.success) {
         return Promise.reject(body.body);

diff --git a/frontend/src/api/main.ts b/frontend/src/api/main.ts
@@ -84,4 +84,4 @@ async function readStream(stream: ReadableStream<Uint8Array>) {
       chunks.push(textDecode.decode(value));
     }
     return chunks.join("");
-}
+}
diff --git a/frontend/src/api/theBackend.ts b/frontend/src/api/theBackend.ts
@@ -17,6 +17,9 @@ class Request {
       if (!result.body) {
         return Promise.reject({ status: 500, message: "Server Connection Failed" });
       }
+      if (result.status == 429) {
+        return Promise.reject(new Error("Too many requests. Try again later"));
+      }
       let stream = await Request.readStream(result.body);
       let body = JSON.parse(stream);
       if (!body.success) {
@@ -34,6 +37,9 @@ class Request {
         method: "GET",
         credentials: "include"
       });
+      if (result.status == 429) {
+        return Promise.reject(new Error("Too many requests. Try again later"));
+      }
       const body = await result.json();
       if (!body.success) {
         return Promise.reject(body.body);