AmineTMIRI/backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfixing websockets

Browse Source
This commit is contained in:
MuslemRahimi 2024-09-09 18:19:19 +02:00
parent fc03007901
commit 5dda91a1a1
2 changed files with 338 additions and 330 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
app/cron_export_price.py
View File

@ -19,35 +19,6 @@ def date_range_days(steps=20):
yield start_date.strftime("%Y-%m-%d"), min(next_date, end_date).strftime("%Y-%m-%d") yield start_date.strftime("%Y-%m-%d"), min(next_date, end_date).strftime("%Y-%m-%d")
start_date = next_date start_date = next_date
def get_existing_data(symbol, interval):
file_path = f"json/export/price/{interval}/{symbol}.json"
if os.path.exists(file_path):
with open(file_path, 'r') as file:
return ujson.load(file)
return []
def get_missing_date_ranges(existing_data, start_date, end_date):
existing_dates = set(item['date'].split()[0] for item in existing_data)
current_date = datetime.strptime(start_date, "%Y-%m-%d")
end = datetime.strptime(end_date, "%Y-%m-%d")
missing_ranges = []
range_start = None
while current_date <= end:
date_str = current_date.strftime("%Y-%m-%d")
if date_str not in existing_dates:
if range_start is None:
range_start = date_str
elif range_start is not None:
missing_ranges.append((range_start, (current_date - timedelta(days=1)).strftime("%Y-%m-%d")))
range_start = None
current_date += timedelta(days=1)
if range_start is not None:
missing_ranges.append((range_start, end_date))
return missing_ranges
async def get_data_batch(session, symbol, url_list): async def get_data_batch(session, symbol, url_list):
tasks = [] tasks = []
for url in url_list: for url in url_list:
@ -60,6 +31,40 @@ async def get_data_batch(session, symbol, url_list):
data.extend(result) data.extend(result)
return data return data
def get_existing_data(symbol, interval):
file_path = f"json/export/price/{interval}/{symbol}.json"
if os.path.exists(file_path):
with open(file_path, 'r') as file:
return ujson.load(file)
return []
def get_missing_date_ranges(existing_data, start_date, end_date):
existing_dates = {item['date'].split()[0] for item in existing_data} # Use a set for O(1) lookup time
start_date = datetime.strptime(start_date, "%Y-%m-%d")
end_date = datetime.strptime(end_date, "%Y-%m-%d")
missing_ranges = []
current_date = start_date
range_start = None
while current_date <= end_date:
date_str = current_date.strftime("%Y-%m-%d")
if date_str not in existing_dates:
if range_start is None:
range_start = current_date
else:
# If we found an existing date, and we have a start for a missing range, add it
if range_start:
missing_ranges.append((range_start.strftime("%Y-%m-%d"), date_str))
range_start = None
current_date += timedelta(days=1)
# If the loop ends and we still have an open range, add it
if range_start:
missing_ranges.append((range_start.strftime("%Y-%m-%d"), end_date.strftime("%Y-%m-%d")))
return missing_ranges
async def fetch_data(session, url): async def fetch_data(session, url):
try: try:
async with session.get(url) as response: async with session.get(url) as response:

381
fastify/app.js
View File

@ -1,11 +1,10 @@
let serverRunning = false; let serverRunning = false;
const fastify = require('fastify')({}) const fastify = require("fastify")({});
const cors = require('@fastify/cors'); const cors = require("@fastify/cors");
//Load API KEYS //Load API KEYS
//require('dotenv').config({ path: '../app/.env' }); require("dotenv").config({ path: "../app/.env" });
const fmpAPIKey = process.env.FMP_API_KEY; const fmpAPIKey = process.env.FMP_API_KEY;
//const mixpanelAPIKey = process.env.MIXPANEL_API_KEY; //const mixpanelAPIKey = process.env.MIXPANEL_API_KEY;
const twitchAPIKey = process.env.TWITCH_API_KEY; const twitchAPIKey = process.env.TWITCH_API_KEY;
@ -14,117 +13,124 @@ const twitchSecretKey = process.env.TWITCH_SECRET_KEY;
//const Mixpanel = require('mixpanel'); //const Mixpanel = require('mixpanel');
//const UAParser = require('ua-parser-js'); //const UAParser = require('ua-parser-js');
const got = require('got'); //Only version npm i got@11.8.3 works with ESM const got = require("got"); //Only version npm i got@11.8.3 works with ESM
const cheerio = require('cheerio'); const cheerio = require("cheerio");
const sharp = require('sharp'); const sharp = require("sharp");
const axios = require('axios'); const axios = require("axios");
const fs = require('fs'); const fs = require("fs");
const path = require('path'); const path = require("path");
//const pino = require('pino'); //const pino = require('pino');
//const mixpanel = Mixpanel.init(mixpanelAPIKey, { debug: false }); //const mixpanel = Mixpanel.init(mixpanelAPIKey, { debug: false });
const PocketBase = require("pocketbase/cjs");
const PocketBase = require('pocketbase/cjs') const pb = new PocketBase("http://127.0.0.1:8090");
const pb = new PocketBase('http://127.0.0.1:8090');
// globally disable auto cancellation // globally disable auto cancellation
//See https://github.com/pocketbase/js-sdk#auto-cancellation //See https://github.com/pocketbase/js-sdk#auto-cancellation
//Bug happens that get-post gives an error of auto-cancellation. Hence set it to false; //Bug happens that get-post gives an error of auto-cancellation. Hence set it to false;
pb.autoCancellation(false); pb.autoCancellation(false);
const { serialize } = require('object-to-formdata'); const { serialize } = require("object-to-formdata");
// Register the CORS plugin // Register the CORS plugin
//Add Cors so that only localhost and my stocknear.com can send acceptable requests //Add Cors so that only localhost and my stocknear.com can send acceptable requests
fastify.register(cors); fastify.register(cors);
const corsMiddleware = (request, reply, done) => { const corsMiddleware = (request, reply, done) => {
const allowedOrigins = ['http://localhost:4173','http://127.0.0.1:4173','http://localhost:5173', 'http://127.0.0.1:5173', 'https://stocknear.com','https://www.stocknear.com', 'http://stocknear.com', 'http://www.stocknear.com']; const allowedOrigins = [
"http://localhost:4173",
"http://127.0.0.1:4173",
"http://localhost:5173",
"http://127.0.0.1:5173",
"https://stocknear.com",
"https://www.stocknear.com",
"http://stocknear.com",
"http://www.stocknear.com",
];
const origin = request?.headers?.origin; const origin = request?.headers?.origin;
if (!origin || allowedOrigins?.includes(origin)) { if (!origin || allowedOrigins?.includes(origin)) {
reply.header('Access-Control-Allow-Origin', origin || '*'); reply.header("Access-Control-Allow-Origin", origin || "*");
reply.header('Access-Control-Allow-Methods', 'GET,POST'); reply.header("Access-Control-Allow-Methods", "GET,POST");
reply.header('Access-Control-Allow-Headers', 'Content-Type'); reply.header("Access-Control-Allow-Headers", "Content-Type");
done(); done();
} else { } else {
reply.code(403).send({ error: 'Forbidden' }); reply.code(403).send({ error: "Forbidden" });
} }
}; };
fastify.addHook('onRequest', corsMiddleware); fastify.addHook("onRequest", corsMiddleware);
//fastify.register(require('./mixpanel/server'), { mixpanel, UAParser }); //fastify.register(require('./mixpanel/server'), { mixpanel, UAParser });
fastify.register(require('./get-user-stats/server'), { pb }); fastify.register(require("./get-user-stats/server"), { pb });
fastify.register(require('./get-community-stats/server'), { pb }); fastify.register(require("./get-community-stats/server"), { pb });
fastify.register(require('./get-moderators/server'), { pb }); fastify.register(require("./get-moderators/server"), { pb });
fastify.register(require('./get-user-data/server'), { pb }); fastify.register(require("./get-user-data/server"), { pb });
fastify.register(require('./get-all-comments/server'), { pb }); fastify.register(require("./get-all-comments/server"), { pb });
fastify.register(require('./get-post/server'), { pb }); fastify.register(require("./get-post/server"), { pb });
fastify.register(require('./get-one-post/server'), { pb }); fastify.register(require("./get-one-post/server"), { pb });
fastify.register(require('./update-watchlist/server'), { pb, serialize }); fastify.register(require("./update-watchlist/server"), { pb, serialize });
fastify.register(require('./get-portfolio-data/server'), { pb }); fastify.register(require("./get-portfolio-data/server"), { pb });
fastify.register(require('./create-portfolio/server'), { pb, serialize }); fastify.register(require("./create-portfolio/server"), { pb, serialize });
fastify.register(require('./buy-stock/server'), { pb }); fastify.register(require("./buy-stock/server"), { pb });
fastify.register(require('./sell-stock/server'), { pb }); fastify.register(require("./sell-stock/server"), { pb });
fastify.register(require('./create-post-link/server'), { got,cheerio,sharp }); fastify.register(require("./create-post-link/server"), { got, cheerio, sharp });
fastify.register(require('./create-post-image/server'), { sharp }); fastify.register(require("./create-post-image/server"), { sharp });
fastify.register(require('./delete-comment/server'), { pb }); fastify.register(require("./delete-comment/server"), { pb });
fastify.register(require('./delete-post/server'), { pb }); fastify.register(require("./delete-post/server"), { pb });
fastify.register(require('./leaderboard/server'), { pb }); fastify.register(require("./leaderboard/server"), { pb });
fastify.register(require('./feedback/server'), { pb }); fastify.register(require("./feedback/server"), { pb });
fastify.register(require('./create-watchlist/server'), { pb }); fastify.register(require("./create-watchlist/server"), { pb });
fastify.register(require('./delete-watchlist/server'), { pb }); fastify.register(require("./delete-watchlist/server"), { pb });
fastify.register(require('./edit-name-watchlist/server'), { pb }); fastify.register(require("./edit-name-watchlist/server"), { pb });
fastify.register(require('./all-watchlists/server'), { pb }); fastify.register(require("./all-watchlists/server"), { pb });
fastify.register(require('./get-notifications/server'), { pb }); fastify.register(require("./get-notifications/server"), { pb });
fastify.register(require('./update-notifications/server'), { pb }); fastify.register(require("./update-notifications/server"), { pb });
fastify.register(require('./create-strategy/server'), { pb }); fastify.register(require("./create-strategy/server"), { pb });
fastify.register(require('./delete-strategy/server'), { pb }); fastify.register(require("./delete-strategy/server"), { pb });
fastify.register(require('./all-strategies/server'), { pb }); fastify.register(require("./all-strategies/server"), { pb });
fastify.register(require('./save-strategy/server'), { pb }); fastify.register(require("./save-strategy/server"), { pb });
fastify.register(require('./get-strategy/server'), { pb }); fastify.register(require("./get-strategy/server"), { pb });
fastify.register(require('./get-twitch-status/server'), { axios,twitchAPIKey, twitchSecretKey }); fastify.register(require("./get-twitch-status/server"), {
fastify.register(require('./get-portfolio/server'), { pb }); axios,
fastify.register(require('./create-price-alert/server'), { pb }); twitchAPIKey,
fastify.register(require('./get-price-alert/server'), { pb, fs, path }); twitchSecretKey,
fastify.register(require('./delete-price-alert/server'), { pb }); });
fastify.register(require('./upvote/server'), { pb }); fastify.register(require("./get-portfolio/server"), { pb });
fastify.register(require('./downvote/server'), { pb }); fastify.register(require("./create-price-alert/server"), { pb });
fastify.register(require('./upvote-comment/server'), { pb }); fastify.register(require("./get-price-alert/server"), { pb, fs, path });
fastify.register(require('./downvote-comment/server'), { pb }); fastify.register(require("./delete-price-alert/server"), { pb });
fastify.register(require("./upvote/server"), { pb });
fastify.register(require("./downvote/server"), { pb });
fastify.register(require("./upvote-comment/server"), { pb });
fastify.register(require("./downvote-comment/server"), { pb });
//fastify.register(require('./create-comment/server'), { pb }); //fastify.register(require('./create-comment/server'), { pb });
function wait(ms) {
function wait(ms){
var start = new Date().getTime(); var start = new Date().getTime();
var end = start; var end = start;
while(end < start + ms) { while (end < start + ms) {
end = new Date().getTime(); end = new Date().getTime();
} }
} }
fastify.register(require('@fastify/websocket')) fastify.register(require("@fastify/websocket"));
const WebSocket = require('ws'); const WebSocket = require("ws");
let isSend = false; let isSend = false;
let sendInterval; let sendInterval;
fastify.register(async function (fastify) { fastify.register(async function (fastify) {
fastify.get('/realtime-data', { websocket: true }, (connection, req) => { fastify.get("/realtime-data", { websocket: true }, (connection, req) => {
// Send a welcome message to the client // Send a welcome message to the client
//connection.socket.send('hi from server'); //connection.socket.send('hi from server');
// Listen for incoming messages from the client // Listen for incoming messages from the client
connection.socket.on('message', (message) => { connection.socket.on("message", (message) => {
symbol = message.toString('utf-8'); symbol = message.toString("utf-8");
console.log('Received message from client:', symbol); console.log("Received message from client:", symbol);
// If you want to dynamically update the subscription based on client's message // If you want to dynamically update the subscription based on client's message
updateSubscription(); updateSubscription();
@ -132,55 +138,51 @@ fastify.register(async function (fastify) {
//====================== //======================
const login = { const login = {
'event': 'login', event: "login",
'data': { data: {
'apiKey': fmpAPIKey, apiKey: fmpAPIKey,
} },
} };
const subscribe = { const subscribe = {
'event': 'subscribe', event: "subscribe",
'data': { data: {
'ticker': '', // Initial value; will be updated dynamically ticker: "", // Initial value; will be updated dynamically
} },
} };
function updateSubscription() { function updateSubscription() {
subscribe.data.ticker = symbol; subscribe.data.ticker = symbol;
} }
// Create a new WebSocket instance for your backend // Create a new WebSocket instance for your backend
const ws = new WebSocket('wss://websockets.financialmodelingprep.com'); const ws = new WebSocket("wss://websockets.financialmodelingprep.com");
// Handle WebSocket connection open // Handle WebSocket connection open
ws.on('open', function open() { ws.on("open", function open() {
ws.send(JSON.stringify(login)); ws.send(JSON.stringify(login));
wait(2000); //2 seconds in milliseconds wait(2000); //2 seconds in milliseconds
ws.send(JSON.stringify(subscribe)); ws.send(JSON.stringify(subscribe));
}); });
// Handle WebSocket errors // Handle WebSocket errors
ws.on('error', function (error) { ws.on("error", function (error) {
console.error('WebSocket error:', error); console.error("WebSocket error:", error);
// Handle the error gracefully, you might want to notify the client or log it. // Handle the error gracefully, you might want to notify the client or log it.
// For now, let's close the connection if an error occurs // For now, let's close the connection if an error occurs
connection.socket.close(); connection.socket.close();
}); });
ws.on("message", function (data, flags) {
const stringData = data.toString("utf-8");
ws.on('message', function (data, flags) {
const stringData = data.toString('utf-8');
try { try {
const jsonData = JSON.parse(stringData); const jsonData = JSON.parse(stringData);
const bpData = jsonData.bp; const bpData = jsonData.bp;
// Check if bpData is a number and not equal to zero // Check if bpData is a number and not equal to zero
if (typeof bpData === 'number' && bpData !== 0) { if (typeof bpData === "number" && bpData !== 0) {
if (connection.socket.readyState === WebSocket.OPEN && !isSend) { if (connection.socket.readyState === WebSocket.OPEN && !isSend) {
connection.socket.send(JSON.stringify({ bp: bpData })); connection.socket.send(JSON.stringify({ bp: bpData }));
isSend = true; isSend = true;
@ -190,43 +192,39 @@ fastify.register(async function (fastify) {
} }
} }
} catch (error) { } catch (error) {
console.error('Error parsing JSON:', error); console.error("Error parsing JSON:", error);
} }
}); });
//====================== //======================
// Handle client disconnect // Handle client disconnect
connection.socket.on('close', () => { connection.socket.on("close", () => {
console.log('Client disconnected'); console.log("Client disconnected");
connection?.socket?.close(); connection?.socket?.close();
// Check if the WebSocket is open before trying to close it // Check if the WebSocket is open before trying to close it
if (ws.readyState === WebSocket.OPEN) { if (ws.readyState === WebSocket.OPEN) {
try { try {
ws.close(); ws.close();
} catch (e) { } catch (e) {
console.error('Error while closing WebSocket:', e); console.error("Error while closing WebSocket:", e);
} }
} }
}); });
}); });
}); });
fastify.register(async function (fastify) { fastify.register(async function (fastify) {
fastify.get('/realtime-crypto-data', { websocket: true }, (connection, req) => { fastify.get(
"/realtime-crypto-data",
{ websocket: true },
(connection, req) => {
// Send a welcome message to the client // Send a welcome message to the client
// Listen for incoming messages from the client // Listen for incoming messages from the client
connection.socket.on('message', (message) => { connection.socket.on("message", (message) => {
symbol = message.toString('utf-8'); symbol = message.toString("utf-8");
console.log('Received message from client:', symbol); console.log("Received message from client:", symbol);
// If you want to dynamically update the subscription based on client's message // If you want to dynamically update the subscription based on client's message
updateSubscription(); updateSubscription();
@ -234,106 +232,106 @@ fastify.register(async function (fastify) {
//====================== //======================
const login = { const login = {
'event': 'login', event: "login",
'data': { data: {
'apiKey': fmpAPIKey, apiKey: fmpAPIKey,
} },
} };
const subscribe = { const subscribe = {
'event': 'subscribe', event: "subscribe",
'data': { data: {
'ticker': '', // Initial value; will be updated dynamically ticker: "", // Initial value; will be updated dynamically
} },
} };
function updateSubscription() { function updateSubscription() {
subscribe.data.ticker = symbol; subscribe.data.ticker = symbol;
} }
// Create a new WebSocket instance for your backend // Create a new WebSocket instance for your backend
const ws = new WebSocket('wss://crypto.financialmodelingprep.com'); const ws = new WebSocket("wss://crypto.financialmodelingprep.com");
// Handle WebSocket connection open // Handle WebSocket connection open
ws.on('open', function open() { ws.on("open", function open() {
ws.send(JSON.stringify(login)); ws.send(JSON.stringify(login));
wait(2000); //2 seconds in milliseconds wait(2000); //2 seconds in milliseconds
ws.send(JSON.stringify(subscribe)); ws.send(JSON.stringify(subscribe));
}); });
// Handle WebSocket errors // Handle WebSocket errors
ws.on('error', function (error) { ws.on("error", function (error) {
console.error('WebSocket error:', error); console.error("WebSocket error:", error);
// Handle the error gracefully, you might want to notify the client or log it. // Handle the error gracefully, you might want to notify the client or log it.
// For now, let's close the connection if an error occurs // For now, let's close the connection if an error occurs
connection.socket.close(); connection.socket.close();
}); });
// Handle WebSocket messages // Handle WebSocket messages
ws.on('message', function (data, flags) { ws.on("message", function (data, flags) {
const stringData = data.toString('utf-8'); const stringData = data.toString("utf-8");
if (connection.socket.readyState === WebSocket.OPEN && !isSend) { if (connection.socket.readyState === WebSocket.OPEN && !isSend) {
connection.socket.send(stringData); connection.socket.send(stringData);
//console.log(stringData) //console.log(stringData)
isSend = true isSend = true;
setTimeout(() => { setTimeout(() => {
isSend = false}, 1000); isSend = false;
}, 1000);
//wait(2000); //wait(2000);
} }
//wait(2000); //wait(2000);
}); });
//====================== //======================
// Handle client disconnect // Handle client disconnect
connection.socket.on('close', () => { connection.socket.on("close", () => {
console.log('Client disconnected'); console.log("Client disconnected");
connection?.socket?.close(); connection?.socket?.close();
// Check if the WebSocket is open before trying to close it // Check if the WebSocket is open before trying to close it
if (ws.readyState === WebSocket.OPEN) { if (ws.readyState === WebSocket.OPEN) {
try { try {
ws.close(); ws.close();
} catch (e) { } catch (e) {
console.error('Error while closing WebSocket:', e); console.error("Error while closing WebSocket:", e);
} }
} }
}); });
}
);
});
}); });
fastify.register(async function (fastify) { fastify.register(async function (fastify) {
fastify.get('/options-flow-reader', { websocket: true }, (connection, req) => { fastify.get(
"/options-flow-reader",
{ websocket: true },
(connection, req) => {
let jsonData; let jsonData;
let sendInterval; let sendInterval;
// Function to send data to the client // Function to send data to the client
const sendData = async () => { const sendData = async () => {
const filePath = path.join(__dirname, '../app/json/options-flow/feed/data.json'); const filePath = path.join(
__dirname,
"../app/json/options-flow/feed/data.json"
);
try { try {
if (fs.existsSync(filePath)) { if (fs.existsSync(filePath)) {
const fileData = fs.readFileSync(filePath, 'utf8'); const fileData = fs.readFileSync(filePath, "utf8");
jsonData = JSON.parse(fileData); jsonData = JSON.parse(fileData);
connection.socket.send(JSON.stringify(jsonData)); connection.socket.send(JSON.stringify(jsonData));
} else { } else {
console.error('File not found:', filePath); console.error("File not found:", filePath);
clearInterval(sendInterval); clearInterval(sendInterval);
connection?.socket?.close(); connection?.socket?.close();
console.error('Connection closed') console.error("Connection closed");
throw new Error('This is an intentional uncaught exception!'); throw new Error("This is an intentional uncaught exception!");
} }
} catch (err) { } catch (err) {
console.error('Error sending data to client:', err); console.error("Error sending data to client:", err);
} }
}; };
@ -344,49 +342,55 @@ fastify.register(async function (fastify) {
sendInterval = setInterval(sendData, 5000); sendInterval = setInterval(sendData, 5000);
// Handle client disconnect // Handle client disconnect
connection.socket.on('close', () => { connection.socket.on("close", () => {
console.log('Client disconnected'); console.log("Client disconnected");
clearInterval(sendInterval); clearInterval(sendInterval);
}); });
// Handle server crash cleanup // Handle server crash cleanup
const closeHandler = () => { const closeHandler = () => {
console.log('Server is closing. Cleaning up resources...'); console.log("Server is closing. Cleaning up resources...");
clearInterval(sendInterval); clearInterval(sendInterval);
connection.socket.close(); connection.socket.close();
}; };
// Add close handler to process event // Add close handler to process event
process.on('exit', closeHandler); process.on("exit", closeHandler);
process.on('SIGINT', closeHandler); process.on("SIGINT", closeHandler);
process.on('SIGTERM', closeHandler); process.on("SIGTERM", closeHandler);
process.on('uncaughtException', closeHandler); process.on("uncaughtException", closeHandler);
process.on('unhandledRejection', closeHandler); process.on("unhandledRejection", closeHandler);
}); }
);
}); });
fastify.register(async function (fastify) { fastify.register(async function (fastify) {
fastify.get('/options-zero-dte-reader', { websocket: true }, (connection, req) => { fastify.get(
"/options-zero-dte-reader",
{ websocket: true },
(connection, req) => {
let jsonData; let jsonData;
let sendInterval; let sendInterval;
// Function to send data to the client // Function to send data to the client
const sendData = async () => { const sendData = async () => {
const filePath = path.join(__dirname, '../app/json/options-flow/zero-dte/data.json'); const filePath = path.join(
__dirname,
"../app/json/options-flow/zero-dte/data.json"
);
try { try {
if (fs.existsSync(filePath)) { if (fs.existsSync(filePath)) {
const fileData = fs.readFileSync(filePath, 'utf8'); const fileData = fs.readFileSync(filePath, "utf8");
jsonData = JSON.parse(fileData); jsonData = JSON.parse(fileData);
connection.socket.send(JSON.stringify(jsonData)); connection.socket.send(JSON.stringify(jsonData));
} else { } else {
console.error('File not found:', filePath); console.error("File not found:", filePath);
clearInterval(sendInterval); clearInterval(sendInterval);
console.error('Connection closed') console.error("Connection closed");
throw new Error('This is an intentional uncaught exception!'); throw new Error("This is an intentional uncaught exception!");
} }
} catch (err) { } catch (err) {
console.error('Error sending data to client:', err); console.error("Error sending data to client:", err);
} }
}; };
@ -397,45 +401,42 @@ fastify.register(async function (fastify) {
sendInterval = setInterval(sendData, 5000); sendInterval = setInterval(sendData, 5000);
// Handle client disconnect // Handle client disconnect
connection.socket.on('close', () => { connection.socket.on("close", () => {
console.log('Client disconnected'); console.log("Client disconnected");
connection?.socket?.close(); connection?.socket?.close();
clearInterval(sendInterval); clearInterval(sendInterval);
}); });
// Handle server crash cleanup // Handle server crash cleanup
const closeHandler = () => { const closeHandler = () => {
console.log('Server is closing. Cleaning up resources...'); console.log("Server is closing. Cleaning up resources...");
clearInterval(sendInterval); clearInterval(sendInterval);
connection?.socket?.close(); connection?.socket?.close();
}; };
// Add close handler to process event // Add close handler to process event
process.on('exit', closeHandler); process.on("exit", closeHandler);
process.on('SIGINT', closeHandler); process.on("SIGINT", closeHandler);
process.on('SIGTERM', closeHandler); process.on("SIGTERM", closeHandler);
process.on('uncaughtException', closeHandler); process.on("uncaughtException", closeHandler);
process.on('unhandledRejection', closeHandler); process.on("unhandledRejection", closeHandler);
}); }
);
}); });
// Function to start the server // Function to start the server
function startServer() { function startServer() {
if (!serverRunning) { if (!serverRunning) {
fastify.listen(2000, (err) => { fastify.listen(2000, (err) => {
if (err) { if (err) {
console.error('Error starting server:', err); console.error("Error starting server:", err);
process.exit(1); // Exit the process if server start fails process.exit(1); // Exit the process if server start fails
} }
serverRunning = true; serverRunning = true;
console.log('Server started successfully on port 2000!'); console.log("Server started successfully on port 2000!");
}); });
} else { } else {
console.log('Server is already running.'); console.log("Server is already running.");
} }
} }
@ -445,17 +446,17 @@ function stopServer() {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
fastify.close((err) => { fastify.close((err) => {
if (err) { if (err) {
console.error('Error closing server:', err); console.error("Error closing server:", err);
reject(err); reject(err);
} else { } else {
serverRunning = false; serverRunning = false;
console.log('Server closed successfully!'); console.log("Server closed successfully!");
resolve(); resolve();
} }
}); });
}); });
} else { } else {
console.log('Server is not running.'); console.log("Server is not running.");
return Promise.resolve(); return Promise.resolve();
} }
} }
@ -463,28 +464,30 @@ function stopServer() {
// Function to gracefully close and restart the server // Function to gracefully close and restart the server
function restartServer() { function restartServer() {
if (serverRunning) { if (serverRunning) {
stopServer().then(() => { stopServer()
console.log('Restarting server...'); .then(() => {
console.log("Restarting server...");
startServer(); startServer();
}).catch((error) => { })
console.error('Failed to restart server:', error); .catch((error) => {
console.error("Failed to restart server:", error);
process.exit(1); // Exit the process if server restart fails process.exit(1); // Exit the process if server restart fails
}); });
} else { } else {
console.log('Server is not running. Starting server...'); console.log("Server is not running. Starting server...");
startServer(); startServer();
} }
} }
// Add a global error handler for uncaught exceptions // Add a global error handler for uncaught exceptions
process.on('uncaughtException', (err) => { process.on("uncaughtException", (err) => {
console.error('Uncaught Exception:', err); console.error("Uncaught Exception:", err);
restartServer(); restartServer();
}); });
// Add a global error handler for unhandled promise rejections // Add a global error handler for unhandled promise rejections
process.on('unhandledRejection', (reason, promise) => { process.on("unhandledRejection", (reason, promise) => {
console.error('Unhandled Rejection at:', promise, 'reason:', reason); console.error("Unhandled Rejection at:", promise, "reason:", reason);
restartServer(); restartServer();
}); });