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页面客户端：

以上是一个很简单的消息服务机制。但是在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。