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

commend websocket

Browse source
This commit is contained in:
AdisakKanthawilang 2025-03-05 12:26:01 +07:00
parent fc9e68e5aa
commit 46fad04ee0
2 changed files with 56 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

63
src/app.ts
View file

@ -12,15 +12,38 @@ 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 });
// export const wss = new WebSocketServer({
// port: 8080,
// perMessageDeflate: {
// zlibDeflateOptions: {
// // See zlib defaults.
// chunkSize: 1024,
// memLevel: 7,
// level: 3
// },
// zlibInflateOptions: {
// chunkSize: 10 * 1024
// },
// // Other options settable:
// clientNoContextTakeover: true, // Defaults to negotiated value.
// serverNoContextTakeover: true, // Defaults to negotiated value.
// serverMaxWindowBits: 10, // Defaults to negotiated value.
// // Below options specified as default values.
// concurrencyLimit: 10, // Limits zlib concurrency for perf.
// threshold: 1024 // Size (in bytes) below which messages
// // should not be compressed if context takeover is disabled.
// }
// });
async function main() { async function main() {
await AppDataSource.initialize(); await AppDataSource.initialize();
const app = express(); const app = express();
const server = http.createServer(app); // const server = http.createServer(app);
app.use( app.use(
cors({ cors({
@ -72,24 +95,24 @@ async function main() {
}); });
// การจัดการคำขออัปเกรดจาก 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");
}); // });
}); // });
// app.listen(APP_PORT, APP_HOST, () => console.log(`Listening on: http://localhost:${APP_PORT}`)); // app.listen(APP_PORT, APP_HOST, () => console.log(`Listening on: http://localhost:${APP_PORT}`));
server.listen( app.listen(
APP_PORT, APP_PORT,
APP_HOST, APP_HOST,
() => ( () => (
@ -112,7 +135,7 @@ async function main() {
// เริ่มเซิร์ฟเวอร์ที่พอร์ตปัจจุบัน // เริ่มเซิร์ฟเวอร์ที่พอร์ตปัจจุบัน
// server.listen(APP_PORT, () => { // server.listen(8080, () => {
// console.log("[APP] HTTP Server is listening on current port"); // console.log("[APP] HTTP Server is listening on current port");
// }); // });

26
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")
@ -3565,17 +3565,17 @@ 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 } });
} }