refactor code

src/controller/website.controller.ts

@@ -1,4 +1,4 @@
-import { get_websites, store_website_stats } from "../database/website.query";
+import { get_stats_q, get_websites, store_website_stats } from "../database/website.query";
 import { fetchPageSpeedData } from "../services/pagespeed.service";
 import pg from "pg";
 import { config } from "../config/env";
@@ -13,8 +13,6 @@ export const fetchAndStoreWebsiteStats = async () => {
 
       const query = get_websites(); 
       const websites = await client.query(query);
-
-      // Process all websites in parallel using Promise.allSettled
       await Promise.allSettled(
         websites.rows.map(async (website) => {
           try {
@@ -33,4 +31,32 @@ export const fetchAndStoreWebsiteStats = async () => {
       console.error("Error fetching and storing stats:", error);
     }
   };
+  export const get_states = async (param: Param) => {
+    try {
+      await client.connect();
+
+      const query = get_stats_q(param);
+      const stats = await client.query(query);
+
+      if (stats.rows.length > 0) {
+
+        const parsedStats = stats.rows.map(row => {
+          return {
+            ...row,
+            stats: row.stats ? JSON.parse(row.stats) : null 
+          };
+        });
+
+        return parsedStats;
+      } else {
+        console.log("No data found");
+        return [];
+      }
+    } catch (error) {
+      console.error("Error fetching and storing stats:", error);
+      return { message: "Error fetching stats", error: error.message };
+    } finally {
+      await client.end(); // Make sure to close the client connection
+    }
+  };
 

src/database/website.query.ts

@@ -3,4 +3,17 @@ const store_website_stats = (websiteId: number,  stats: object) => `
   INSERT INTO website_stats (website_id, stats)
   VALUES (${websiteId},  '${JSON.stringify(stats)}'::jsonb) returning id;
 `;
-export {get_websites,store_website_stats};
+
+  interface Param {
+    url?: string;
+  }
+
+  const get_stats_q = (param: Param) => `
+    SELECT ws.website_id, w.url,ws.stats
+    FROM ${process.env.schema}.website_stats ws
+    JOIN ${process.env.schema}.websites w
+    ON ws.website_id = w.id
+    ${param.url && param.url !== "" ? `WHERE w.url = '${param.url}'` : ""}
+  `;
+
+export {get_websites,store_website_stats,get_stats_q};

src/routes/website.routes.ts

@@ -1,12 +1,18 @@
 import { Elysia } from "elysia";
-import { fetchAndStoreWebsiteStats } from "../controller/website.controller";
+import { fetchAndStoreWebsiteStats, get_states } from "../controller/website.controller";
 
 const router = new Elysia();
 
-router.post("/fetch-stats", async () => {
+router.post("/store-stats", async () => {
   await   fetchAndStoreWebsiteStats();
   return { message: "Response Stored Successfully." };
 });
 
+router.get("/get-stats", async () => {
+    return   get_states();
+    return { message: "Response Stord Successfully." };
+  });
+
+
 export default router;