原生html+js Websocket鏈接原生node建立web服務器和socket服務進行通訊 socket服務原理

時間 2021-02-17

標籤 javascript html java node web 數組瀏覽器服務器 websocket socket 欄目 HTML 简体版

原文原文鏈接

最近項目要用到websocket技術了，重溫一下當初學習寫的DEMOjavascript

1.爲了運行h5頁面，須要利用NODE開啓一個web服務器，端口號8888，取名爲 web.js , 運行node web.jshtml

var fs = require('fs');
var events = require('events');
var http = require('http');
var url = require('url');java

// 建立服務器
http.createServer(function (request, response) {node

    // 解析請求，包括文件名
    var pathname = url.parse(request.url).pathname;
    // 輸出請求的文件名
    console.log("Request for " + pathname + " received.", pathname.substr(1));web

    // 從文件系統中讀取請求的文件內容
    fs.readFile(pathname.substr(1), function (err, data) {
        if (err) {
            console.log(err);
            // HTTP 狀態碼: 404 : NOT FOUND
            // Content Type: text/plain
            response.writeHead(404, { 'Content-Type': 'text/html;charset=utf-8' });
            response.write('頁面不存在！');
        } else {
            // HTTP 狀態碼: 200 : OK
            // Content Type: text/plain
            response.writeHead(200, { 'Content-Type': 'text/html;charset=utf-8' });數組

            // 響應文件內容
            response.write(data.toString());
        }瀏覽器

        // 發送響應數據
        response.end();
    });服務器

}).listen(8888);websocket

// 控制檯會輸出如下信息
console.log('Server running at http://127.0.0.1:8888/');socket

而後在web.js同文件夾下建立h5頁面取名 index.html，利用上面建立的web服務打開頁面，http://127.0.0.1:8888/index.html，爲了測試，能夠打開兩個不一樣的瀏覽器，打開同一個地址，但如今還不能通訊，須要第三部建立scoket服務器，建立完成，點擊兩個瀏覽器的運行按鈕，而後輸入內容發送，另外一個頁面也實時獲取到了你發的消息

<!DOCTYPE HTML>
<html>
<head>
    <meta charset="utf-8">
    <title>webSocket測試</title>
</head>
<body>
    <div id="sse">
        <a href="javascript:WebSocketTest()">運行(鏈接服務，等於帳號登陸) WebSocket
    </div>
    <input type="text" id="inputBox" />
    <button id="submitFunc">發送</button><button onclick="closeWs()">關閉</button>
    <div id="content"></div>
    <script type="text/javascript">
        var ws = null;
        var inputBox = document.getElementById('inputBox');
        var submitFunc = document.getElementById('submitFunc');
        var content = document.getElementById('content');
        function WebSocketTest() {
            if ("WebSocket" in window) {
                alert("您的瀏覽器支持 WebSocket!");
                // 打開一個 web socket
                ws = new WebSocket("ws://localhost:8124");
                ws.onopen = function () {
                    // Web Socket 已鏈接上，使用 send() 方法發送數據
                    ws.send("笑傲江湖");
                    alert("數據發送中...");
                    submitFunc.onclick = function () {
                        ws.send(inputBox.value);
                        alert("數據發送中...");
                    }
                };
                ws.onmessage = function (evt) {
                    var received_msg = evt.data;
                    console.log('接受的數據', received_msg);
                    var html = content.innerHTML;
                    html += received_msg;
                    content.innerHTML = '<p>' + html + '</p>';
                    //alert("數據已接收...");
                };

                ws.onclose = function () {
                    // 關閉 websocket
                    alert("鏈接已關閉...");
                };
            } else {
                // 瀏覽器不支持 WebSocket
                alert("您的瀏覽器不支持 WebSocket!");
            }
        }

        //獲取信息發送
        function closeWs() {
            if (ws != null) {
                ws.close();
            } else {
                alert('你還沒開啓');
            }
        }
    </script>
</body>
</html>

3.利用net模塊建立socket服務，在同文件下下建立server.js ,代碼以下，運行 node server 端口號：8124 (端口號隨意，只是index.html裏的地址端口一致就能夠)

const crypto = require('crypto');
const net = require('net');
var clientList = [];
var this_client = null;
var options = {
allowHalfOpen: false,
pauseOnConnect: false
}

let tcpServer = net.createServer(options);
//計算websocket校驗
function getSecWebSocketAccept(key) {
    return crypto.createHash('sha1')
        .update(key + '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')
        .digest('base64');
}

//掩碼操做
function unmask(buffer, mask) {
    const length = buffer.length;
    for (var i = 0; i < length; i++) {
        buffer[i] ^= mask[i & 3];
    }
    console.log('解碼後的數據', buffer.toString('utf8'));
}

tcpServer.on('listening', () => {
console.log('開始監聽');
});

tcpServer.on('connection', (socket) => {
    console.log('鏈接已創建' + 'n', socket.name);
    //啓動心跳機制
    /*var isOnline = !0;
    var keepAliveTimer = socket.timer = setInterval(()=>{
    if(!isOnline){
    this_client = socket;
    quit(socket.nick);
    return;
    }

    if(socket.writable){
    isOnline = !1;
    socket.write('::');
    }else{
    this_client = socket;
    quit(socket.nick);
    }
    },3000);

    if(socket._handle==null){
    isOnline = !0;
    return;
    }*/

    tcpServer.getConnections((err, count) => {
        if (err) {
            console.warn(err);
        } else {
            console.log(當前有${count}個鏈接);
        }
    });

    socket.on('data', (data) => {
        this_client = socket;
        if (clientList.indexOf(socket) > -1) {
            let buffer = data;
            let fin = (buffer[0] & 0b10000000) === 0b10000000;
            //取第一個字節的後四位，獲得的一個是十進制數
            let opcode = buffer[0] & 0b00001111;
            //取第二個字節的第一位是不是1，判斷是否掩碼操做
            let mask = buffer[1] & 0b100000000 === 0b100000000;
            //載荷數據的長度
            let payloadLength = buffer[1] & 0b01111111;
            //掩碼鍵，佔4個字節
            let maskingKey = buffer.slice(2, 6);
            //載荷數據，就是客戶端發送的實際數據
            let payloadData = buffer.slice(6);
            console.log('客戶端發送的實際數據', payloadData.toString('utf8'));
            //對數據進行解碼處理
            unmask(payloadData, maskingKey);
            //向客戶端響應數據
            let send = Buffer.alloc(2 + payloadData.length);
            //0b10000000表示發送結束
            send[0] = opcode | 0b10000000;
            //載荷數據的長度
            send[1] = payloadData.length;
            payloadData.copy(send, 2);
            var now = new Date();
            broadcast(send, socket);
            /*const buf2 = Buffer.from('後臺傳過來的時間：'+now, 'utf8');
            const buf=Buffer.alloc(2+buf2.length);
            buf[0]=opcode | 0b10000000;
            buf[1]=buf2.length;
            buf2.copy(buf,2);*/
            /*for(var i=0;i<clientList.length;i++){
            let client=clientList[i];
            if(client._handle==null){
            clientList.splice(clientList.indexOf(client), 1);
            i--;
           }
            }*/

            /* if(send=='end'){
            this.close();
            }
            for(var i=0;i<clientList.length;i++){
            console.log('剩餘個數：',clientList.length);
            let client=clientList[i];
            //console.log('返回數據中---',client);
            console.log(send);
            client.write(send);
            }*/

        } else {
            data = data.toString();
            if (data.match(/Upgrade: websocket/)) {
                let rows = data.split('rn');

//去掉第一行的請求行

                //去掉請求頭的尾部兩個空行
                rows = rows.slice(1, -2);
                let headers = {};
                rows.forEach(function (value) {
                    let [k, v] = value.split(': ');
                    headers[k] = v;
                });

                //判斷websocket的版本
                if (headers['Sec-WebSocket-Version'] == 13) {
                    let secWebSocketKey = headers['Sec-WebSocket-Key'];

//計算websocket校驗
let secWebSocketAccept = getSecWebSocketAccept(secWebSocketKey);

                    //服務端響應的內容
                    let res =
                        'HTTP/1.1 101 Switching Protocols rn' +
                        'Upgrade: websocket rn' +
                        'Sec-WebSocket-Accept: ' + secWebSocketAccept + 'rn' +
                        'Connection: Upgrade rn' +
                        'rn';
                    console.log('發送給客戶端協議', res);
                    //給客戶端發送響應內容
                    socket.write(res);
                    //socket.name = socket.remoteAddress + ':' + socket.remotePort;
                    clientList.push(socket);
                }
            }
        }
    });

    socket.on('disconnect', function () { // 這裏監聽 disconnect，就能夠知道誰斷開鏈接了
        console.log('disconnect: ' + socket.id);
    });
    socket.on('error', (err) => {
        console.warn('錯誤', err);
        socket.destroy();
    });

    /*socket.on('close', function(data) {
    console.log('客戶端關閉了!',data);
    //clientList.splice(clientList.indexOf(socket), 1);
    // socket.remoteAddress + ' ' + socket.remotePort);
    });*/

    //結束
    socket.on('end', () => {
        console.log('' + socket + '-quit'); //若是某個客戶端斷開鏈接，node控制檯就會打印出來
        //this.destroy();
        clientList.splice(clientList.indexOf(socket), 1);
    });

        /刪除掉服務器的客戶端數組中，已斷開的客戶端/
        for (var i = 0; i < cleanup.length; i++) {
            clientList.splice(clientList.indexOf(cleanup[i]), 1);
        }
    }
});

tcpServer.on('close', () => {
console.log('close');
});

tcpServer.listen(8124, () => { console.log("8124 服務器 ok");});