nodejs消息推送之ws

74 阅读 0 评论 49 点赞

我是靠谱客的博主默默蜗牛，最近开发中收集的这篇文章主要介绍nodejs消息推送之ws，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

nodejs服务器端，如何消息的类型为test就返回消息，反之则群发：

const WebSocket = require('ws');
const wss = new WebSocket.Server({port:3030})
const connection = {}

wss.on('connection',ws => {
    ws.on('message',message => {
        console.log(JSON.stringify(connection))
        message = JSON.parse(message)
        if(message.type === 'test') {
            connection[message.id] = ws
            console.log(connection)
            console.log('received: %s',JSON.stringify(message))
            if(!!connection[message.id])
                connection[message.id].send(JSON.stringify(message))
        }else {
            ws.clients.forEach(function each(client) {
                client.send(message);
            });
        }
            
    });
})

nodejs之客户端：

const WebSocket = require('ws');
const ws = new WebSocket('ws://127.0.0.1:3030');


ws.on('open',() =>{
    let msg = {type:'test',id:1}
    ws.send(JSON.stringify(msg));
})

ws.on('error', err => {
    console.log(err)
})

ws.on('message',data => {
    console.log(data)
})

ws.on('close',(code,reason) => {
    console.log(code);
    console.log(reason+'=========='+typeof reason)
})

html页面客户端：


<!DOCTYPE html>
<html>
<head>
	<title>skynet websocket example</title>
</head>
<body>
	<script>
	var ws = new WebSocket("ws://127.0.0.1:3030");
	
	ws.onopen = function(){
		alert("open");
		ws.send("WebSocket  hellowrold!!");
	};
	ws.onmessage = function(ev){
		alert(ev.data);
	};
	ws.onclose = function(ev){
		alert("close");
	};
	ws.onerror = function(ev){
		console.log(ev);
		alert("error");
	};
	</script>
</body>
<html>

以上是一个很简单的消息服务机制。但是在nodejs中，大家都知道，是可以使用pm2启动多个进程的。在多个进程启动之后，就不能保证消息能够正常的发送到指定的客服端了。下面给大家说一下一个普遍的示例，用来解决这个问题：

单进程消息机制：只需要在本地启动appServer，客服端连接即可。要实现群聊，只需要在appServer中，转发消息即可。私聊，则在用户每一次连接的时候，使用用户的userId为key将连接保存到服务器即可。

多进程消息机制：

1，单独在服务器启一个消息服务。作为中间服务器转发功能命名：syncWsServer。

2，在本地根据express来开启一个Server，命名为appServer。首先。appServer会连接到syncWsServer服务上，并保存连接。其次在appServer中如果有用户连接，则使用用户的userId为key将此连接保存起来。

3，在发消息的时候，首先判断客服端（cliet）在appServer中是否有连接，如有连接则直接通过连接发送给用户。如果没有，将消息发送给syncWsServer。通过syncWsServer来群发消息保证，消息发送给已连接的用户。

如果我们的消息Server是启动的多个进程。假如aClient连接到的是AServer，bClient连接到的是BServer。当然AServer和BServer是同一份带码的不同进程。这个时候，aClient如果想要和bClient进行通信。则需要aClient将消息发送给AServer，AServer在服务端找不到bClient的连接，则将消息发送给syncWsServer。syncWsServer会将收到的消息群发。因为AServer和BServer都会带代码启动的时候，默认连接到syncWsServer。此时就能保证BServer能收到消息。在BServer中收到消息后，BServer会根据消息中的userId来找到bClient的连接，然后将消息正确的传送给bClient。