This commit is contained in:
AdisakKanthawilang 2025-03-06 18:25:27 +07:00
parent cfcf63b9aa
commit fa073242b8
4 changed files with 56 additions and 5 deletions

View file

@ -13,7 +13,12 @@ import { OrganizationController } from "./controllers/OrganizationController";
import logMiddleware from "./middlewares/logs";
import { CommandController } from "./controllers/CommandController";
import { ProfileSalaryController } from "./controllers/ProfileSalaryController";
import { WebSocketServer } from "ws";
import http from "http";
export const wss = new WebSocketServer({ noServer: true,
path: "/api/vi/org/socket",
});
async function main() {
await AppDataSource.initialize();
@ -102,9 +107,32 @@ async function main() {
runMessageQueue();
app.listen(APP_PORT, APP_HOST, () => {
console.log(`[APP] Application is running on: http://${APP_HOST}:${APP_PORT}`);
console.log(`[APP] Swagger on: http://${APP_HOST}:${APP_PORT}/api-docs`);
const server = http.createServer(app);
// การจัดการคำขออัปเกรดจาก HTTP เป็น WebSocket
server.on("upgrade", (request:any, socket:any, head:any) => {
console.log("🔹 Handling upgrade request...");
wss.handleUpgrade(request, socket, head, (ws:any) => {
console.log("🔹 WebSocket connection established");
wss.emit("connection", ws, request);
});
});
wss.on("connection", (ws:any) => {
console.log("✅ Client connected to WebSocket");
ws.on("close", () => {
console.log("❌ Client disconnected");
});
ws.on("error", (error:any) => {
console.error("WebSocket error:", error);
});
});
server.listen(APP_PORT, APP_HOST, () => {
console.log(`[APP] Application is running on: http://localhost:${APP_PORT}`);
console.log(`[APP] Swagger on: http://localhost:${APP_PORT}/api-docs`);
console.log("[APP] HTTP Server is listening on current port");
});
}

View file

@ -55,6 +55,7 @@ export class PermissionController extends Controller {
}
let reply = await getAsync("role_" + profile.id);
console.log(">>>reply",reply);
if (reply != null) {
reply = JSON.parse(reply);
} else {
@ -84,11 +85,13 @@ export class PermissionController extends Controller {
throw new HttpError(HttpStatus.NOT_FOUND, "ไม่พบข้อมูลสิทธิ์");
}
}
console.log(">>>posMaster",posMaster);
const getDetail = await this.authRoleRepo.findOne({
select: ["id", "roleName", "roleDescription"],
where: { id: posMaster.authRoleId },
});
console.log(">>>detail",getDetail);
if (!getDetail) {
throw new HttpError(HttpStatus.NOT_FOUND, "ไม่พบข้อมูล");
}

View file

@ -31,6 +31,7 @@ import { Profile } from "../entities/Profile";
import { viewRegistryOfficer } from "../entities/view/viewRegistryOfficer";
import { viewRegistryEmployee } from "../entities/view/viewRegistryEmployee";
import { EmployeeTempPosMaster } from "../entities/EmployeeTempPosMaster";
import { sendWebSocket } from "../services/webSocket";
@Route("api/v1/org/report")
@Tags("Report")
@ -1369,6 +1370,7 @@ export class ReportController extends Controller {
}
}
}
return new HttpSuccess({ template: "report1", reportName: "report1", data: { data } });
}
@ -3537,8 +3539,10 @@ export class ReportController extends Controller {
}
}
}
const metaData = { template: "report2", reportName: "report2", data: { data } };
sendWebSocket(metaData)
return new HttpSuccess({ template: "report2", reportName: "report2", data: { data } });
// return new HttpSuccess({ template: "report2", reportName: "report2", data: { data } });
}
/**

16
src/services/webSocket.ts Normal file
View file

@ -0,0 +1,16 @@
import { WebSocket } from "ws";
import { wss } from "../app";
export async function sendWebSocket(data:any){
wss.clients.forEach((client: any) => {
if (client.readyState === WebSocket.OPEN) {
const message = JSON.stringify(data);
console.log("📤 Sending data to client:", message);
client.send(message, (err:any) => {
if (err) {
console.error("❌ Error sending message:", err);
}
});
}
});
}