前言
上两篇文章,分别对产品需求和Electron端的实现做了相关的介绍,有兴趣的朋友可以看看之前的文章。本文,将介绍服务端的实现。
需求分析
之前的文章也有提到,服务端的主要任务是对数据进行转发以及生成小程序二维码。
对数据进行转发,主要是把用户在小程序端发送的消息,推送给Electron接收端。因此,我们需要创建一个websocket服务。在这里,我们选择使用ws模块。
生成小程序二维码,我们需要使用到微信的API。我们将使用request这个模块向微信服务器发起相关的请求。
创建websocket服务
入口文件代码如下:
const WebSocketServer = require('ws').Server;
const port = require('./config').port;
const log4js = require('./helpers/logger');
const logger = log4js.Logger('APP');
const wsServer = new WebSocketServer({ port });
logger.info("Server listening on port:" + port);
wsServer.on('connection', function (ws) {
logger.debug('a client connect');
ws.on('message', async function (message) {
// 消息处理逻辑
});
ws.on('close', function () {
logger.info('ws断开');
})
ws.on('error', function (error) {
ws.close();
logger.error(error);
})
});
复制代码
连接上这个服务的客户端,有可能是小程序发送端,也有可能是Electron接收端,所以我们需要对收到的消息进行了简单的类型判断。
根据之前的约定,如果消息的类型type为INIT,我们就认为是接收端初始化的消息。传递的参数中,包含接收端的UID,我们将会使用这个id作为参数,生成带参数的小程序二维码。
由于我们会在公众场合使用弹幕,所以我们需要对内容进行校验,防止一些不符合社会主义核心价值观的言论出现在弹幕中。这里,我们直接使用微信的msgSecCheck接口。
ws.on('message', async function (message) {
try {
const data = JSON.parse(message);
if (data.type === "INIT") {
logger.info('获取二维码', data);
// 将接收端的clientId作为连接的标识
ws.danmuId = data.clientId;
// 获取微信小程序二维码
const result = await WechatService.getQrcode(data.clientId);
if(result){
const respMsg = {
type: 'qrcode',
data: result
}
ws.send(JSON.stringify(respMsg));
}
} else {
logger.info('收到消息', ws.danmuId, message);
// 对消息内容进行安全性校验
const isSecurity = await WechatService.checkContentSecurity(data.content);
if(isSecurity){
// 将消息转发给对应的接收端
wsServer.clients.forEach(function (client) {
if (client.danmuId === data.scene && client.readyState === 1) {
client.send(message);
}
});
}
}
} catch (error) {
logger.error(error);
}
});
复制代码
生成带参数的小程序二维码
微信提供生成二维码的接口有三个,分别应用于不同的应用场景。具体见小程序官方文档
本项目使用getWXACodeUnlimit这个API来获取小程序码,通过该接口生成的小程序码,永久有效,数量暂无限制。
请求地址
POST https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token=ACCESS_TOKEN
复制代码
如果调用成功,会直接返回图片***二进制内容***,如果请求失败,会返回 JSON 格式的数据。
WTF,官方文档说了,会返回二进制的内容,但是为什么我的请求返回乱码呢?
既然乱码,那应该就是编解码的问题了,看看所用的请求模块request是否有相关的参数设置,果不其然:
encoding - encoding to be used on setEncoding of response data. If null, the body is returned as a Buffer. Anything else (including the default value of undefined) will be passed as the encoding parameter to toString() (meaning this is effectively utf8 by default). (Note: if you expect binary data, you should set encoding: null.)
默认是utf8,如果我们需要二进制内容,那需要将encoding设置为null。结果输出如下:
看起来是好点了,我们通过Buffer的toString方法将其转为base64,然后传给Electron端进行展示,生成二维码的功能就完成了。access_token获取与缓存
有微信开发经验的朋友都知道,基本上所有的微信API,请求的时候都需要带上access_token。而每天请求access_token的次数是有限制的,超出一定的限制则无法获取,只能坐等明天的到来。所以我们需要对access_token进行缓存。
由于我们的项目是单机部署,所以我们可以简单处理,将access_token存在内存中,每次先判断内存中的access_token是否过期,未过期则使用,反之则向微信服务器请求最新的,并同步更新到内存中。
如果考虑多实例部署,可以将access_token存储到redis中。
日志打印
鲁迅曾经说过,没有打印日志的服务都是耍流氓。日志,是后台服务不可或缺的一部分,这里我们使用log4js进行日志的打印。
最简单的使用示例:
var log4js = require('log4js');
var logger = log4js.getLogger();
logger.level = 'debug';
logger.debug("Some debug messages");
复制代码
详情请查看官方文档
GitHub项目地址
我们将陆续整理各端的代码放在GitHub上,有兴趣的可以查阅一下。
总结
本文使用ws模块搭建了一个简单的websocket服务,并使用request进行微信API的请求,用于获取小程序码等。在日志打印方面,我们使用log4js,记录一些关键信息。
下一篇文章,我们将会探讨小程序端的实现。
以上,如有错漏,欢迎指正!
@Author: TDGarden