cgserver 8.6.8 → 8.6.10

package/dist/lib/Framework/SocketServer/ISocketServer.js

@@ -1,202 +1,205 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ISocketServer = void 0;
-const MysqlManager_1 = require("../Database/MysqlManager");
-const RedisManager_1 = require("../Database/RedisManager");
-const ws = require("websocket");
-const Log_1 = require("./../Logic/Log");
-const fs = require("fs");
-const http = require("http");
-const https = require("https");
-const MongoManager_1 = require("../Database/MongoManager");
-const MSSqlManager_1 = require("../Database/MSSqlManager");
-const EventTool_1 = require("../Logic/EventTool");
-const IServerConfig_1 = require("../Config/IServerConfig");
-const cgserver_1 = require("../cgserver");
-class ISocketServer {
-    _cfg = null;
-    //方便提示
-    get cfg() {
-        return this._cfg;
-    }
-    //服务器已被关闭
-    _is_closed = false;
-    get isClosed() {
-        return this._is_closed;
-    }
-    get name() {
-        return this._cfg.serverName;
-    }
-    //监听websocket
-    _listening_websocket = null;
-    get listeningWebSocket() {
-        return this._listening_websocket;
-    }
-    //来自于用户的链接
-    _ws_clients = {};
-    get listenPort() {
-        return this._cfg.port;
-    }
-    /**
-     * websocket can accepted protocol
-     */
-    _accepted_protocol = null;
-    //服务器管理相关
-    /*
-        item=servername:class
-    */
-    _name_vs_class = {};
-    registerWebSocketHandleClass(name, cls) {
-        if (!name || name == "") {
-            throw "registerWebSocketHandleClass name must be not null or empty!";
-        }
-        this._name_vs_class[name] = cls;
-    }
-    constructor(cfg) {
-        this._cfg = cfg;
-    }
-    removeServerWebSocketBySocketId(socketId) {
-        this._ws_clients[socketId] = null;
-        delete this._ws_clients[socketId];
-    }
-    async run() {
-        cgserver_1.GCgServer.addSocketServer(this);
-        await MSSqlManager_1.GMSSqlMgr.init();
-        await MysqlManager_1.GMysqlMgr.init();
-        await RedisManager_1.GRedisMgr.init(IServerConfig_1.GServerCfg.db?.redis);
-        await MongoManager_1.GMongoMgr.init(IServerConfig_1.GServerCfg.db?.mongo);
-        this.initWebSocket();
-    }
-    stop() {
-        this._is_closed = true;
-        this._listening_websocket.closeAllConnections();
-    }
-    /*
-        把所有的客户端链接保存起来
-        方便统计，广播等
-    */
-    addClient(ws_client) {
-        this._ws_clients[ws_client.socketId] = ws_client;
-    }
-    isOriginAllowed(origin) {
-        return true;
-    }
-    initWebSocket(wss) {
-        let server = null;
-        if (wss) {
-            console.log("wss:---");
-            let options = {
-                key: fs.readFileSync("ssl/ssl2.key"),
-                cert: fs.readFileSync("ssl/ssl2.crt"),
-                //passphrase:'1234'//如果秘钥文件有密码的话，用这个属性设置密码
-            };
-            server = https.createServer(options, (request, response) => {
-                Log_1.GLog.info((new Date()) + 'wss Received request for ' + request.url);
-                response.writeHead(404);
-                response.end();
-            });
-        }
-        else {
-            server = http.createServer((request, response) => {
-                Log_1.GLog.info((new Date()) + 'ws Received request for ' + request.url);
-                response.writeHead(404);
-                response.end();
-            });
-        }
-        server.listen(this._cfg.port, this.onListenning.bind(this));
-        this._listening_websocket = new ws.server({
-            httpServer: server,
-            // You should not use autoAcceptConnections for production 
-            // applications, as it defeats all standard cross-origin protection 
-            // facilities built into the protocol and the browser.  You should 
-            // *always* verify the connection's origin and decide whether or not 
-            // to accept it. 
-            autoAcceptConnections: false
-        });
-        this._listening_websocket.on('request', this.onRequest.bind(this));
-        this._listening_websocket.on('close', this.onClose.bind(this));
-    }
-    onClose(connection, reason, desc) {
-    }
-    onListenning() {
-        EventTool_1.GEventTool.emit("socket_server_init_done");
-        let info = (new Date()) + "  Server " + this.name + " is listening on port " + this._cfg.port;
-        Log_1.GLog.info(info);
-        console.log(info);
-    }
-    onRequest(req) {
-        if (this._is_closed) {
-            return;
-        }
-        let protocol = null;
-        if (req.requestedProtocols.length > 0) {
-            protocol = req.requestedProtocols[0];
-        }
-        let allowed = this.isOriginAllowed(req.origin);
-        if (!allowed) {
-            req.reject();
-            Log_1.GLog.info(' Connection from origin ' + req.origin + ' rejected.');
-            return;
-        }
-        try {
-            let conn = req.accept(this._accepted_protocol, req.origin);
-            if (!conn) {
-                Log_1.GLog.info(' protocol reject');
-                return;
-            }
-            Log_1.GLog.info((new Date()) + ' Connection accepted.');
-            let server_name = this._getServerNameByCookies(req.cookies);
-            this.createWebSocketObjectByProtocol(server_name, conn);
-        }
-        catch (e) {
-            Log_1.GLog.info(' protocol reject');
-        }
-    }
-    createWebSocketObjectByProtocol(server_name, _ws) {
-        server_name = server_name || "default";
-        let cls = this._name_vs_class[server_name];
-        if (!cls) {
-            Log_1.GLog.error("(createWebSocketObjectByProtocol in server(" + this.name + "))no this websocket handle class=" + server_name);
-            return null;
-        }
-        let ws_server = (new cls(this));
-        this.addClient(ws_server);
-        ws_server.onConnect(_ws);
-        return ws_server;
-    }
-    _getServerNameByCookies(cookies) {
-        let server_name = "";
-        if (!cookies) {
-            return server_name;
-        }
-        for (let index in cookies) {
-            let cookie = cookies[index];
-            if (cookie.name == "server") {
-                server_name = cookie.value;
-                break;
-            }
-        }
-        return server_name;
-    }
-    /**
-     * 广播消息
-     * @param msg
-     */
-    broadCast(msg) {
-        for (var key in this._ws_clients) {
-            let ws = this._ws_clients[key];
-            ws.send(msg);
-        }
-    }
-    /**
-     * 获取任意客户端连接
-     * @returns
-     */
-    getAnyWebSocket() {
-        for (var key in this._ws_clients) {
-            return this._ws_clients[key];
-        }
-        return null;
-    }
-}
-exports.ISocketServer = ISocketServer;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ISocketServer = void 0;
+const MysqlManager_1 = require("../Database/MysqlManager");
+const RedisManager_1 = require("../Database/RedisManager");
+const ws = require("websocket");
+const Log_1 = require("./../Logic/Log");
+const fs = require("fs");
+const http = require("http");
+const https = require("https");
+const MongoManager_1 = require("../Database/MongoManager");
+const MSSqlManager_1 = require("../Database/MSSqlManager");
+const EventTool_1 = require("../Logic/EventTool");
+const IServerConfig_1 = require("../Config/IServerConfig");
+const cgserver_1 = require("../cgserver");
+class ISocketServer {
+    _cfg = null;
+    //方便提示
+    get cfg() {
+        return this._cfg;
+    }
+    //服务器已被关闭
+    _is_closed = false;
+    get isClosed() {
+        return this._is_closed;
+    }
+    get name() {
+        return this._cfg.serverName;
+    }
+    //监听websocket
+    _listening_websocket = null;
+    get listeningWebSocket() {
+        return this._listening_websocket;
+    }
+    //来自于用户的链接
+    _ws_clients = {};
+    get listenPort() {
+        return this._cfg.port;
+    }
+    /**
+     * websocket can accepted protocol
+     */
+    _accepted_protocol = null;
+    //服务器管理相关
+    /*
+        item=servername:class
+    */
+    _name_vs_class = {};
+    registerWebSocketHandleClass(name, cls) {
+        if (!name || name == "") {
+            throw "registerWebSocketHandleClass name must be not null or empty!";
+        }
+        this._name_vs_class[name] = cls;
+    }
+    constructor(cfg) {
+        this._cfg = cfg;
+    }
+    removeServerWebSocketBySocketId(socketId) {
+        this._ws_clients[socketId] = null;
+        delete this._ws_clients[socketId];
+    }
+    async run() {
+        cgserver_1.GCgServer.addSocketServer(this);
+        let dbcfg = IServerConfig_1.GServerCfg.db;
+        if (dbcfg) {
+            await RedisManager_1.GRedisMgr.init(dbcfg.redis);
+            await MysqlManager_1.GMysqlMgr.init(dbcfg.mysql);
+            await MSSqlManager_1.GMSSqlMgr.init(dbcfg.mssql);
+            await MongoManager_1.GMongoMgr.init(dbcfg.mongo);
+        }
+        this.initWebSocket();
+    }
+    stop() {
+        this._is_closed = true;
+        this._listening_websocket.closeAllConnections();
+    }
+    /*
+        把所有的客户端链接保存起来
+        方便统计，广播等
+    */
+    addClient(ws_client) {
+        this._ws_clients[ws_client.socketId] = ws_client;
+    }
+    isOriginAllowed(origin) {
+        return true;
+    }
+    initWebSocket(wss) {
+        let server = null;
+        if (wss) {
+            console.log("wss:---");
+            let options = {
+                key: fs.readFileSync("ssl/ssl2.key"),
+                cert: fs.readFileSync("ssl/ssl2.crt"),
+                //passphrase:'1234'//如果秘钥文件有密码的话，用这个属性设置密码
+            };
+            server = https.createServer(options, (request, response) => {
+                Log_1.GLog.info((new Date()) + 'wss Received request for ' + request.url);
+                response.writeHead(404);
+                response.end();
+            });
+        }
+        else {
+            server = http.createServer((request, response) => {
+                Log_1.GLog.info((new Date()) + 'ws Received request for ' + request.url);
+                response.writeHead(404);
+                response.end();
+            });
+        }
+        server.listen(this._cfg.port, this.onListenning.bind(this));
+        this._listening_websocket = new ws.server({
+            httpServer: server,
+            // You should not use autoAcceptConnections for production 
+            // applications, as it defeats all standard cross-origin protection 
+            // facilities built into the protocol and the browser.  You should 
+            // *always* verify the connection's origin and decide whether or not 
+            // to accept it. 
+            autoAcceptConnections: false
+        });
+        this._listening_websocket.on('request', this.onRequest.bind(this));
+        this._listening_websocket.on('close', this.onClose.bind(this));
+    }
+    onClose(connection, reason, desc) {
+    }
+    onListenning() {
+        EventTool_1.GEventTool.emit("socket_server_init_done");
+        let info = (new Date()) + "  Server " + this.name + " is listening on port " + this._cfg.port;
+        Log_1.GLog.info(info);
+        console.log(info);
+    }
+    onRequest(req) {
+        if (this._is_closed) {
+            return;
+        }
+        let protocol = null;
+        if (req.requestedProtocols.length > 0) {
+            protocol = req.requestedProtocols[0];
+        }
+        let allowed = this.isOriginAllowed(req.origin);
+        if (!allowed) {
+            req.reject();
+            Log_1.GLog.info(' Connection from origin ' + req.origin + ' rejected.');
+            return;
+        }
+        try {
+            let conn = req.accept(this._accepted_protocol, req.origin);
+            if (!conn) {
+                Log_1.GLog.info(' protocol reject');
+                return;
+            }
+            Log_1.GLog.info((new Date()) + ' Connection accepted.');
+            let server_name = this._getServerNameByCookies(req.cookies);
+            this.createWebSocketObjectByProtocol(server_name, conn);
+        }
+        catch (e) {
+            Log_1.GLog.info(' protocol reject');
+        }
+    }
+    createWebSocketObjectByProtocol(server_name, _ws) {
+        server_name = server_name || "default";
+        let cls = this._name_vs_class[server_name];
+        if (!cls) {
+            Log_1.GLog.error("(createWebSocketObjectByProtocol in server(" + this.name + "))no this websocket handle class=" + server_name);
+            return null;
+        }
+        let ws_server = (new cls(this));
+        this.addClient(ws_server);
+        ws_server.onConnect(_ws);
+        return ws_server;
+    }
+    _getServerNameByCookies(cookies) {
+        let server_name = "";
+        if (!cookies) {
+            return server_name;
+        }
+        for (let index in cookies) {
+            let cookie = cookies[index];
+            if (cookie.name == "server") {
+                server_name = cookie.value;
+                break;
+            }
+        }
+        return server_name;
+    }
+    /**
+     * 广播消息
+     * @param msg
+     */
+    broadCast(msg) {
+        for (var key in this._ws_clients) {
+            let ws = this._ws_clients[key];
+            ws.send(msg);
+        }
+    }
+    /**
+     * 获取任意客户端连接
+     * @returns
+     */
+    getAnyWebSocket() {
+        for (var key in this._ws_clients) {
+            return this._ws_clients[key];
+        }
+        return null;
+    }
+}
+exports.ISocketServer = ISocketServer;