From 81bc1f8f0c948438391248f42e441fc7ec898be6 Mon Sep 17 00:00:00 2001
From: ajnart <ajnart@pm.me>
Date: Sun, 11 Dec 2022 14:25:27 +0900
Subject: [PATCH] Add a timeout of 2 seconds for the ping module

---
 src/pages/api/modules/ping.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/pages/api/modules/ping.ts b/src/pages/api/modules/ping.ts
index de5dcc79d..add269e54 100644
--- a/src/pages/api/modules/ping.ts
+++ b/src/pages/api/modules/ping.ts
@@ -7,13 +7,15 @@ async function Get(req: NextApiRequest, res: NextApiResponse) {
   const { url } = req.query;
   const agent = new https.Agent({ rejectUnauthorized: false });
   await axios
-    .get(url as string, { httpsAgent: agent })
+    .get(url as string, { httpsAgent: agent, timeout: 2000 })
     .then((response) => {
       res.status(response.status).json(response.statusText);
     })
     .catch((error) => {
       if (error.response) {
         res.status(error.response.status).json(error.response.statusText);
+      } else if (error.code === 'ECONNABORTED') {
+        res.status(408).json('Request Timeout');
       } else {
         res.status(500).json('Server Error');
       }