nodejs 使用ws模块包实现指定用户 发送消息

nodejs 使用ws模块包实现指定用户 发送消息

思路

1. 前端连接（websoket） 发送自身的唯一id
2. 连接上后在 ws客户端中存下标识id
3. 添加进clients 数组
4. 再通过服务器端请求后端接口 如GET ,POST
5. 在请求中 遍历每个客户端应用 如果复合条件 则client.send返回 回去

前端方法(H5,微信小程序)

微信小程序

第一步 连接后端websoket

	
	wx.connectSocket({
		//url 是连接后端websocket的地址
		//http 就用ws  https 就用wss
		url: 'ws://localhost:4000',
		success: function() {
			console.log("connect socket success.");
		},
		fail: function() {
			console.log("connect socket fail.");
		},
	});

第二步 连接成功后 向后端发送自己的唯一标识

wx.onSocketOpen(function() {
	wx.sendSocketMessage({
	//比如你的本地存储中用来区别每个用户的是手机号  那就把手机号传上去
		data: wx.getStorageSync('usemsg').phone,
			success: () => {
				console,log("成功的回调")
  			}
	})
})

第三步 前端设置 接收返回数据

wx.onSocketMessage(function(msg) {
	console.log('这个就是返回回来的数据');
})

下面是nodejs 后端的部分

我使用的是express框架

npm i ws --save
const WebSocket = require(‘ws’); //模块引入

//开一个4000端口来存
let clients=[];	//存所有连接上的用户
const wss = new WebSocket.Server({
	port: 4000
});

//连接上就会多一个client
wss.on('connection', function connection(client, req) {
	//有数据就会触发client.on     'message'    data是传上来的数据
	client.on('message', function incoming(data) {
		//这是连接上发送过来的数据  唯一标识的phone
		console.log(data)
		//每个客户端都存上自己发来的phone做为唯一的id
		client.id = data
		//连接上后就压进数组
		clients.push(client)
	});
	
	client.on("close",(msg)=>{
		console.log("与前端断开连接")
	})
});

到这里前端连接后端的websoket 就已经连好了 唯一标识也在client中记下了

当你后端收到数据的时候

// 当有人发送消息的时候
router.post('/sendmsg', (req, res) =>{
//前端传过来的数据
	let {Pimg,Dimg,Pphone,Dphone,yuyin,face,content,imgs,nickName} = req.body
	const datalist = {
		Pimg,Dimg,Pphone,Dphone,yuyin,face,content,imgs,nickName
	}
	//发送给自己客户端的数据
	res.send(datalist)
	//循环遍历     找到你要发送的人  他的id  就发送
	wss.clients.forEach(function each(client) {
		if(Dphone==client.id){
			client.send(JSON.stringify(datalist));
		}	
	});
})

哈哈现在就可以指定发给谁消息了