59 lines
1.6 KiB
JavaScript
59 lines
1.6 KiB
JavaScript
const axios = require("axios");
|
|
const fs = require("fs").promises;
|
|
const dotenv = require("dotenv");
|
|
const { StoreRecord } = require("./models/db_store_controller");
|
|
const { checkUrl } = require("./models/health_status_controller");
|
|
|
|
const { init, broadcast } = require("./utils/utils");
|
|
const urlPath = "./database/urls.json";
|
|
dotenv.config();
|
|
|
|
async function setupWebSocket(wss) {
|
|
init(wss);
|
|
let Store = [];
|
|
try {
|
|
const urls = await fs.readFile(urlPath, "utf8");
|
|
const parsedUrls = JSON.parse(urls);
|
|
|
|
await Promise.allSettled(
|
|
parsedUrls.map(async (url) => {
|
|
const response = await checkUrl(url);
|
|
broadcast(response);
|
|
Store.push(response);
|
|
}),
|
|
);
|
|
StoreRecord(Store);
|
|
broadcast({ type: "checking" });
|
|
} catch (err) {
|
|
console.error("WebSocket polling error:", err.message);
|
|
broadcast({
|
|
type: "error",
|
|
message: "Unable to read urls.json or parse data",
|
|
});
|
|
}
|
|
setInterval(async () => {
|
|
try {
|
|
const urls = await fs.readFile(urlPath, "utf8");
|
|
const parsedUrls = JSON.parse(urls);
|
|
|
|
await Promise.allSettled(
|
|
parsedUrls.map(async (url) => {
|
|
const response = await checkUrl(url);
|
|
broadcast(response);
|
|
Store.push(response);
|
|
}),
|
|
);
|
|
StoreRecord(Store);
|
|
broadcast({ type: "checking" });
|
|
} catch (err) {
|
|
console.error("WebSocket polling error:", err.message);
|
|
broadcast({
|
|
type: "error",
|
|
message: "Unable to read urls.json or parse data",
|
|
});
|
|
}
|
|
}, process.env.TIME || 60000);
|
|
}
|
|
|
|
module.exports = { setupWebSocket };
|