hrms-bangkok/hrms-api-org
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

comment websocket

Browse source
This commit is contained in:
AdisakKanthawilang 2025-03-04 18:25:31 +07:00
parent 4647120172
commit ad6d694c51
2 changed files with 37 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

46
src/app.ts
View file

@ -12,9 +12,9 @@ import { RegisterRoutes } from "./routes";
import { OrganizationController } from "./controllers/OrganizationController"; import { OrganizationController } from "./controllers/OrganizationController";
import logMiddleware from "./middlewares/logs"; import logMiddleware from "./middlewares/logs";
import { CommandController } from "./controllers/CommandController"; import { CommandController } from "./controllers/CommandController";
import { WebSocketServer } from "ws"; // import { WebSocketServer } from "ws";
import http from "http"; // import http from "http";
export const wss = new WebSocketServer({ noServer: true }); // export const wss = new WebSocketServer({ noServer: true });
async function main() { async function main() {
await AppDataSource.initialize(); await AppDataSource.initialize();
@ -90,30 +90,30 @@ async function main() {
runMessageQueue(); runMessageQueue();
// สร้างเซิร์ฟเวอร์ HTTP // // สร้างเซิร์ฟเวอร์ HTTP
const server = http.createServer(app); // const server = http.createServer(app);
// การจัดการคำขออัปเกรดจาก HTTP เป็น WebSocket // // การจัดการคำขออัปเกรดจาก HTTP เป็น WebSocket
server.on("upgrade", (request:any, socket:any, head:any) => { // server.on("upgrade", (request:any, socket:any, head:any) => {
console.log("🔹 Handling upgrade request..."); // console.log("🔹 Handling upgrade request...");
wss.handleUpgrade(request, socket, head, (ws:any) => { // wss.handleUpgrade(request, socket, head, (ws:any) => {
console.log("🔹 WebSocket connection established"); // console.log("🔹 WebSocket connection established");
wss.emit("connection", ws, request); // wss.emit("connection", ws, request);
}); // });
}); // });
wss.on("connection", (ws:any) => { // wss.on("connection", (ws:any) => {
console.log("✅ Client connected to WebSocket"); // console.log("✅ Client connected to WebSocket");
ws.on("close", () => { // ws.on("close", () => {
console.log("❌ Client disconnected"); // console.log("❌ Client disconnected");
}); // });
}); // });
// เริ่มเซิร์ฟเวอร์ที่พอร์ต 5000 // // เริ่มเซิร์ฟเวอร์ที่พอร์ต 5000
server.listen(APP_PORT, () => { // server.listen(APP_PORT, () => {
console.log("✅ HTTP Server is listening on port 5000"); // console.log("✅ HTTP Server is listening on port 5000");
}); // });
} }

28
src/controllers/ReportController.ts
View file

@ -31,8 +31,8 @@ import { Profile } from "../entities/Profile";
import { viewRegistryOfficer } from "../entities/view/viewRegistryOfficer"; import { viewRegistryOfficer } from "../entities/view/viewRegistryOfficer";
import { viewRegistryEmployee } from "../entities/view/viewRegistryEmployee"; import { viewRegistryEmployee } from "../entities/view/viewRegistryEmployee";
import { EmployeeTempPosMaster } from "../entities/EmployeeTempPosMaster"; import { EmployeeTempPosMaster } from "../entities/EmployeeTempPosMaster";
import { WebSocket } from "ws"; // import { WebSocket } from "ws";
import { wss } from "../app"; // import { wss } from "../app";
@Route("api/v1/org/report") @Route("api/v1/org/report")
@Tags("Report") @Tags("Report")
@ -3516,19 +3516,19 @@ export class ReportController extends Controller {
} }
// console.log(">>",data); // console.log(">>",data);
wss.clients.forEach((client: any) => { // wss.clients.forEach((client: any) => {
if (client.readyState === WebSocket.OPEN) { // if (client.readyState === WebSocket.OPEN) {
const message = JSON.stringify({ template: "report2", reportName: "report2", data: { data } }); // const message = JSON.stringify({ template: "report2", reportName: "report2", data: { data } });
console.log("📤 Sending data to client:", message); // console.log("📤 Sending data to client:", message);
client.send(message, (err:any) => { // client.send(message, (err:any) => {
if (err) { // if (err) {
console.error("❌ Error sending message:", err); // console.error("❌ Error sending message:", err);
} // }
}); // });
} // }
}); // });
// return new HttpSuccess({ template: "report2", reportName: "report2", data: { data } }); return new HttpSuccess({ template: "report2", reportName: "report2", data: { data } });
} }
/** /**