接上一篇文章的话题,我们这次来讲一下如何建立一个基于node.js的WebSocket服务器,并在小程序中使用这个提供实时服务的服务器。
node.js中已经有很多现成的第三方库,用于构建WebSocket服务。我们今天选用一个叫做websocket的库,来构建一个可以提供标准WebSocket接口的node.js服务端程序。
先建一个空文件夹,名叫ws-server,然后进入该文件夹,在命令行执行:
npm install websocket
然后在ws-server文件夹下,再新建一个名为app.js的文件,内容如下:
const http = require('http')
const WebSocketServer = require('websocket').server
const httpServer = http.createServer((request, response) => {
console.log('[' + new Date + '] Received request for ' + request.url)
response.writeHead(404)
response.end()
})
const wsServer = new WebSocketServer({
httpServer,
autoAcceptConnections: true
})
wsServer.on('connect', connection => {
connection.on('message', message => {
if (message.type === 'utf8') {
console.log('>> message content from client: ' + message.utf8Data)
connection.sendUTF('[from server] ' + message.utf8Data)
}
}).on('close', (reasonCode, description) => {
console.log('[' + new Date() + '] Peer ' + connection.remoteAddress + ' disconnected.')
})
})
httpServer.listen(8080, () => {
console.log('[' + new Date() + '] Serveris listening on port 8080')
})
由于WebSocket服务是建立在HTTP之上的,因此我们从上面的代码中可以看到,代码先建立了一个http server, 然后再建立了一个使用该http server的WebSocket server,并让http server监听8080端口对外提供服务。
这个服务端的功能也很简单,就是收到客户端发送的消息并打印出来,然后在接收到的消息前面加上一个[from server]的前缀后,返回给客户端。
写完这个app.js后,通过以下命令把它运行起来:
node app.js
好,接着我们来实现一个调用该WebSocket服务的微信小程序端的代码:
Page({
onLoad: function (options) {
wx.connectSocket({
url: 'ws://localhost:8080'
})
wx.onSocketOpen(function (res) {
console.log('WebSocket连接已打开!')
wx.sendSocketMessage({
data: 'Hello,World:' + Math.round(Math.random() * 0xFFFFFF).toString(),
})
})
wx.onSocketMessage(function (res) {
console.log(res)
})
wx.onSocketClose(function (res) {
console.log('WebSocket连接已关闭!')
})
}
})
上面的代码中,我们首先需要用wx.connectSocket()方法去连接目标服务器,因为我们的开发环境用的是非安全的http,所以这里的url参数是ws://打头的。在微信的实际运行环境中,你的服务端必须使用SSL,因此连接url就会改成wss://。
接着,需要调用wx.onSocketOpen()方法来设置WebSocket连接打开时的回调函数。当连接打开后,就可以开始向服务端发送数据了,我们在这里使用wx.sendSocketMessage()方法,向服务端发送了一个后面跟随一个随机数的Hello,World字符串。我们运行一下程序,可以看到,服务端的控制台上会打印出这样的结果:
>> message content from client: Hello,World:16119566
说明服务端已经成功接收到了客户端发送的字符串消息。
之后服务端会向客户端再反馈这个消息,那客户端这边如何接收这个从服务端过来的消息呢?我们可以在小程序中,使用wx.onSocketMessage()方法,监听服务端发送到客户端的消息,正如我们上面的示例代码写的那样:
wx.onSocketMessage(function (msg) {
console.log(msg)
})
我们在小程序的Console上,简单的打印了从服务端过来的消息,如下所示:
这样,一个简单但完整的客户端和服务器端的WebSocket交互就算完成了。如果你想关闭这个WebSocket连接,可以调用wx.closeSocket()来进行关闭。
好了,关于微信小程序和WebSocket服务器通信的例子,就讲到这里吧。不对之处,欢迎指正。
谢谢大家阅读本文。