基于 Swoole 的微信扫码登录功能实现代码
随着微信的普及,扫码登录方式越来越被现在的应用所使用。它因为不用去记住密码,只要有微信号即可方便快捷登录。微信的开放平台原生就有支持扫码登录的功能,不过大部分人还是在用公众平台,所以扫码登录只能自行实现。这里基于微信公众平台的带参数临时二维码,并且结合 swoole 的 websocket 服务实现扫码登录。大体流程如下:
- 客户端打开登录界面,连接到 websocket 服务
- webscoket 服务生成带参数二维码返回给客户端
- 用户扫描展示的带参数二维码
- 微信服务器回调扫码事件并通知开发者服务器
- 开发者服务器通知 websocket 服务
- websocket 服务通知客户端登录成功
连接 websocket 服务
安装完 swoole 之后,我们需用使用到 websocket 服务。新建一个 websocket 服务非常简单:
$server = new swoole_websocket_server("0.0.0.0", 1099); $server->on('open', function (swoole_websocket_server $server, $request) use ($config){ echo "server: handshake success with fd{$request->fd}\n"; }); $server->on('message', function (swoole_websocket_server $server, $frame) { });
这里的 message 回调其实用不到,因为都是服务端下发消息的,但是必须设定一个。设定的端口号如果低于 1024 则必须要有 root 权限,服务器记得去防火墙开启该端口。
生成带参数二维码
websocket 服务在客户端连接成功后,需要生成一张微信的带参数二维码返回给客户端展示:
$server->on('open', function (swoole_websocket_server $server, $request) use ($config){ $app = factory::officialaccount($config['wechat']); $result = $app->qrcode->temporary($request->fd, 120); $url = $app->qrcode->url($result['ticket']); $server->push($request->fd, json_encode([ 'message_type' => 'qrcode_url', 'url' => $url ])); });
我们在 open 回调中,生成一张临时的二维码,二维码的场景值就是客户端连接的文件描述符,这样就可以保证每个客户端的唯一性.有效时间设置为 120 秒,防止一张二维码被多次扫码使用。消息 push 到客户端的时候必须要 json,方便客户端处理。客户端代码也很简单:
const socket = new websocket('ws://127.0.0.1:1099'); socket.addeventlistener('message', function (event) { var data = json.parse(event.data); if (data.message_type == 'qrcode_url'){ $('#qrcode').attr('src', data.url); } });
回调扫码事件
在客户端展示二维码后,需要提示用户扫码。对于用户扫临时的二维码,微信会触发相应的回调事件,我们需要在该回调事件中处理用户的扫码行为。其中我们需要用到微信传递过来的一些参数:
fromusername 发送方帐号(一个openid) msgtype 消息类型,event event 事件类型,subscribe eventkey 事件 key 值,qrscene_为前缀,后面为二维码的参数值
这里要注意一点:微信已关注扫码推送的 eventkey 是没有 qrscene_ 前缀的,只有未关注扫码然后关注才有。
收到微信回调后我们首先要根据不同的事件类型做不同处理:
if ($message['msgtype'] == 'event'){ if ($message['event'] == 'subscribe'){ //关注 return $this->subscribe($message); } if ($message['event'] == 'unsubscribe') { //取消关注 return $this->unsubscribe($message); } if ($message['event'] == 'scan'){ //已关注扫码 return $this->scan($message); } }else{ return "您好!欢迎使用 swoolewechat 扫描登录"; }
这里只讲解一个关注事件的业务逻辑,其他根据需要自行编码:
public function subscribe($message){ $eventkey = intval(str_replace('qrscene_', '', $message['eventkey'])); $openid = $message['fromusername']; $user = $this->app->user->get($openid); $this->notify(json_encode([ 'type' => 'scan', 'fd' => $eventkey, 'nickname' => $user['nickname'] ])); $count = $this->count($openid); $msgtemp = "%s,登录成功!\n这是你第%s次登录,玩的开心!"; return sprintf($msgtemp, $user['nickname'], $count); }
这里的 eventkey 实际上就是连接 websocket 的客户端文件描述符,获取到扫码用户的 open_id ,根据用户的 open_id 获取用户信息,通知 websocket 服务,响应文本消息给微信。
这里一个比较麻烦的点就是如何通知 websocket 服务,我们知道处理微信回调的代码是是不在 websocket 服务上的,那么不同 server 间如何通信呢?swoole 官方给出的解决方案有两个:
- 额外监听一个udp端口
- 使用 swoole_client 作为客户端访问 server
这里我们选择第二个方案,swoole 1.8 版本支持一个 server 监听多个端口,我们在 websocket 服务新增监听一个 tcp 的端口:
$tcp_server = $server->addlistener('0.0.0.0', 9999, swoole_sock_tcp); $tcp_server->set([]); $tcp_server->on('receive', function ($serv, $fd, $threadid, $data) { });
主服务器是 websocket 或 http 协议,新监听的 tcp 端口默认会继承主 server 的协议设置,必须单独调用 set 方法设置新的协议才会启用新协议
然后我们就可以在扫码回调的进程中去通知 websocket 服务:
public function notify($message){ $client = new swoole_client(swoole_sock_tcp); if (!$client->connect('127.0.0.1', $this->config['notify_port'], -1)) { return "connect failed. error: {$client->errcode}\n"; } $ret = $client->send($message); }
通知登录成功
在 websocket 服务收到登录成功的通知后,就可以根据需要处理一下用户信息,然后把用户信息传递给客户端的浏览器展示结果,还记得我们刚刚新监听的 tcp 端口吗?就可以在 receive 事件中处理:
$tcp_server->on('receive', function ($serv, $fd, $threadid, $data) { $data = json_decode($data, true); if ($data['type'] == 'scan'){ $serv->push($data['fd'], json_encode([ 'message_type' => 'scan_success', 'user' => $data['nickname'] ])); } $serv->close($fd); });
最后登录的界面:
总结
整个过程并不难,主要的两个难点就是对应连接用户的扫码用户、不同 server 之间的通信,我们的解决办法就是把连接的文件描述符作为临时二维码场景值(这里也可以采用 redis 来存储映射关系)、监听新的 tcp 端口来接受通知消息。可以访问 试试看,记得要用电脑打开。
以上所述是小编给大家介绍的基于 swoole 的微信扫码登录功能实现代码,希望对大家有所帮助